解决多模态学习中主导模态压制其他模态的问题,提升模型整体表现。
On-the-fly Modulation for Balanced Multimodal Learning
- 训练时动态调节各模态的特征与梯度,防止强势模态主导优化过程。
- 在多个任务上显著提升性能,尤其在复杂多模态场景中效果明显。
- 方法简单有效,适用于通用和任务导向的多模态模型,适合研究者使用。
多模态学习本应通过融合不同模态信息提升模型性能,但现有联合训练策略对所有模态采用统一目标,导致模态间表示不平衡且优化不足。例如,足球视觉或风声等更具判别性的模态可能主导训练过程,使其他模态被严重欠优化。本文从前向传播与反向传播阶段分析该现象,并提出即插即用的On-the-fly Prediction Modulation(OPM)与On-the-fly Gradient Modulation(OGM)策略:前者在前向传播中动态丢弃主导模态特征,后者在反向传播中抑制其梯度。实验表明,该方法在多种多模态任务中均取得显著提升,不仅增强基础与任务导向模型性能,也适用于复杂多模态任务,展现出强有效性与灵活性。源码已公开于 url{https://github.com/GeWu-Lab/BML_TPAMI2024}。
原文摘要 · Abstract (English)
Multimodal learning is expected to boost model performance by integrating information from different modalities. However, its potential is not fully exploited because the widely-used joint training strategy, which has a uniform objective for all modalities, leads to imbalanced and under-optimized uni-modal representations. Specifically, we point out that there often exists modality with more discriminative information, e.g., vision of playing football and sound of blowing wind. They could dominate the joint training process, resulting in other modalities being significantly under-optimized. To alleviate this problem, we first analyze the under-optimized phenomenon from both the feed-forward and the back-propagation stages during optimization. Then, On-the-fly Prediction Modulation (OPM) and On-the-fly Gradient Modulation (OGM) strategies are proposed to modulate the optimization of each modality, by monitoring the discriminative discrepancy between modalities during training. Concretely, OPM weakens the influence of the dominant modality by dropping its feature with dynamical probability in the feed-forward stage, while OGM mitigates its gradient in the back-propagation stage. In experiments, our methods demonstrate considerable improvement across a variety of multimodal tasks. These simple yet effective strategies not only enhance performance in vanilla and task-oriented multimodal models, but also in more complex multimodal tasks, showcasing their effectiveness and flexibility. The source code is available at \url{https://github.com/GeWu-Lab/BML_TPAMI2024}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。