Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

About

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.

VSS Tejaswi Abburi, Saurabh J. Shigwan, Nitin Kumar• 2026

Related benchmarks

TaskDatasetResultRank
ClassificationLiver ultrasound dataset (10% train, 90% test)
Accuracy78.5
40
Semi-supervised classificationADNI CN vs. AD (90% test)
Accuracy77.9
40
Unsupervised ClusteringNIFD (unsupervised)
Accuracy71.2
25
Unsupervised ClusteringBreastMNIST
Accuracy73.1
25
Unsupervised ClusteringPneumoniaMNIST
Accuracy91
25
Unsupervised ClusteringADNI CN vs AD
Accuracy74.8
25
Clusteringliver ultrasound dataset
Accuracy75.4
24
Semi-supervised classificationNIFD (10% train, 90% test)
Accuracy72.3
20
Semi-supervised classificationMedMNIST Breast (10% train 90% test)
Accuracy85.5
20
Semi-supervised classificationADNI CN vs. MCI (10% train, 90% test)
Accuracy79.7
20
Showing 10 of 10 rows

Other info

Follow for update