MDT: semi-supervised medical image segmentation with mixup-decoupling training

被引:0
|
作者
Long, Jianwu [1 ]
Ren, Yan [1 ]
Yang, Chengxin [1 ]
Ren, Pengcheng [1 ]
Zeng, Ziqin [1 ]
机构
[1] Chongqing Univ Technol, Coll Comp Sci & Engn, Chongqing 400054, Peoples R China
来源
PHYSICS IN MEDICINE AND BIOLOGY | 2024年 / 69卷 / 06期
基金
中国国家自然科学基金;
关键词
medical image segmentation; semi-supervised segmentation; consistency; pseudo-labeling;
D O I
10.1088/1361-6560/ad2715
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Objective. In the field of medicine, semi-supervised segmentation algorithms hold crucial research significance while also facing substantial challenges, primarily due to the extreme scarcity of expert-level annotated medical image data. However, many existing semi-supervised methods still process labeled and unlabeled data in inconsistent ways, which can lead to knowledge learned from labeled data being discarded to some extent. This not only lacks a variety of perturbations to explore potential robust information in unlabeled data but also ignores the confirmation bias and class imbalance issues in pseudo-labeling methods. Approach. To solve these problems, this paper proposes a semi-supervised medical image segmentation method 'mixup-decoupling training (MDT)' that combines the idea of consistency and pseudo-labeling. Firstly, MDT introduces a new perturbation strategy 'mixup-decoupling' to fully regularize training data. It not only mixes labeled and unlabeled data at the data level but also performs decoupling operations between the output predictions of mixed target data and labeled data at the feature level to obtain strong version predictions of unlabeled data. Then it establishes a dual learning paradigm based on consistency and pseudo-labeling. Secondly, MDT employs a novel categorical entropy filtering approach to pick high-confidence pseudo-labels for unlabeled data, facilitating more refined supervision. Main results. This paper compares MDT with other advanced semi-supervised methods on 2D and 3D datasets separately. A large number of experimental results show that MDT achieves competitive segmentation performance and outperforms other state-of-the-art semi-supervised segmentation methods. Significance. This paper proposes a semi-supervised medical image segmentation method MDT, which greatly reduces the demand for manually labeled data and eases the difficulty of data annotation to a great extent. In addition, MDT not only outperforms many advanced semi-supervised image segmentation methods in quantitative and qualitative experimental results, but also provides a new and developable idea for semi-supervised learning and computer-aided diagnosis technology research.
引用
收藏
页数:20
相关论文
共 50 条
  • [21] Meta semi-supervised medical image segmentation with label hierarchy
    Xu, Hai
    Xie, Hongtao
    Tan, Qingfeng
    Zhang, Yongdong
    HEALTH INFORMATION SCIENCE AND SYSTEMS, 2023, 11 (01)
  • [22] Mutual consistency learning for semi-supervised medical image segmentation
    Wu, Yicheng
    Ge, Zongyuan
    Zhang, Donghao
    Xu, Minfeng
    Zhang, Lei
    Xia, Yong
    Cai, Jianfei
    Medical Image Analysis, 2022, 81
  • [23] Data augmentation strategies for semi-supervised medical image segmentation
    Wang, Jiahui
    Ruan, Dongsheng
    Li, Yang
    Wang, Zefeng
    Wu, Yongquan
    Tan, Tao
    Yang, Guang
    Jiang, Mingfeng
    PATTERN RECOGNITION, 2025, 159
  • [24] Mutual consistency learning for semi-supervised medical image segmentation
    Wu, Yicheng
    Ge, Zongyuan
    Zhang, Donghao
    Xu, Minfeng
    Zhang, Lei
    Xia, Yong
    Cai, Jianfei
    MEDICAL IMAGE ANALYSIS, 2022, 81
  • [25] Dual Consistency Regularization for Semi-supervised Medical Image Segmentation
    Wei, Lin
    Sha, Runxuan
    Shi, Yucheng
    Wang, Qingxian
    Shi, Lei
    Gao, Yufei
    ADVANCED INTELLIGENT COMPUTING TECHNOLOGY AND APPLICATIONS, PT V, ICIC 2024, 2024, 14866 : 197 - 206
  • [26] PICK: Predict and Mask for Semi-supervised Medical Image Segmentation
    Zeng, Qingjie
    Lu, Zilin
    Xie, Yutong
    Xia, Yong
    INTERNATIONAL JOURNAL OF COMPUTER VISION, 2025,
  • [27] CCA: Contrastive cluster assignment for supervised and semi-supervised medical image segmentation
    Zhu, Jinghua
    Huang, Chengying
    Xi, Heran
    Cui, Hui
    NEURAL NETWORKS, 2025, 188
  • [28] SEMI-SUPERVISED HYPERSPECTRAL IMAGE SEGMENTATION
    Li, Jun
    Bioucas-Dias, Jose M.
    Plaza, Antonio
    2009 FIRST WORKSHOP ON HYPERSPECTRAL IMAGE AND SIGNAL PROCESSING: EVOLUTION IN REMOTE SENSING, 2009, : 215 - +
  • [29] Uncertainty-aware deep co-training for semi-supervised medical image segmentation
    Zheng, Xu
    Fu, Chong
    Xie, Haoyu
    Chen, Jialei
    Wang, Xingwei
    Sham, Chiu-Wing
    COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 149
  • [30] A Dual-Stage Semi-Supervised Pre-Training Approach for Medical Image Segmentation
    Aralikatti R.C.
    Pawan S.J.
    Rajan J.
    IEEE Transactions on Artificial Intelligence, 2024, 5 (02): : 556 - 565