用偏好优化让单模态大模型学会多模态医学知识,提升疾病预测和组织分类性能。
Multimodal Integrated Knowledge Transfer to Large Language Models through Preference Optimization with Biomedical Applications
- 通过偏好优化,将多模态医学模型的知识迁移至仅处理文本或图像的LLM。
- 在罕见病预测任务中,仅用文本输入的模型超越SFT、RAG、DPO及4050亿参数大模型。
- 适用于医学图像与文本分析场景,适合医疗AI研发者和临床辅助系统构建者。
高质量多模态生物医学数据稀缺,限制了预训练大语言模型(LLM)在专业医学任务上的微调效果。为此,我们提出MINT(Multimodal Integrated kNowledge Transfer)框架,通过偏好优化将单模态大解码器模型与来自多模态生物医学数据的领域特定决策模式对齐。尽管支持多种优化技术,我们以奇比率偏好优化(ORPO)为核心实现。该策略使对齐后的LLM能仅凭文本或图像输入完成预测任务,同时保留多模态学习的知识。MINT利用上游多模态机器学习(MML)模型(在高质量多模态数据上训练)生成偏好数据集,向下游文本或图像单模态LLM传递领域知识。我们在两项关键应用中验证其有效性:(1) 基于文本的罕见遗传病预测,使用在面部照片与临床笔记上训练的多模态编码器生成偏好数据,对轻量级Llama 3.2-3B-Instruct进行对齐。仅依赖文本输入的模型优于经SFT、RAG或DPO训练的模型,甚至超过Llama 3.1-405B-Instruct;(2) 基于细胞核图像的组织类型分类,采用视觉-语言基础模型作为偏好生成器,融合文本与病理图像知识,对下游图像单模态模型进行对齐。结果表明,经过MINT对齐的模型显著提升了Llama 3.2-Vision-11B-Instruct在组织分类上的表现。综上,MINT提供了一种有效策略,通过偏好优化将多模态医学专长迁移至单模态大模型。
原文摘要 · Abstract (English)
The scarcity of high-quality multimodal biomedical data limits the ability to effectively fine-tune pretrained Large Language Models (LLMs) for specialized biomedical tasks. To address this challenge, we introduce MINT (Multimodal Integrated kNowledge Transfer), a framework that aligns unimodal large decoder models with domain-specific decision patterns from multimodal biomedical data through preference optimization. While MINT supports different optimization techniques, we primarily implement it with the Odds Ratio Preference Optimization (ORPO) framework as its backbone. This strategy enables the aligned LLMs to perform predictive tasks using text-only or image-only inputs while retaining knowledge learnt from multimodal data. MINT leverages an upstream multimodal machine learning (MML) model trained on high-quality multimodal data to transfer domain-specific insights to downstream text-only or image-only LLMs. We demonstrate its effectiveness through two key applications: (1) Rare genetic disease prediction from texts, where MINT uses a multimodal encoder model, trained on facial photos and clinical notes, to generate a preference dataset for aligning a lightweight Llama 3.2-3B-Instruct. Despite relying on text input only, the MINT-derived model outperforms models trained with SFT, RAG, or DPO, and even outperforms Llama 3.1-405B-Instruct. (2) Tissue type classification using cell nucleus images, where MINT uses a vision-language foundation model as the preference generator, containing knowledge learnt from both text and histopathological images to align downstream image-only models. The resulting MINT-derived model significantly improves the performance of Llama 3.2-Vision-11B-Instruct on tissue type classification. In summary, MINT provides an effective strategy to align unimodal LLMs with high-quality multimodal expertise through preference optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。