Inference-Time Dynamic Modality Selection for Incomplete Multimodal Classification

ICLR 2026 Conference Submission512 Authors

Keywords: Multimodal Learning, Dynamic Neural Network, Missing Modality

TL;DR: We propose DyMo, a novel inference-time dynamic modality selection framework that adaptively selects and integrates reliable recovered modalities for each test sample, maximizing task-relevant information in incomplete multimodal data.

Abstract: 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 integrates 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 datasets show that DyMo significantly outperforms state-of-the-art incomplete/dynamic MDL methods across various missing-data scenarios. Our code will be available at https://github.com/anonymous.

Primary Area: unsupervised, self-supervised, semi-supervised, and supervised representation learning

Submission Number: 512