arXiv:2510.17394cs.LGcs.CV2025-10中稿 · and presented at t…

提出MILES调度器,让多模态模型均衡学习各模态数据。

MILES: Modality-Informed Learning Rate Scheduler for Balancing Multimodal Learning

  • 根据模态使用率动态调整学习率,实现多模态训练平衡。
  • 在四个任务上超越7个基线模型,提升多模态与单模态性能。
  • 适合需平衡模态学习的多模态任务,如跨模态融合与缺失模态场景。

多模态神经网络旨在融合多种数据源(模态)以提升性能,但常因模态过拟合导致模型过度依赖单一模态,限制了性能提升,使结果仅小幅优于单模态模型。本文提出模态感知学习率调度器(MILES),通过分析训练中各模态的条件使用率差异,动态调整学习率,平衡各模态的学习速度。我们在四个多模态联合融合任务上评估MILES,并与七个先进基线比较。结果表明,MILES在所有任务和融合方法中均优于基线,有效平衡模态使用,提升多模态性能,同时增强模态编码器能力,可应用于单模态样本或缺失模态场景。本工作凸显了平衡多模态学习对模型性能的关键影响。

原文摘要 · Abstract (English)

The aim of multimodal neural networks is to combine diverse data sources, referred to as modalities, to achieve enhanced performance compared to relying on a single modality. However, training of multimodal networks is typically hindered by modality overfitting, where the network relies excessively on one of the available modalities. This often yields sub-optimal performance, hindering the potential of multimodal learning and resulting in marginal improvements relative to unimodal models. In this work, we present the Modality-Informed Learning ratE Scheduler (MILES) for training multimodal joint fusion models in a balanced manner. MILES leverages the differences in modality-wise conditional utilization rates during training to effectively balance multimodal learning. The learning rate is dynamically adjusted during training to balance the speed of learning from each modality by the multimodal model, aiming for enhanced performance in both multimodal and unimodal predictions. We extensively evaluate MILES on four multimodal joint fusion tasks and compare its performance to seven state-of-the-art baselines. Our results show that MILES outperforms all baselines across all tasks and fusion methods considered in our study, effectively balancing modality usage during training. This results in improved multimodal performance and stronger modality encoders, which can be leveraged when dealing with unimodal samples or absent modalities. Overall, our work highlights the impact of balancing multimodal learning on improving model performance.

多模态学习学习率调度模型平衡

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