用轻量方法让医学图文模型兼顾通用性与诊断细节。
ACE-LoRA: Graph-Attentive Context Enhancement for Parameter-Efficient Adaptation of Medical Vision-Language Models
- 在冻结主干上加低秩模块,结合图注意力增强局部诊断线索。
- 仅增0.95M参数,零样本测试中多任务表现超现有方法。
- 适合想高效适配医学模型又不丢关键细节的研究者。
CLIP类图文模型在自然图像上成功推动了医学领域的发展,但现有方法存在两种极端:单一领域训练的专用模型虽保留细节却泛化差,多领域训练的通用模型虽具备广泛语义但稀释了细粒度诊断信息。为弥合这一权衡,本文提出ACE-LoRA,一种面向通用医学图文模型的参数高效适配框架,保持强零样本泛化能力。该方法将低秩适配(LoRA)模块嵌入冻结的图像-文本编码器,并引入基于注意力的上下文增强超图神经网络(ACE-HGNN),捕捉超越成对相似性的高阶上下文关联,以全局表征融入局部诊断线索,解决了以往参数高效微调方法忽视细粒度特征的问题。为进一步强化跨模态对齐,设计标签引导的InfoNCE损失,有效抑制语义相关图文对间的误负样本。尽管仅增加0.95M可训练参数,ACE-LoRA在涵盖多个领域的零样本分类、分割与检测基准上持续优于最先进医疗VLM与PEFT基线。代码已开源:https://github.com/icon-lab/ACE-LoRA。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。