Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

ACE-LoRA: Graph-Attentive Context Enhancement for Parameter-Efficient Adaptation of Medical Vision-Language Models

About

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.

M. Arda Ayd{\i}n, Melih B. Yilmaz, Aykut Ko\c{c}, Tolga \c{C}ukur• 2026

Related benchmarks

TaskDatasetResultRank
Object DetectionRSNA
mAP (%)21.29
106
Image ClassificationRSNA (test)
AUC87.19
59
Medical Semantic SegmentationSIIM Pneumothorax
Dice Score46.34
46
Image ClassificationMHIST (test)
Accuracy57.27
41
Image ClassificationSIIM (test)--
30
Image ClassificationCheXpert 5x200 (test)
Accuracy49.8
19
ClassificationLC25000 lung cancer (test)
Accuracy84.03
9
ClassificationLC25000 Colon (test)
Accuracy90.39
5
Showing 8 of 8 rows

Other info

GitHub

Follow for update