让视觉语言模型更懂专业任务,通过融合专用模型提升诊断准确率。
From Generalist to Specialist: Adapting Vision Language Models via Task-Specific Visual Instruction Tuning
- 用任务特化模型引导大模型响应,调整输出分布以适配具体场景。
- 在12个医学诊断数据集上表现优于纯指令微调模型和专用模型。
- 支持灵活接入专用模型,对不完整指令也有强鲁棒性。
大型视觉语言模型(VLMs)结合了大语言模型与视觉编码器,在多种任务中展现出潜力。然而,由于预训练与微调之间的领域差距,其在特定任务上的表现常不理想。本文提出VITask框架,通过集成任务特化模型(TSMs)来增强VLM的任务适应能力。该方法采用三种策略:示例提示(EP)、响应分布对齐(RDA)和对比响应调优(CRT),以调整VLM的输出分布。EP使TSM特征引导VLM,RDA让VLM在推理时无需TSM也能适应,通过示例提示模型学习;CRT则优化正确图像-响应对的排序,降低生成错误响应的风险。在涵盖9种成像模态的12个医学诊断数据集上的实验表明,VITask优于基线指令微调的VLM和单独的TSM,有效融合两者互补特征。此外,VITask具备灵活集成TSM和对不完整指令的鲁棒性等实用优势,是高效通用的任务特定微调方案。代码已开源。
原文摘要 · Abstract (English)
Large vision language models (VLMs) combine large language models with vision encoders, demonstrating promise across various tasks. However, they often underperform in task-specific applications due to domain gaps between pre-training and fine-tuning. We introduce VITask, a novel framework that enhances task-specific adaptability of VLMs by integrating task-specific models (TSMs). VITask employs three key strategies: exemplar prompting (EP), response distribution alignment (RDA), and contrastive response tuning (CRT) to improve the task-specific performance of VLMs by adjusting their response distributions. EP allows TSM features to guide VLMs, while RDA enables VLMs to adapt without TSMs during inference by learning from exemplar-prompted models. CRT further optimizes the ranking of correct image-response pairs, thereby reducing the risk of generating undesired responses. Experiments on 12 medical diagnosis datasets across 9 imaging modalities show that VITask outperforms both vanilla instruction-tuned VLMs and TSMs, showcasing its ability to integrate complementary features from both models effectively. Additionally, VITask offers practical advantages such as flexible TSM integration and robustness to incomplete instructions, making it a versatile and efficient solution for task-specific VLM tuning. Our code are available at https://github.com/baiyang4/VITask.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。