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 条
  • [1] Semi-supervised medical image segmentation with dual-branch mixup-decoupling confidence training
    Long, Jianwu
    Liu, Yuanqin
    Ren, Yan
    MEDICAL ENGINEERING & PHYSICS, 2025, 136
  • [2] Dual-decoder data decoupling training for semi-supervised medical image segmentation
    Wang, Bing
    Huang, Taifeng
    Yang, Shuo
    Yang, Ying
    Zhai, Junhai
    Zhang, Xin
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2025, 100
  • [3] Semi-supervised structure attentive temporal mixup coherence for medical image segmentation
    Pawan, S. J.
    Jeevan, Govind
    Rajan, Jeny
    BIOCYBERNETICS AND BIOMEDICAL ENGINEERING, 2022, 42 (04) : 1149 - 1161
  • [4] Balanced feature fusion collaborative training for semi-supervised medical image segmentation
    Zhao, Zhongda
    Wang, Haiyan
    Lei, Tao
    Wang, Xuan
    Shen, Xiaohong
    Yao, Haiyang
    PATTERN RECOGNITION, 2025, 157
  • [5] Decoupled Consistency for Semi-supervised Medical Image Segmentation
    Chen, Faquan
    Fei, Jingjing
    Chen, Yaqi
    Huang, Chenxi
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2023, PT I, 2023, 14220 : 551 - 561
  • [6] Semi-supervised Medical Image Segmentation with Confidence Calibration
    Xu, Qisen
    Wu, Qian
    Hu, Yiqiu
    Jin, Bo
    Hu, Bin
    Zhu, Fengping
    Li, Yuxin
    Wang, Xiangfeng
    2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2021,
  • [7] Semi-supervised Image Segmentation
    Lazarova, Gergana Angelova
    ARTIFICIAL INTELLIGENCE: METHODOLOGY, SYSTEMS, AND APPLICATIONS, 2014, 8722 : 59 - 68
  • [8] Deep Tri-Training for Semi-Supervised Image Segmentation
    An, Shan
    Zhu, Haogang
    Zhang, Jiaao
    Ye, Junjie
    Wang, Siliang
    Yin, Jianqin
    Zhang, Hong
    IEEE ROBOTICS AND AUTOMATION LETTERS, 2022, 7 (04) : 10097 - 10104
  • [9] Deep co-training for semi-supervised image segmentation
    Peng, Jizong
    Estrada, Guillermo
    Pedersoli, Marco
    Desrosiers, Christian
    PATTERN RECOGNITION, 2020, 107 (107)
  • [10] Adaptive Bidirectional Displacement for Semi-Supervised Medical Image Segmentation
    Chi, Hanyang
    Pang, Jian
    Zhang, Bingfeng
    Liu, Weifeng
    2024 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2024, 2024, : 4070 - 4080