ACE-LoRA: Graph-Attentive Context Enhancement for Parameter-Efficient Adaptation of Medical Vision-Language Models
Organizations: Bilkent University · National Magnetic Resonance Research Center (UMRAM)
Abstract
The success of CLIP-like vision-language models (VLMs) on natural images has inspired medical counterparts, yet existing approaches largely fall into two extremes: specialist models trained on single-domain data, which capture domain-specific details but generalize poorly, and generalist medical VLMs trained on multi-domain data, which retain broad semantics but dilute fine-grained diagnostic cues. Bridging this specialization-generalization trade-off remains challenging. To address this problem, we propose ACE-LoRA, a parameter-efficient adaptation framework for generalist medical VLMs that maintains robust zero-shot generalization. ACE-LoRA integrates Low-Rank Adaptation (LoRA) modules into frozen image-text encoders and introduces an Attention-based Context Enhancement Hypergraph Neural Network (ACE-HGNN) module that captures higher-order contextual interactions beyond pairwise similarity to enrich global representations with localized diagnostic cues, addressing a key limitation of prior Parameter-Efficient Fine-Tuning (PEFT) methods that overlook fine-grained details. To further enhance cross-modal alignment, we formulate a label-guided InfoNCE loss to effectively suppress false negatives between semantically related image-text pairs. Despite adding only 0.95M trainable parameters, ACE-LoRA consistently outperforms state-of-the-art medical VLMs and PEFT baselines across zero-shot classification, segmentation, and detection benchmarks spanning multiple domains. Our code is available at https://github.com/icon-lab/ACE-LoRA.
Figures & tables
| Method | CheXpert 5 200 | RSNA | SIIM | |||
|---|---|---|---|---|---|---|
| ACC (%) | AUC (%) | ACC (%) | AUC (%) | ACC (%) | AUC (%) | |
| BiomedCLIP [ 62 ] | 35.50 | 67.46 | 74.34 | 81.14 | 61.40 | 64.31 |
| CoOp [ 69 ] | 40.90 | 69.32 | 75.07 | 79.92 | 65.69 | 69.87 |
| CoCoOp [ 68 ] | 40.10 | 69.74 | 75.02 | 79.95 | 67.24 | 71.21 |
| CLIP-Adapter [ 11 ] | 42.70 | 77.55 | 68.47 | 75.14 | 60.38 | 61.26 |
| TaskRes [ 58 ] | 36.50 | 68.17 | 74.77 | 81.36 | 63.53 | 67.14 |
| Method | CheXpert 5 200 | RSNA | SIIM | |||
|---|---|---|---|---|---|---|
| ACC (%) | AUC (%) | ACC (%) | AUC (%) | ACC (%) | AUC (%) | |
| Specialist Medical VLMs | ||||||
| ConVIRT [ 64 ] | 45.10 | 75.82 | 42.64 | 42.13 | 46.54 | 49.09 |
| GLoRIA [ 18 ] | 24.80 | 71.60 | 68.42 | 76.73 | 48.67 | 50.13 |
| MGCA [ 49 ] | 21.40 | 52.62 | 63.75 | 68.21 | 58.52 | 62.55 |
| BioVIL [ 2 ] | 48.40 | 76.99 | 56.66 | 62.23 | 53.32 | 59.99 |
| Components | Datasets | ||||
|---|---|---|---|---|---|
| LoRA | ACE-HGNN | Label | CheXpert | RSNA | SIIM |
| ✗ | ✗ | ✗ | 35.50 | 74.34 | 61.40 |
| ✓ | ✗ | ✗ | 45.60 | 67.07 | 68.29 |
| ✓ | ✓ | ✗ | 49.20 | 74.64 | 70.56 |
| ✓ | ✗ | ✓ | 45.80 | 72.85 | 72.50 |
| ✓ | ✓ | ✓ | 49.80 | 79.54 | 73.35 |
| Method | CheXpert | RSNA | SIIM |
|---|---|---|---|
| BMC-CLIP | 24.70 | 53.83 | 48.67 |
| + ACE-LoRA | 50.50 | 67.35 | 70.70 |
| +25.80 | +13.52 | +22.03 |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Task | Dataset | Classes | Total Size | Train/Val/Test (%) |
|---|---|---|---|---|
| Classification | CheXpert 5 200 [ 21 ] | 5 | 1000 | -/-/100 |
| Classification | RSNA [ 43 ] | 2 | 29,684 | -/–/15 |
| Classification | SIIM [ 61 ] | 2 | 12,047 | –/–/30 |
| Classification | NIH ChestX-ray [ 50 ] | 14 | 112,120 | –/5/- |
| Classification | LC25000 (Lung) [ 3 ] | 3 | 15,000 | –/–/100 |
| Classification | LC25000 (Colon) [ 3 ] | 2 | 10,000 | –/–/100 |
| Dataset | Classes |
|---|---|
| CheXpert 5 200 [ 21 ] | "Atelectasis", "Cardiomegaly", "Consolidation", "Edema", "Pleural Effusion" |
| RSNA [ 43 ] | "No Finding", "Pneumonia" |
| SIIM [ 61 ] | "No Finding", "Pneumothorax" |
| NIH ChestX-ray [ 50 ] | "Atelectasis", "Cardiomegaly", "Effusion", "Infiltration", "Mass", ’"Nodule", "Pneumonia", "Pneumothorax", "Consolidation", "Edema", "Emphysema", "Fibrosis", "Pleural Thickening", "Hernia" |
| LC25000 (Lung) [ 3 ] | "Lung adenocarcinoma", "Benign lung", "Lung squamous cell carcinoma" |
| LC25000 (Colon) [ 3 ] | "Colon adenocarcinoma", "Benign colonic tissue" |
| IE | TE | CheXpert | RSNA | SIIM |
|---|---|---|---|---|
| ✗ | ✗ | 45.80 | 72.85 | 72.50 |
| ✓ | ✗ | 49.00 | 75.84 | 73.60 |
| ✗ | ✓ | 50.40 | 76.92 | 72.80 |
| ✓ | ✓ | 49.80 | 79.54 | 73.35 |
| Method | CheXpert | RSNA | SIIM |
|---|---|---|---|
| BiomedCLIP [ 62 ] | 35.50 | 74.34 | 61.40 |
| TransCLIP [ 60 ] | 37.50 | 73.54 | 65.36 |
VLM Adaptation
Vision-language model adaptation tailors pretrained image-text and multimodal models to new tasks and domains while retaining transferable capabilities.
See all 545 papers