arXiv:2605.25657cs.CV2026-05

基于对比学习的图神经网络,用少量标签实现精准疾病分类

ARMA-C3: A Contrastive ARMA Convolutional Framework for Unsupervised and Semi-supervised Classification

论文配图:ARMA-C3: A Contrastive ARMA Convolutional Framework for Unsupervised and Semi-supervised Classification
图 1 · 摘自论文原文
  • 用对比学习和图切割正则化构建统一框架
  • 在五个医学影像数据集上表现优于主流方法,尤其在小样本下
  • 适合医疗图像分析、标签稀缺场景的科研与临床应用

在生物医学与神经退行性疾病中,由于标注数据稀少和影像模式复杂,准确且早期的疾病识别仍具挑战。为此,我们提出ARMA-C3,一种基于对比学习与图切割正则化的统一无监督与半监督图学习框架,用于节点分类,以学习结构有意义且具有判别性的表征。通过将样本或图像建模为图节点,并利用样本间关系,该框架捕捉传统机器学习方法常忽略的个体级依赖。我们在五个临床相关数据集上开展广泛二分类实验:阿尔茨海默病神经影像计划(ADNI)、额颞叶痴呆神经影像数据集(NIFD)及三个医学影像基准(BreastMNIST、PneumoniaMNIST 和肝脏超声数据集)。结果表明,相较于经典聚类方法、先进机器学习模型及现有图深度学习方法,ARMA-C3在多种评估设置下均取得具有竞争力甚至更优的表现,尤其在监督信息有限和类别严重不平衡条件下。该框架还展现出稳健的表征学习能力与跨模态泛化性能,适用于多种生物医学影像模态。

原文摘要 · Abstract (English)

In biomedical and neurodegenerative disorders, accurate and early disease identification remains challenging due to the scarcity of labeled data and the complexity of imaging patterns. To address these challenges, we introduce ARMA-C3, a unified unsupervised and semi-supervised graph learning framework for node classification based on contrastive learning and graph-cut regularization to learn structurally meaningful and discriminative representations. By modeling samples or images as graph nodes and exploiting inter-sample relationships, the proposed framework captures subject-level dependencies that conventional machine learning methods typically overlook. We conduct extensive binary classification experiments across five clinically relevant datasets: the Alzheimer's Disease Neuroimaging Initiative (ADNI), the Neuroimaging in Frontotemporal Dementia (NIFD) dataset, and three medical imaging benchmarks (BreastMNIST, PneumoniaMNIST, and a liver ultrasound dataset). Experimental results demonstrate that ARMA-C3 achieves competitive and frequently superior performance compared to classical clustering techniques, state-of-the-art machine learning models, and existing graph-based deep learning approaches across multiple evaluation settings, particularly under limited supervision and severe class imbalance. The proposed framework further demonstrates robust representation learning and strong cross-modal generalization across diverse biomedical imaging modalities.

图神经网络医学影像半监督学习对比学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。