arXiv:2410.08245cs.LGcs.AI2024-10NeurIPS被引 93

Flex-MoE可灵活处理任意缺失模态组合,提升多模态模型在真实场景的适应性。

Flex-MoE: Modeling Arbitrary Modality Combination via the Flexible Mixture-of-Experts

  • 设计可泛化模态组合的稀疏专家混合框架,用双路由机制分离通用与特定模态知识。
  • 在ADNI和MIMIC-IV数据集上验证,即使部分模态缺失仍保持高准确率。
  • 适合医疗等模态缺失情况复杂的实际应用,尤其适用于个性化诊疗建模。

多模态学习在多个领域日益重要,能整合图像、文本及个人记录等多元数据,常见于医疗场景。然而,当某些模态缺失时,现有框架难以支持任意模态组合,往往依赖单一模态或完整数据,限制了其在真实场景的应用。为此,我们提出Flex-MoE(Flexible Mixture-of-Experts),一个可灵活处理任意模态组合并具备抗缺失能力的新框架。核心思想是:首先通过新的缺失模态库,将观测到的模态组合与其对应的缺失组合进行融合;随后采用独特的稀疏专家混合(Sparse MoE)结构。具体地,先使用全模态样本训练专家,通过通用路由器($Γ$-Router)注入泛化知识;再由专用路由器($Γ$-Router)根据观测到的模态组合,将最高权重分配给对应专家。我们在阿尔茨海默病领域的ADNI数据集(含4种模态)和MIMIC-IV数据集上评估,结果表明Flex-MoE在多种缺失模态场景下均表现出色,有效建模任意模态组合。代码已开源:https://github.com/UNITES-Lab/flex-moe。

原文摘要 · Abstract (English)

Multimodal learning has gained increasing importance across various fields, offering the ability to integrate data from diverse sources such as images, text, and personalized records, which are frequently observed in medical domains. However, in scenarios where some modalities are missing, many existing frameworks struggle to accommodate arbitrary modality combinations, often relying heavily on a single modality or complete data. This oversight of potential modality combinations limits their applicability in real-world situations. To address this challenge, we propose Flex-MoE (Flexible Mixture-of-Experts), a new framework designed to flexibly incorporate arbitrary modality combinations while maintaining robustness to missing data. The core idea of Flex-MoE is to first address missing modalities using a new missing modality bank that integrates observed modality combinations with the corresponding missing ones. This is followed by a uniquely designed Sparse MoE framework. Specifically, Flex-MoE first trains experts using samples with all modalities to inject generalized knowledge through the generalized router ($\mathcal{G}$-Router). The $\mathcal{S}$-Router then specializes in handling fewer modality combinations by assigning the top-1 gate to the expert corresponding to the observed modality combination. We evaluate Flex-MoE on the ADNI dataset, which encompasses four modalities in the Alzheimer's Disease domain, as well as on the MIMIC-IV dataset. The results demonstrate the effectiveness of Flex-MoE highlighting its ability to model arbitrary modality combinations in diverse missing modality scenarios. Code is available at https://github.com/UNITES-Lab/flex-moe.

多模态专家混合缺失数据医疗AI

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。