Pseudo Labeling based Consistency Learning for Semi-supervised Medical Image Segmentation

被引:0
作者
Cheng, Yintao [1 ]
Zhang, Manli [1 ]
Yang, Rui [1 ]
Kang, Guixia [1 ]
机构
[1] Beijing Univ Posts & Telecommun, Beijing, Peoples R China
来源
2024 11TH INTERNATIONAL CONFERENCE ON BIOMEDICAL AND BIOINFORMATICS ENGINEERING, ICBBE 2024 | 2024年
基金
中国国家自然科学基金;
关键词
Semi-supervised Learning; Medical Image Segmentation; Entropy Minimization; Consistency Regularization;
D O I
10.1145/3707127.3707135
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Semi-supervised medical image segmentation has garnered much attention in recent years due to its potential to significantly alleviate the dependence on expert annotations. In this paper, we present a novel Pseudo Labeling based Consistency Learning (PLCL) method, designed to effectively exploit unlabeled data by encouraging low entropy outputs while maintaining the consistency of results in ambiguous regions like border area. In practice, the framework consists of a shared encoder and two different decoders, and we generate soft pseudo labels from the outputs. On this foundation, we adopt a rectified intra entropy loss for the dual decoder branch and design a novel uncertainty-aware scheme to derive the mutual consistency loss in the fuzzy region. The proposed method is evaluated on the left atrium (LA) and the BraTS datasets, with experimental results demonstrating superior performance compared to state-of-the-art semi-supervised methods. Code is available at: https://github.com/krid-up/PLCL
引用
收藏
页码:46 / 52
页数:7
相关论文
共 23 条
[1]  
Berthelot D, 2019, Arxiv, DOI arXiv:1905.02249
[2]   MTANS: Multi-Scale Mean Teacher Combined Adversarial Network with Shape-Aware Embedding for Semi-Supervised Brain Lesion Segmentation [J].
Chen, Gaoxiang ;
Ru, Jintao ;
Zhou, Yilin ;
Rekik, Islem ;
Pan, Zhifang ;
Liu, Xiaoming ;
Lin, Yezhi ;
Lu, Beichen ;
Shi, Jialin .
NEUROIMAGE, 2021, 244
[3]  
Chen XK, 2021, Arxiv, DOI arXiv:2106.01226
[4]   Recent advances and clinical applications of deep learning in medical image analysis [J].
Chen, Xuxin ;
Wang, Ximin ;
Zhang, Ke ;
Fung, Kar-Ming ;
Thai, Theresa C. ;
Moore, Kathleen ;
Mannel, Robert S. ;
Liu, Hong ;
Zheng, Bin ;
Qiu, Yuchen .
MEDICAL IMAGE ANALYSIS, 2022, 79
[5]   MultiResUNet : Rethinking the U-Net architecture for multimodal biomedical image segmentation [J].
Ibtehaz, Nabil ;
Rahman, M. Sohel .
NEURAL NETWORKS, 2020, 121 :74-87
[6]   nnU-Net: a self-configuring method for deep learning-based biomedical image segmentation [J].
Isensee, Fabian ;
Jaeger, Paul F. ;
Kohl, Simon A. A. ;
Petersen, Jens ;
Maier-Hein, Klaus H. .
NATURE METHODS, 2021, 18 (02) :203-+
[7]  
Ke ZH, 2019, Arxiv, DOI arXiv:1909.01804
[8]  
khurram Ejaz, 2020, Brats MICCAI Brain Tumor Dataset
[9]  
Laine S, 2017, Arxiv, DOI [arXiv:1610.02242, DOI 10.48550/ARXIV.1610.02242]
[10]   A Review of Deep-Learning-Based Medical Image Segmentation Methods [J].
Liu, Xiangbin ;
Song, Liping ;
Liu, Shuai ;
Zhang, Yudong .
SUSTAINABILITY, 2021, 13 (03) :1-29