Fu, Xingyue, Deng, Ruoxi, Ahn, Euijoon, Korgaonkar, Mayuresh S., and Kim, Jinman (2026) CARformer: Class-Aware Representation Learning for Small-Cohort, Imbalanced Psychiatric Disorder Classification in Brain MRI. In: Lecture Notes in Computer Science (16885) pp. 98-107. From: Medical Image Computing and Computer Assisted Intervention – MICCAI 2026, 27 Sep - 1 Oct 2026, Strasbourg, France.

PDF (Published Version) - Published Version Restricted to Repository staff only

Abstract

Structural MRI (sMRI) has revealed morphological correlates associated with psychiatric disorders, motivating neuroimaging-based diagnostic support tools. However, multi-class classification from sMRI remains challenging in small-cohort psychiatric settings, where site-level datasets often contain only a few hundred subjects and exhibit class imbalance, and where subtle inter-class differences further complicate discrimination. We propose a class-aware representation learning framework (CARformer) that adapts domain-pretrained encoders for 3D sMRI multi-class psychiatric classification. The framework includes (1) a class-aware Bag-of-Queries (CA-BoQ) module that learns class-aligned query representations to aggregate class-specific dis-criminative evidence under inter-class differences, and (2) a Class-Aware Contrastive (CAC) learning strategy that strengthens class separation under imbalance. We evaluated the framework across two sites from a public brain MRI dataset to assess robustness under differences in diagnostic composition, class imbalance, and imaging acquisition protocols. Our results demonstrate consistent improvements in macro AUC and AUPRC over existing standard 3D Convolutional Neural Network (CNN) backbones and domain-pretrained fine-tuning frameworks across both sites, supporting the practical utility of class-aware adaptation of domain pre-training in small-cohort, imbalanced psychiatric imaging tasks. Source code is available at: https://github.com/xyfu33/CARformer.

Item ID:	94494
Item Type:	Conference Item (Research - E1)
ISSN:	0302-9743
Copyright Information:	© The Editor(s) (if applicable) and The Author(s), under exclusive license to Springer Nature Switzerland AG 2027.
Date Deposited:	26 Oct 2026 01:42
FoR Codes:	46 INFORMATION AND COMPUTING SCIENCES > 4603 Computer vision and multimedia computation > 460304 Computer vision @ 40% 40 ENGINEERING > 4003 Biomedical engineering > 400304 Biomedical imaging @ 20% 46 INFORMATION AND COMPUTING SCIENCES > 4611 Machine learning > 461103 Deep learning @ 40%
SEO Codes:	22 INFORMATION AND COMMUNICATION SERVICES > 2204 Information systems, technologies and services > 220403 Artificial intelligence @ 100%
	More Statistics