Beyond Classification: Structured Supervision Aligns Visual Evidence with Medical Semantics
Authors: Hexiang Bai, Hanyang Xu, Xiaoxue Li, Xiaoliang Wu, Shangde Gao, Hongxia Xu, Ke Liu
Organizations: Zhejiang University Hangzhou, China · Henan Normal University Xinxiang, China · University of Southampton Southampton, United Kingdom · The Second Affiliated Hospital Zhejiang University Hangzhou, China · Transvascular Implantation Devices Research Institute and State Key Laboratory of Transvascular Implantation Devices Hangzhou, China · School of Software Technology Zhejiang University Ningbo, China
Vision Transformers (ViTs) have shown immense potential in medical image analysis. However, standard pre-training via global image classification suffers from spatial collapse, where models rely heavily on background shortcuts rather than localising critical foreground lesions. To overcome this limitation and align visual evidence with precise medical semantics, we systematically investigate alternative pre-training paradigms.Specifically, we evaluate three independent forms of structured supervision: topological priors via graph self-supervision, dense pixel-level constraints via segmentation, and cross-modal semantic grounding via image-text pairs. Notably, our empirical analysis reveals that while all three forms of structured supervision successfully alleviate the global pooling bottleneck and steer visual attention towards foreground regions, image-text alignment achieves the most superior performance. By embedding high-dimensional diagnostic logic, the cross-modal approach not only anchors attention on precise visual evidence but also enables profound abstract reasoning. Extensive experiments demonstrate that this semantically enriched pre-training fundamentally enhances the model's feature representation. Consequently, when fine-tuned for downstream clinical classification tasks, our models achieve superior accuracy and yield highly interpretable attention maps focused on true pathological features, vastly outperforming vanilla classification baselines.
Figures & tables
Figure 1: Overview of the three pre-training paradigms. Graph self-supervision introduces topological priors, segmentation-guided co-pretraining provides dense foreground constraints, and image-text alignment grounds visual representations in medical semantics.
Figure 2: Patch-score distribution of a standard ViT classifier. High-relevance patches can concentrate outside the lesion foreground, indicating weak spatial grounding of the class-token representation.
Figure 3: Qualitative comparison of ViT and ResNet-50 evidence localization. From left to right: original image, ground-truth lesion mask with bounding box, patch-score heatmap, and the top-scoring region.
Dataset
Modality
Task
# Imgs.
# Cls.
Mask
BUSI
Ultrasound
Breast lesion cls.
780
3
Lesion
HAM10000
Dermoscopy
Skin lesion cls.
10,015
7
Lesion
SIIM-ACR
X-ray
Pneumothorax cls.
12,047
2
Pneumothorax
Table 1: Summary of the datasets used in experiments.
Dataset
ViT-B/16
ResNet50-L4
Δ PIB
ACC
PIB
ACC
PIB
BUSI
0.888
0.071
0.862
0.690
+0.619
HAM10000
0.830
0.393
0.867
0.954
+0.561
SIIM-ACR
0.789
0.040
0.884
0.410
+0.370
Average
0.836
0.168
0.871
0.685
+0.517
Table 2: Evidence localization comparison between a standard ViT and a spatially structured convolutional reference. Δ PIB is computed as the PIB difference between ResNet50-L4 and ViT-B/16.
Model
Method
BUSI
HAM10000
Pneumothorax
Avg. PIB
ACC
PIB
ACC
PIB
ACC
PIB
ViT-B/16
Standard
0.888
0.071
0.830
0.393
0.847
0.040
0.168
BiomedCLIP
Zero-shot
0.362
0.346
0.228
0.561
0.798
0.088
0.332
BiomedCLIP
Linear probe
0.759
0.346
0.747
0.561
0.811
0.088
0.332
BiomedCLIP
Fine-tuning
0.888
0.385
0.843
0.768
0.829
0.076
0.410
LVM-Med
Linear probe
0.750
0.135
0.790
0.231
0.835
0.063
0.143
Table 3: Structured supervision results across three medical classification datasets. Avg. PIB is computed over datasets with available PIB values.
Figure 4: PCA visualization of encoder class-token features on BUSI under three input settings: original image, background-only image, and foreground-only image.
Department of Computer Science Wayne State University · Department of Radiation Oncology Henry Ford Health · Institute for AI and Data Science Wayne State University