Ensemble models have emerged as a powerful technique for improving robustness in medical image segmentation. However, traditional ensembles suffer from limitations such as under-confidence and over-reliance on poor performing models. In this work, we introduce an Adaptive Uncertainty-based Ensemble (AUE) model for tumor segmentation in histopathological slides. Our approach leverages uncertainty estimates from Monte Carlo dropout during testing to dynamically select the optimal pair of models for each whole slide image. The AUE model combines predictions from the two most reliable models (K-Net, ResNeSt, Segformer, Twins), identified through uncertainty quantification, to enhance segmentation performance. We validate the AUE model on the ACDC@LungHP challenge dataset, systematically comparing it against state-of-the-art approaches. Results demonstrate that our uncertainty-guided ensemble achieves a mean Dice score of 0.8653 and outperforms traditional ensemble techniques and top-ranked methods from the challenge by over 3 %. Our adaptive ensemble approach provides accurate and reliable lung tumor delineation in histopathology images by managing model uncertainty.
A dynamic uncertainty-aware ensemble model: Application to lung cancer segmentation in digital pathology / Salvi, Massimo; Mogetta, Alessandro; Raghavendra, U.; Gudigar, Anjan; Acharya, U. Rajendra; Molinari, Filippo. - In: APPLIED SOFT COMPUTING. - ISSN 1568-4946. - STAMPA. - 165:(2024). [10.1016/j.asoc.2024.112081]
A dynamic uncertainty-aware ensemble model: Application to lung cancer segmentation in digital pathology
Salvi, Massimo;Mogetta, Alessandro;Molinari, Filippo
2024
Abstract
Ensemble models have emerged as a powerful technique for improving robustness in medical image segmentation. However, traditional ensembles suffer from limitations such as under-confidence and over-reliance on poor performing models. In this work, we introduce an Adaptive Uncertainty-based Ensemble (AUE) model for tumor segmentation in histopathological slides. Our approach leverages uncertainty estimates from Monte Carlo dropout during testing to dynamically select the optimal pair of models for each whole slide image. The AUE model combines predictions from the two most reliable models (K-Net, ResNeSt, Segformer, Twins), identified through uncertainty quantification, to enhance segmentation performance. We validate the AUE model on the ACDC@LungHP challenge dataset, systematically comparing it against state-of-the-art approaches. Results demonstrate that our uncertainty-guided ensemble achieves a mean Dice score of 0.8653 and outperforms traditional ensemble techniques and top-ranked methods from the challenge by over 3 %. Our adaptive ensemble approach provides accurate and reliable lung tumor delineation in histopathology images by managing model uncertainty.File | Dimensione | Formato | |
---|---|---|---|
(2024) paper - AUE lung cancer.pdf
accesso aperto
Tipologia:
2a Post-print versione editoriale / Version of Record
Licenza:
Creative commons
Dimensione
7.28 MB
Formato
Adobe PDF
|
7.28 MB | Adobe PDF | Visualizza/Apri |
Pubblicazioni consigliate
I documenti in IRIS sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione.
https://hdl.handle.net/11583/2991584