GraphCL: Graph-based Clustering for Semi-Supervised Medical Image Segmentation
Hebei University of Technology · Hong Kong University of Science and Technology · University of Electronic Science and Technology of China · National University of Defense Technology · Mohamed bin Zayed University of Artificial Intelligence · Sun Yat-Sen University · The Hong Kong Polytechnic University
PDF 由论文原始站点提供，PaperCompass 不保存论文文件。
摘要
Semi-supervised learning (SSL) has made notable advancements in medical image segmentation (MIS), particularly in scenarios with limited labeled data and significantly enhancing data utilization efficiency. Previous methods primarily focus on complex training strategies to utilize unlabeled data but neglect the importance of graph structural information. Different from existing methods, we propose a graph-based clustering for semi-supervised medical image segmentation (GraphCL) by jointly modeling graph data structure in a unified deep model. The proposed GraphCL model enjoys several advantages. Firstly, to the best of our knowledge, this is the first work to model the data structure information for semi-supervised medical image segmentation (SSMIS). Secondly, to get the clustered features across different graphs, we integrate both pairwise affinities between local image features and raw features as inputs. Extensive experimental results on three standard benchmarks show that the proposed GraphCL algorithm outperforms state-of-the-art semi-supervised medical image segmentation methods.