arXiv:2601.22853cs.CV2026-01中稿 · ICLR被引 3

动态选择可用模态,提升缺失数据下的多模态分类准确率

Inference-Time Dynamic Modality Selection for Incomplete Multimodal Classification

  • 测试时动态选择可靠模态,避免盲目丢弃或补全
  • 通过任务损失估算信息量,实现精准模态筛选
  • 适用于医疗影像等真实场景中不完整数据的分类

多模态深度学习在多个领域取得显著进展,但实际部署常受限于模态数据缺失。现有方法要么丢弃缺失模态(损失有用信息),要么恢复缺失模态(引入无关噪声),陷入“丢弃-补全”困境。本文提出 DyMo 框架,在推理阶段动态选择并融合可信的恢复模态,突破传统二元策略。核心是设计一种新选择算法,最大化每条测试样本的多模态任务相关性信息。由于测试时数据分布未知,无法直接估计信息量,我们理论建立信息与任务损失的关联,并以任务损失作为可计算的代理。基于此构建了合理的奖励函数指导模态选择。同时设计兼容任意模态组合的灵活网络结构及针对性训练策略,实现鲁棒表示学习。在多种自然图像与医学图像数据集上的大量实验表明,DyMo 在各类缺失场景下均显著优于当前最优的不完整/动态多模态学习方法。代码已开源。

原文摘要 · Abstract (English)

Multimodal deep learning (MDL) has achieved remarkable success across various domains, yet its practical deployment is often hindered by incomplete multimodal data. Existing incomplete MDL methods either discard missing modalities, risking the loss of valuable task-relevant information, or recover them, potentially introducing irrelevant noise, leading to the discarding-imputation dilemma. To address this dilemma, in this paper, we propose DyMo, a new inference-time dynamic modality selection framework that adaptively identifies and fuses reliable recovered modalities, fully exploring task-relevant information beyond the conventional discard-or-impute paradigm. Central to DyMo is a novel selection algorithm that maximizes multimodal task-relevant information for each test sample. Since direct estimation of such information at test time is intractable due to the unknown data distribution, we theoretically establish a connection between information and the task loss, which we compute at inference time as a tractable proxy. Building on this, a novel principled reward function is proposed to guide modality selection. In addition, we design a flexible multimodal network architecture compatible with arbitrary modality combinations, alongside a tailored training strategy for robust representation learning. Extensive experiments on diverse natural and medical image datasets show that DyMo significantly outperforms state-of-the-art incomplete/dynamic MDL methods across various missing-data scenarios. Our code is available at https://github.com//siyi-wind/DyMo.

多模态学习缺失数据动态选择

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