ARC-CT: Anatomy-Routed Contrastive Vision-Language Learning for 3D Chest CT
Authors: Huseyin Umut Isik, Mehmet Alp Ozaydin, Sila Kurugol, Şeyda Ertekin
Organizations: Department of Computer Engineering, METU, Ankara, Turkey · Quantitative Intelligent Imaging Lab, Boston Children’s Hospital and Harvard Medical School, Boston, MA, USA · METU-DTX Digital Transformation and Innovation Center, Ankara, Turkey
Contrastive vision-language learning uses paired chest CT volumes and radiology reports to learn abnormality classifiers without manually annotated labels. However, two characteristics of chest CT challenge conventional global contrastive learning. First, many critical abnormalities are small or anatomically localized, and pooling an en- tire volume into a single embedding may dilute their visual evidence. Second, the standard contrastive objective treats every other scan in a batch as a negative. Because many chest CTs share abnormalities, this objective incorrectly pushes co-positive pairs apart. We propose Anatomy-Routed Contrastive Learning for 3D Chest CT (ARC-CT), a region-aware framework that addresses these limitations using only la- bels extracted from reports by an LLM, with no manual annotations or bounding boxes. ARC-CT combines three components: (1) an Anato- myQFormer localizing evidence via queries constrained by automatically generated organ masks; (2) a label-Jaccard soft InfoNCE objective in- tegrating the standard one-hot target with the label-set overlap of each pair, which reduces false-negative penalties between studies that share clinical findings; and (3) an organ-level alignment loss connecting mask- pooled visual features to organ-specific report text extracted offline with a large language model. ARC-CT achieves a 0.86 mask-free macro AUC across 18 abnormalities using a compact 3D ResNet-18 backbone. Over- all, ARC-CT outperforms both comparable efficient baselines and sev- eral larger transformer models. Our code and weights are available at https://github.com/arc-ct/arc-ct.
Figures & tables
Figure 1: ARC-CT overview. Stage 1 pretrains a 3D ResNet-18 on 18 LLM-extracted abnormality labels, warm-starting Stage 2. In Stage 2, the encoder’s feature map is cross-attended by the AnatomyQFormer’s anatomy, pathology, and global queries, while an LLM parses reports into per-organ sentences. Embeddings are projected into a shared space and trained with a label-Jaccard soft InfoNCE contrastive loss with per-organ alignment and query supervision.
CT-RATE (in-domain)
RAD-ChestCT (external)
Method ∗
AUC
Prec
Acc
F1
AUC
Prec
Acc
F1
ARC-CT (ours)
0.855
0.455
0.787
0.809
0.734
0.392
0.679
0.716
GreenRFM [ 4 ]
0.848
0.444
0.777
0.800
0.780
0.461
0.730
0.757
MPS-CT [ 3 ]
0.838
0.435
0.774
0.796
0.773
0.450
0.719
0.748
BrgSA [ 5 ]
0.792
0.385
0.733
0.762
0.742
0.422
0.686
0.720
ViSD-Boost [ 7 ]
0.790
0.387
0.731
0.759
0.694
0.342
0.652
0.693
Table 1: Prompt-based abnormality classification (no manual labels) on CT-RATE (in-domain) and RAD-ChestCT (external). The best per column in bold , second underlined .
Class
ARC-CT
GreenRFM
MPS-CT
CT-CLIP
Pleural effusion
0.97
0.97
0.96
0.90
Cardiomegaly
0.93
0.93
0.94
0.86
Arterial wall calcification
0.94
0.94
0.93
0.85
Coronary artery wall calcification
0.93
0.94
0.93
0.85
Consolidation
0.91
0.91
0.89
0.72
Medical material
0.94
0.91
0.89
0.72
Table 2: Per-class AUC on CT-RATE validation (3,039 volumes), three-seed mean. GreenRFM [ 4 ] , MPS-CT [ 3 ] , and CT-CLIP [ 1 ] are the baselines with a published 18-class breakdown. Best per row in bold , second underlined .
Figure 2: Mask-free Grad-CAM for four positive findings (cardiomegaly, arterial wall calcification, pleural effusion, consolidation) on one CT-RATE validation volume. Top row (a–d): ARC-CT; bottom row (e–h): CT-CLIP [ 1 ] . ARC-CT’s activation is sharply localized to the relevant structure, while CT-CLIP’s is diffuse and largely unchanged between findings.
Configuration
Macro AUC
Δ
95% CI
ARC-CT
0.855±0.001
–
[0.848,0.862]
− Stage-1 warm-start (Kinetics only)
0.826±0.001
−0.028
[−0.032,−0.024]
− anatomy routing (plain Q-Former)
0.837±0.004
−0.017
[−0.019,−0.014]
− label-Jaccard soft target
0.845±0.001
−0.009
[−0.011,−0.007]
− learnable temperature (fixed)
0.847±0.002
−0.007
[−0.009,−0.006]
− per-organ alignment
0.850±0.002
−0.004
[−0.006,−0.002]
Table 3: Leave-one-out ablation on CT-RATE (three-seed mean ± s.d.; 95% CIs from a paired patient bootstrap; all p<0.001 ). The weakly supervised Stage-1 warm-start contributes the largest gain, followed by anatomy routing and the label-Jaccard target.
Image → image (MAP@ K )
Report → image (R@ K )
Method
@5
@10
@50
@5
@10
@50
@100
ARC-CT (ours)
71.3
61.7
54.3
11.4
18.0
40.4
52.1
GreenRFM [ 4 ]
–
–
–
9.5
16.0
39.9
54.3
MPS-CT [ 3 ]
71.3
61.5
53.7
10.0
16.3
39.0
52.2
BrgSA [ 5 ]
69.2
58.5
50.5
5.8
10.1
28.6
42.0
CT-CLIP [ 1 ]
68.3
57.2
48.9
2.9
5.0
18.0
28.7
Table 4: Cross-modal image-to-image and report-to-image retrieval on CT-RATE validation, no fine-tuning. ARC-CT leads on all metrics but R@100. The best per column in bold , second underlined .
Recent 3D CT vision-language models align volumes with reports via contrastive pretraining, but typically rely on limited public data and provide only coarse global supervision. We train a 3D CT vision-language model on 98k report-volume pairs (50k patients) collected at a single hospital, combined with public datasets, using SigLIP-style contrastive pretraining together with prompt-based disease supervision in the shared vision-text embedding space. On CT-RATE, our model achieves state-of-the-art text-to-image retrieval (R@10 31.5 vs. 22.2) and competitive disease classification (AUC 83.8 vs. 83.8), with consistent results on Rad-ChestCT (AUC 77.0 vs. 77.3). We further observe that radiologists routinely reference specific images within their reports (e.g., ``series X, image Y''), linking textual descriptions to precise axial locations. We automatically mine 262k such snippet-slice pairs and introduce the task of intra-scan snippet localization -- predicting the axial depth referred to by a text snippet -- reducing mean absolute error to 36.3 mm at 12 mm feature resolution, compared with 67.0 mm for the best baseline. Adding this localization objective leaves retrieval and classification broadly unchanged within confidence bounds, yielding a single unified model for retrieval, classification, and intra-scan grounding.
Simon Ging, Philipp Arnold, Sebastian Walter +6
Computer Vision Group, University of Freiburg, Germany · Adaptive & Agentic AI · Adaptive & Agentic AI (A3) Lab, Aarhus University, Denmark +5
Computed tomography (CT) vision-language pretraining from paired volumes and radiology reports is a scalable yet challenging task. Existing methods commonly adopt global scan-report contrast, which is scalable but obscures heterogeneous organ evidence. Meanwhile, direct organ-level alignment remains coarse, since the same anatomy can exhibit multiple distinct radiological appearances. Therefore, pretraining requires a finer alignment unit: the organ-conditioned radiological pattern. In this work, we propose OCP-CT, an organ-conditioned pattern-token alignment framework for CT vision-language pretraining. Specifically, OCP-CT preserves a stable global CT-report contrastive branch and introduces an organ pattern interface: sparse Mixture-of-Experts (MoE) routes image and text tokens according to latent radiological patterns, learnable slots query the routed tokens into continuous pattern tokens, and paired token contrast aligns image-text pattern tokens with structured soft targets built from report-derived clinical similarity. On the publicly available CT-RATE and RAD-ChestCT benchmarks, OCP-CT achieves average AUROCs of 84.5% and 69.9% for zero-shot abnormality diagnosis, respectively. Compared with the strongest prior reported results, these results yield absolute AUROC gains of 6.7 and 0.8 percentage points.
Vision-language models (VLMs) pretrained on large-scale image-text pairs demonstrate strong image-level understanding, but are primarily optimized for global alignment and do not explicitly encode fine-grained anatomical structure, limiting their suitability for spatially precise tasks such as segmentation. We introduce CheXanatomy, a framework that integrates explicit anatomical knowledge into a pretrained VLM through autoregressive token-space supervision. Instead of adding task-specific decoder heads, the model is trained to generate anatomical segmentation masks via next-token prediction. To enable scalable supervision, we synthesize realistic chest radiographs from CT volumes and forward-project CT segmentation labels to obtain anatomically consistent 2D masks. We evaluate the approach on synthetic and real chest radiographs against a U-Net baseline, including ablations on model scale, input resolution, and vision encoder fine-tuning. Autoregressive anatomical supervision achieves performance comparable to specialized convolutional models in-distribution and demonstrates improved geometric robustness under domain shift to real CXR data. In addition, anatomy-pretrained models exhibit improved sample efficiency when adapting to novel localization tasks under limited supervision. Larger models and higher input image resolution improve performance, while vision encoder fine-tuning has limited effect. These results show that embedding anatomical structure directly into the generative objective promotes spatially grounded representations and supports anatomy-aware medical vision-language modeling.
Sergios Gatidis, Curtis Langlotz, Christian Bluethgen
Stanford Center for Artificial Intelligence in Medicine and Imaging, Stanford University, Palo Alto, CA, USA · Department of Radiology, Stanford University, Stanford, CA, USA