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 条
  • [31] Decoupled Training for Semi-supervised Medical Image Segmentation with Worst-Case-Aware Learning
    Das, Ankit
    Gautam, Chandan
    Cholakkal, Hisham
    Agrawal, Pritee
    Yang, Feng
    Savitha, Ramasamy
    Liu, Yong
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT XII, 2024, 15012 : 45 - 55
  • [32] Reliable semi-supervised mutual learning framework for medical image segmentation
    Hang, Wenlong
    Bai, Kui
    Liang, Shuang
    Zhang, Qingfeng
    Wu, Qiang
    Jin, Yukun
    Wang, Qiong
    Qin, Jing
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2025, 99
  • [33] Analysing the effectiveness of a generative model for semi-supervised medical image segmentation
    Rosnati, Margherita
    Ribeiro, Fabio De Sousa
    Monteiro, Miguel
    de Castro, Daniel Coelho
    Glocker, Ben
    MACHINE LEARNING FOR HEALTH, VOL 193, 2022, 193 : 290 - 310
  • [34] Contour-aware consistency for semi-supervised medical image segmentation
    Li, Lei
    Lian, Sheng
    Luo, Zhiming
    Wang, Beizhan
    Li, Shaozi
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2024, 89
  • [35] Dual-branch Transformer for semi-supervised medical image segmentation
    Huang, Xiaojie
    Zhu, Yating
    Shao, Minghan
    Xia, Ming
    Shen, Xiaoting
    Wang, Pingli
    Wang, Xiaoyan
    JOURNAL OF APPLIED CLINICAL MEDICAL PHYSICS, 2024, 25 (10):
  • [36] Towards Generic Semi-Supervised Framework for Volumetric Medical Image Segmentation
    Wang, Haonan
    Li, Xiaomeng
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 36 (NEURIPS 2023), 2023,
  • [37] Federated Semi-supervised Medical Image Segmentation Based on Asynchronous Transmission
    Liu, Fangbo
    Yang, Feng
    ADVANCED INTELLIGENT COMPUTING TECHNOLOGY AND APPLICATIONS, ICIC 2023, PT III, 2023, 14088 : 55 - 66
  • [38] Overlay Mantle-Free for Semi-supervised Medical Image Segmentation
    Liu, Jiacheng
    Qian, Wenhua
    Cao, Jinde
    Liu, Peng
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT X, 2024, 15010 : 589 - 598
  • [39] Cross co-teaching for semi-supervised medical image segmentation
    Zhang, Fan
    Liu, Huiying
    Wang, Jinjiang
    Lyu, Jun
    Cai, Qing
    Li, Huafeng
    Dong, Junyu
    Zhang, David
    PATTERN RECOGNITION, 2024, 152
  • [40] FRCNet: Frequency and Region Consistency for Semi-supervised Medical Image Segmentation
    He, Along
    Li, Tao
    Wu, Yanlin
    Zou, Ke
    Fu, Huazhu
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT VIII, 2024, 15008 : 305 - 315