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 in questo prodotto:
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.

Utilizza questo identificativo per citare o creare un link a questo documento: https://hdl.handle.net/11583/2991584