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 条
  • [41] MCF: Mutual Correction Framework for Semi-Supervised Medical Image Segmentation
    Wang, Yongchao
    Xiao, Bin
    Bi, Xiuli
    Li, Weisheng
    Gao, Xinbo
    2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, : 15651 - 15660
  • [42] Semi-Supervised Medical Image Segmentation With Voxel Stability and Reliability Constraints
    Zhao, Yang
    Lu, Ke
    Xue, Jian
    Wang, Shuhua
    Lu, Jian
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2023, 27 (08) : 3912 - 3923
  • [43] Inherent Consistent Learning for Accurate Semi-supervised Medical Image Segmentation
    Zhu, Ye
    Yang, Jie
    Liu, Si-Qi
    Zhang, Ruimao
    MEDICAL IMAGING WITH DEEP LEARNING, VOL 227, 2023, 227 : 1581 - 1601
  • [44] Bidirectional Copy-Paste for Semi-Supervised Medical Image Segmentation
    Bai, Yunhao
    Chen, Duowen
    Li, Qingli
    Shen, Wei
    Wang, Yan
    2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, : 11514 - 11524
  • [45] Semi-Supervised Learning With Fact-Forcing for Medical Image Segmentation
    Bui, Phuoc-Nguyen
    Le, Duc-Tai
    Bum, Junghyun
    Kim, Seongho
    Song, Su Jeong
    Choo, Hyunseung
    IEEE ACCESS, 2023, 11 : 99413 - 99425
  • [46] Boundary-Aware Prototype in Semi-Supervised Medical Image Segmentation
    Wang, Yongchao
    Xiao, Bin
    Bi, Xiuli
    Li, Weisheng
    Gao, Xinbo
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2024, 33 : 5456 - 5467
  • [47] Exploring Feature Representation Learning for Semi-Supervised Medical Image Segmentation
    Wu, Huimin
    Li, Xiaomeng
    Cheng, Kwang-Ting
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (11) : 16589 - 16601
  • [48] Semi-supervised medical image segmentation network based on mutual learning
    Sun, Junmei
    Wang, Tianyang
    Wang, Meixi
    Li, Xiumei
    Xu, Yingying
    MEDICAL PHYSICS, 2025, 52 (03) : 1589 - 1600
  • [49] Boundary-Aware Prototype in Semi-Supervised Medical Image Segmentation
    Wang, Yongchao
    Xiao, Bin
    Bi, Xiuli
    Li, Weisheng
    Gao, Xinbo
    IEEE Transactions on Image Processing, 2024, 33 : 5456 - 5467
  • [50] Multidimensional perturbed consistency learning for semi-supervised medical image segmentation
    Yuan, Enze
    Zhao, Bin
    Qin, Xiao
    Ding, Shuxue
    INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2024, 34 (03)