The method to measure the uncertainty of the model could be improved: - adapt corrective weights and transformations (cf Euclidean Distance Transform) - cross-entropy, mutual information, ... instead of variance - Ensembles instead of MC-dropout