Dynamic graph consistency and self-contrast learning for semi-supervised medical image segmentation

被引:2
作者
Li, Gang [1 ]
Xie, Jinjie [1 ]
Zhang, Ling [1 ]
Cheng, Guijuan [1 ]
Zhang, Kairu [1 ]
Bai, Mingqi [1 ]
机构
[1] Taiyuan Univ Technol, Coll Software, Taiyuan, Peoples R China
关键词
Semi-supervised learning; Dynamic graph consistency; Self-contrast learning; Medical image segmentation;
D O I
10.1016/j.neunet.2024.107063
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Semi-supervised medical image segmentation endeavors to exploit a limited set of labeled data in conjunction with a substantial corpus of unlabeled data, with the aim of training models that can match or even exceed the efficacy of fully supervised segmentation models. Despite the potential of this approach, most existing semisupervised medical image segmentation techniques that employ consistency regularization predominantly focus on spatial consistency at the image level, often neglecting the crucial role of feature-level channel information. To address this limitation, we propose an innovative method that integrates graph convolutional networks with a consistency regularization framework to develop a dynamic graph consistency approach. This method imposes channel-level constraints across different decoders by leveraging high-level features within the network. Furthermore, we introduce a novel self-contrast learning strategy, which performs image-level comparison within the same batch and engages in pixel-level contrast learning based on pixel positions. This approach effectively overcomes traditional contrast learning challenges related to identifying positive and negative samples, reduces computational resource consumption, and significantly improves model performance. Our experimental evaluation on three distinct medical image segmentation datasets indicates that the proposed method demonstrates superior performance across a variety of test scenarios.
引用
收藏
页数:14
相关论文
共 54 条
[1]   Pseudo-label Guided Contrastive Learning for Semi-supervised Medical Image Segmentation [J].
Basak, Hritam ;
Yin, Zhaozheng .
2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, :19786-19797
[2]   Deep Learning Techniques for Automatic MRI Cardiac Multi-Structures Segmentation and Diagnosis: Is the Problem Solved? [J].
Bernard, Olivier ;
Lalande, Alain ;
Zotti, Clement ;
Cervenansky, Frederick ;
Yang, Xin ;
Heng, Pheng-Ann ;
Cetin, Irem ;
Lekadir, Karim ;
Camara, Oscar ;
Gonzalez Ballester, Miguel Angel ;
Sanroma, Gerard ;
Napel, Sandy ;
Petersen, Steffen ;
Tziritas, Georgios ;
Grinias, Elias ;
Khened, Mahendra ;
Kollerathu, Varghese Alex ;
Krishnamurthi, Ganapathy ;
Rohe, Marc-Michel ;
Pennec, Xavier ;
Sermesant, Maxime ;
Isensee, Fabian ;
Jaeger, Paul ;
Maier-Hein, Klaus H. ;
Full, Peter M. ;
Wolf, Ivo ;
Engelhardt, Sandy ;
Baumgartner, Christian F. ;
Koch, Lisa M. ;
Wolterink, Jelmer M. ;
Isgum, Ivana ;
Jang, Yeonggul ;
Hong, Yoonmi ;
Patravali, Jay ;
Jain, Shubham ;
Humbert, Olivier ;
Jodoin, Pierre-Marc .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2018, 37 (11) :2514-2525
[3]  
Chen T, 2020, PR MACH LEARN RES, V119
[4]   Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision [J].
Chen, Xiaokang ;
Yuan, Yuhui ;
Zeng, Gang ;
Wang, Jingdong .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :2613-2622
[5]   Combining contrastive learning and shape awareness for semi-supervised medical image segmentation [J].
Chen, Yaqi ;
Chen, Faquan ;
Huang, Chenxi .
EXPERT SYSTEMS WITH APPLICATIONS, 2024, 242
[6]   Improving Anatomical Plausibility in Medical Image Segmentation via Hybrid Graph Neural Networks: Applications to Chest X-Ray Analysis [J].
Gaggion, Nicolas ;
Mansilla, Lucas ;
Mosquera, Candelaria ;
Milone, Diego H. ;
Ferrante, Enzo .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2023, 42 (02) :546-556
[7]  
Gai D., 2024, Computers in Biology and Medicine, V174
[8]   ConvUNeXt: An efficient convolution neural network for medical image segmentation [J].
Han, Zhimeng ;
Jian, Muwei ;
Wang, Gai-Ge .
KNOWLEDGE-BASED SYSTEMS, 2022, 253
[9]   Bilateral Supervision Network for Semi-Supervised Medical Image Segmentation [J].
He, Along ;
Li, Tao ;
Yan, Juncheng ;
Wang, Kai ;
Fu, Huazhu .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2024, 43 (05) :1715-1726
[10]   Momentum Contrast for Unsupervised Visual Representation Learning [J].
He, Kaiming ;
Fan, Haoqi ;
Wu, Yuxin ;
Xie, Saining ;
Girshick, Ross .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2020), 2020, :9726-9735