arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.09705cs.CVcs.AIcs.MM

面向平衡多模态学习的诊断与再学习

Diagnosing and Re-learning for Balanced Multimodal Learning

Gaoling School of Artificial Intelligence, Renmin University of China(中国人民大学高瓴人工智能学院);Department of Electronic Engineering, Tsinghua University(清华大学电子工程系);Engineering Research Center of Next-Generation Search and Recommendation(下一代搜索与推荐技术工程研究中心)

Yake Wei, Siwei Li, Ruoxuan Feng, Di Hu

AI总结:

针对多模态学习不平衡问题,提出基于单模态表示空间可分离性的诊断与再学习方法,兼顾模态容量限制与所有模态,实现多模态学习的平衡与增强,经多类实验验证性能优异。

摘要:

为解决多模态学习不平衡问题(即模型更偏好特定模态的训练),现有方法以模态间性能差异为依据,从不同角度调控单模态编码器的训练,但忽略了模态容量的内在限制。信息量匮乏的模态可能被判定为“学习效果较差”的模态,这会迫使模型记忆更多噪声,反而对多模态模型的能力产生负面影响。此外,当前的模态调制方法仅狭隘地聚焦于选定的学习效果较差的模态,甚至会抑制其他模态的训练。因此,在平衡过程中必须考虑模态容量的内在限制,并兼顾所有模态。为此,我们提出了诊断与再学习(Diagnosing & Re-learning)方法。该方法首先基于单模态表示空间的可分离性估计每个模态的学习状态,再据此对对应的单模态编码器进行软重初始化,从而避免对信息量匮乏模态的过度强调。同时,该方法增强了学习效果较差模态的编码器,也避免了其他模态的过训练,由此有效平衡并增强了多模态学习。覆盖多种模态类型与多模态框架的实验表明,我们这一简单却有效的方法在平衡多模态学习方面性能优异。源代码与数据集可访问https://github.com/GeWu-Lab/Diagnosing_Relearning_ECCV2024 获取。

英文摘要:

To overcome the imbalanced multimodal learning problem, where models prefer the training of specific modalities, existing methods propose to control the training of uni-modal encoders from different perspectives, taking the inter-modal performance discrepancy as the basis. However, the intrinsic limitation of modality capacity is ignored. The scarcely informative modalities can be recognized as ``worse-learnt'' ones, which could force the model to memorize more noise, counterproductively affecting the multimodal model ability. Moreover, the current modality modulation methods narrowly concentrate on selected worse-learnt modalities, even suppressing the training of others. Hence, it is essential to consider the intrinsic limitation of modality capacity and take all modalities into account during balancing. To this end, we propose the Diagnosing \& Re-learning method. The learning state of each modality is firstly estimated based on the separability of its uni-modal representation space, and then used to softly re-initialize the corresponding uni-modal encoder. In this way, the over-emphasizing of scarcely informative modalities is avoided. In addition, encoders of worse-learnt modalities are enhanced, simultaneously avoiding the over-training of other modalities. Accordingly, multimodal learning is effectively balanced and enhanced. Experiments covering multiple types of modalities and multimodal frameworks demonstrate the superior performance of our simple-yet-effective method for balanced multimodal learning. The source code and dataset are available at \url{https://github.com/GeWu-Lab/Diagnosing_Relearning_ECCV2024}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑