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.
Vision-language pretraining has driven significant progress in medical image analysis. However, current methods typically supervise visual encoders using one-hot labels or free-form text, neither of which effectively captures the complex semantic relationships among clinical findings. In this study, we introduce VIVID-Med, a novel framework that leverages a frozen large language model (LLM) as a structured semantic teacher to pretrain medical vision transformers (ViTs). VIVID-Med translates clinical findings into verifiable JSON field-state pairs via a Unified Medical Schema (UMS), utilizing answerability-aware masking to focus optimization. It then employs Structured Prediction Decomposition (SPD) to partition cross-attention into orthogonality-regularized query groups, extracting complementary visual aspects. Crucially, the LLM is discarded post-training, yielding a lightweight, deployable ViT-only backbone. We evaluated VIVID-Med across multiple settings: on CheXpert linear probing, it achieves a macro-AUC of 0.8588, outperforming BiomedCLIP by +6.65 points while using 500x less data. It also demonstrates robust zero-shot cross-domain transfer to NIH ChestX-ray14 (0.7225 macro-AUC) and strong cross-modality generalization to CT, achieving 0.8413 AUC on LIDC-IDRI lung nodule classification and 0.9969 macro-AUC on OrganAMNIST 11-organ classification. VIVID-Med offers a highly efficient, scalable alternative to deploying resource-heavy vision-language models in clinical settings.
Xiyao Wang, Xiaoyu Tan, Yang Dai +3
Shanghai University of Engineering Science, China · National University of Singapore, Singapore · Case Western Reserve University, United States
Fine-grained visual representations are essential for medical image analysis, particularly when diagnostically relevant evidence is subtle and spatially localized. Modern transformer-based medical vision encoders must therefore learn patch-level representations that are both clinically meaningful and spatially consistent. Without these properties, large vision-language models (LVLMs) operate on an ambiguous visual foundation, limiting their ability to generate clinically reliable and spatially grounded responses. However, existing training strategies for medical vision encoders rarely achieve both objectives. Image-text alignment provides clinically meaningful supervision primarily at the image level, leaving the spatial localization of diagnostic evidence weakly constrained. In contrast, self-supervised learning promotes spatial consistency but lacks the semantic supervision needed to distinguish visually similar yet clinically distinct regions. To address this gap, we present LoFi, a medical vision foundation model built on location-aware fine-grained representation learning. LoFi trains a vision encoder with a lightweight large language model under grounding and grounded captioning objectives. Because these objectives require predicting location from clinical text and vice versa, spatial consistency emerges without any explicit patch-level regularization. To enable training at scale, we construct MedG, a large-scale medical grounding dataset of 4.48M image-text-box triplets curated from 84 datasets spanning 7 modalities. Across phrase grounding, visual question answering, and region-based organ classification under perturbations, LoFi consistently outperforms general-purpose and medical vision foundation models as well as state-of-the-art LVLMs. Code is available at https://github.com/myeongkyunkang/lofi-medg.
Myeongkyun Kang, Yanting Yang, Xiaoxiao Li
The University of British Columbia · Vector Institute
Medical image segmentation is still largely treated as a vision-only problem, although clinical interpretation often relies on textual knowledge of anatomy, location, appearance, and surrounding context. Existing text-guided segmentation methods within the Vision-Language Model (VLM) paradigm often use language only as a late conditioning signal, limiting its influence on visual representation learning. We introduce MedPlex (Medical Plexus of Vision and Language), an end-to-end VLM framework that makes text guidance a continuous, clinically grounded component of segmentation learning. Through Bi-Fusion (Bidirectional Fusion), visual and textual representations evolve jointly across the encoding hierarchy. MedPlex further introduces class-level and region-level concept alignment to organize the shared representation at complementary granularities. Class-level alignment anchors each anatomical target to an aggregated clinical concept profile, while region-level alignment preserves individual concepts, such as shape, location, appearance, and texture, through class-specific visual evidence. In this way, language provides structured supervision throughout the encoder rather than serving only as a late-stage cue. MedPlex achieves state-of-the-art performance across CT and MR benchmarks for multi-organ, cardiac substructure, and tumor segmentation, including settings with real free-text clinical supervision. Code: https://github.com/rafiibnsultan/MedPlex.
Rafi Ibn Sultan, Hui Zhu, Chengyin Li +1
Department of Computer Science Wayne State University · Department of Radiation Oncology Henry Ford Health · Institute for AI and Data Science Wayne State University