Timezone: »
In many medical segmentation tasks, it is crucial to provide valid confidence intervals to machine learning predictions. In the case of segmenting amniotic fluid using fetal MRIs, this allows doctors to better understand and control the segmentation masks, bound the fluid volume, and statistically detect anomalies such as cysts. In this work, we propose and evaluate different ways of creating confidence intervals for segmentation masks and volume predictions using tools from the field of conformal prediction. We show that simple but well-suited modifications of current methods, such as volume normalization and tuning of a leniency hyperparameter, lead to significant improvements, resulting in more consistent coverage and narrower confidence sets. These advances are thoroughly illustrated in the amniotic fluid segmentation problem.
Author Information
Daniel Csillag (IMPA)
Lucas Monteiro Paes (IMPA)
Thiago Ramos (IMPA)
João Vitor Romano (Instituto de Matemática Pura e Aplicada)
Roberto Oliveira (IMPA)
Paulo Orenstein (Instituto de Matemática Pura e Aplicada)
More from the Same Authors
-
2021 Poster: Online Learning with Optimism and Delay »
Genevieve Flaspohler · Francesco Orabona · Judah Cohen · Soukayna Mouatadid · Miruna Oprescu · Paulo Orenstein · Lester Mackey -
2021 Spotlight: Online Learning with Optimism and Delay »
Genevieve Flaspohler · Francesco Orabona · Judah Cohen · Soukayna Mouatadid · Miruna Oprescu · Paulo Orenstein · Lester Mackey