Semi-Supervised Unpaired Medical Image Segmentation Through Task-Affinity Consistency. 2023

Jingkun Chen, and Jianguo Zhang, and Kurt Debattista, and Jungong Han

Deep learning-based semi-supervised learning (SSL) algorithms are promising in reducing the cost of manual annotation of clinicians by using unlabelled data, when developing medical image segmentation tools. However, to date, most existing semi-supervised learning (SSL) algorithms treat the labelled images and unlabelled images separately and ignore the explicit connection between them; this disregards essential shared information and thus hinders further performance improvements. To mine the shared information between the labelled and unlabelled images, we introduce a class-specific representation extraction approach, in which a task-affinity module is specifically designed for representation extraction. We further cast the representation into two different views of feature maps; one is focusing on low-level context, while the other concentrates on structural information. The two views of feature maps are incorporated into the task-affinity module, which then extracts the class-specific representations to aid the knowledge transfer from the labelled images to the unlabelled images. In particular, a task-affinity consistency loss between the labelled images and unlabelled images based on the multi-scale class-specific representations is formulated, leading to a significant performance improvement. Experimental results on three datasets show that our method consistently outperforms existing state-of-the-art methods. Our findings highlight the potential of consistency between class-specific knowledge for semi-supervised medical image segmentation. The code and models are to be made publicly available at https://github.com/jingkunchen/TAC.

UI MeSH Term Description Entries
D000069553 Supervised Machine Learning A MACHINE LEARNING paradigm used to make predictions about future instances based on a given set of labeled paired input-output training (sample) data. Active Machine Learning,Inductive Machine Learning,Learning from Labeled Data,Machine Learning with a Teacher,Semi-supervised Learning,Learning, Active Machine,Learning, Inductive Machine,Learning, Semi-supervised,Learning, Supervised Machine,Machine Learning, Active,Machine Learning, Inductive,Machine Learning, Supervised,Semi supervised Learning
D000465 Algorithms A procedure consisting of a sequence of algebraic formulas and/or logical steps to calculate or determine a given task. Algorithm

Related Publications

Jingkun Chen, and Jianguo Zhang, and Kurt Debattista, and Jungong Han
October 2022, Medical image analysis,
Jingkun Chen, and Jianguo Zhang, and Kurt Debattista, and Jungong Han
July 2023, Computers in biology and medicine,
Jingkun Chen, and Jianguo Zhang, and Kurt Debattista, and Jungong Han
April 2023, Artificial intelligence in medicine,
Jingkun Chen, and Jianguo Zhang, and Kurt Debattista, and Jungong Han
January 2025, IEEE transactions on medical imaging,
Jingkun Chen, and Jianguo Zhang, and Kurt Debattista, and Jungong Han
May 2025, Scientific reports,
Jingkun Chen, and Jianguo Zhang, and Kurt Debattista, and Jungong Han
February 2021, Medical image analysis,
Jingkun Chen, and Jianguo Zhang, and Kurt Debattista, and Jungong Han
August 2023, Medical image analysis,
Jingkun Chen, and Jianguo Zhang, and Kurt Debattista, and Jungong Han
May 2023, IEEE transactions on medical imaging,
Jingkun Chen, and Jianguo Zhang, and Kurt Debattista, and Jungong Han
October 2025, Journal of imaging informatics in medicine,
Jingkun Chen, and Jianguo Zhang, and Kurt Debattista, and Jungong Han
August 2025, IEEE transactions on medical imaging,
Copied contents to your clipboard!