用大模型补全缺失多模态数据,提升生存预测准确性
Distilled Prompt Learning for Incomplete Multimodal Survival Prediction
- 分两阶段提示:先提取单模态知识,再用已有模态引导大模型推断缺失模态
- 在多种缺失场景下优于现有方法,显著提升生存预测性能
- 适合临床中多模态数据不全的生存分析任务
病理图像与基因谱型等多模态数据融合广泛应用于精准生存预测。尽管多模态生存模型进展迅速,但完整收集各模态数据仍面临重大挑战,限制了其在临床中的应用。现有方法通常仅能补偿部分缺失模态知识。为此,我们提出一种基于大语言模型(LLM)的蒸馏提示学习框架(DisPro),通过两阶段提示实现对缺失模态信息的全面补全。第一阶段,单模态提示(UniPro)提取各模态的知识分布,为后续补充缺失模态的特异性知识做准备;第二阶段,多模态提示(MultiPro)利用可用模态作为提示,让大模型推断缺失模态,获得共性信息,同时将第一阶段获取的单模态知识注入多模态推理,以弥补缺失模态的特异性知识。在多种缺失情景下的大量实验验证了该方法的优越性。
原文摘要 · Abstract (English)
The integration of multimodal data including pathology images and gene profiles is widely applied in precise survival prediction. Despite recent advances in multimodal survival models, collecting complete modalities for multimodal fusion still poses a significant challenge, hindering their application in clinical settings. Current approaches tackling incomplete modalities often fall short, as they typically compensate for only a limited part of the knowledge of missing modalities. To address this issue, we propose a Distilled Prompt Learning framework (DisPro) to utilize the strong robustness of Large Language Models (LLMs) to missing modalities, which employs two-stage prompting for compensation of comprehensive information for missing modalities. In the first stage, Unimodal Prompting (UniPro) distills the knowledge distribution of each modality, preparing for supplementing modality-specific knowledge of the missing modality in the subsequent stage. In the second stage, Multimodal Prompting (MultiPro) leverages available modalities as prompts for LLMs to infer the missing modality, which provides modality-common information. Simultaneously, the unimodal knowledge acquired in the first stage is injected into multimodal inference to compensate for the modality-specific knowledge of the missing modality. Extensive experiments covering various missing scenarios demonstrated the superiority of the proposed method. The code is available at https://github.com/Innse/DisPro.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。