Mutual Evidential Deep Learning for Semi-supervised Medical Image Segmentation
Existing semi-supervised medical segmentation co-learning frameworks have realized that model performance can be diminished by the biases in model recognition caused by low-quality pseudo-labels. Due to the averaging nature of their pseudo-label integration strategy, they fail to explore the reliability of pseudo-labels from different sources. In this paper, we propose a mutual evidential deep learning (MEDL) framework that offers a potentially viable solution for pseudo-label generation in semi-supervised learning from two perspectives. First, we introduce networks with different architectures to generate complementary evidence for unlabeled samples and adopt an improved class-aware evidential fusion to guide the confident synthesis of evidential predictions sourced from diverse architectural networks. Second, utilizing the uncertainty in the fused evidence, we design an asymptotic Fisher information-based evidential learning strategy. This strategy enables the model to initially focus on unlabeled samples with more reliable pseudo-labels, gradually shifting attention to samples with lower-quality pseudo-labels while avoiding over-penalization of mislabeled classes in high data uncertainty samples. Additionally, for labeled data, we continue to adopt an uncertainty-driven asymptotic learning strategy, gradually guiding the model to focus on challenging voxels. Extensive experiments on five mainstream datasets have demonstrated that MEDL achieves state-of-the-art performance.
Authors
- Lijian Li (ORCID: https://orcid.org/0009-0006-2009-3269)
- Chi‐Man Pun (ORCID: https://orcid.org/0000-0003-1788-3746)
- Yali Bi (ORCID: https://orcid.org/0000-0001-5931-2592)
- Zhi Jin (ORCID: https://orcid.org/0000-0001-9670-7366)
- Yuanpeng He (ORCID: https://orcid.org/0000-0001-9071-2260)
- Wenpin Jiao (ORCID: https://orcid.org/0000-0001-9374-3900)
Institutions
- Southwest University (CN)
- Peking University (CN)
- University of Macau (MO)
Publication Details
- Published
- 2024-12-03
- DOI
- https://doi.org/10.1109/bibm62325.2024.10822008
- Citations
- 2
- Primary Topic
- AI in cancer detection
- Type
- article
- Field-Weighted Citation Impact
- 0.69
Funders
- National Natural Science Foundation of China