提出解耦梯度学习,解决多模态模型性能不如单模态的问题
Boosting Multimodal Learning via Disentangled Gradient Learning
- 分离模态编码器与融合模块的优化路径,切断相互干扰的梯度
- 在多个任务和框架上提升性能,最高达+5.2%准确率
- 适合需要强跨模态融合的视觉-语言任务研究者
多模态学习常面临优化不足问题,表现甚至劣于单模态学习。现有方法归因于模态间学习不平衡,通过梯度调制进行再平衡,但无法解释为何多模态中主导模态仍逊于单模态表现。本文揭示多模态模型中模态编码器与模态融合模块间的优化冲突:跨模态融合会降低反馈至各模态编码器的梯度强度,导致其性能低于单模态模型。为此,提出解耦梯度学习(DGL)框架,将多模态损失反传至模态编码器的梯度截断,并替换为单模态损失梯度;同时移除单模态损失反传至融合模块的梯度。该设计消除两者间的梯度干扰,保障独立优化。在多种模态、任务及具强跨模态交互的框架上进行广泛实验,验证了DGL的有效性与通用性。代码已开源。
原文摘要 · Abstract (English)
Multimodal learning often encounters the under-optimized problem and may have worse performance than unimodal learning. Existing methods attribute this problem to the imbalanced learning between modalities and rebalance them through gradient modulation. However, they fail to explain why the dominant modality in multimodal models also underperforms that in unimodal learning. In this work, we reveal the optimization conflict between the modality encoder and modality fusion module in multimodal models. Specifically, we prove that the cross-modal fusion in multimodal models decreases the gradient passed back to each modality encoder compared with unimodal models. Consequently, the performance of each modality in the multimodal model is inferior to that in the unimodal model. To this end, we propose a disentangled gradient learning (DGL) framework to decouple the optimization of the modality encoder and modality fusion module in the multimodal model. DGL truncates the gradient back-propagated from the multimodal loss to the modality encoder and replaces it with the gradient from unimodal loss. Besides, DGL removes the gradient back-propagated from the unimodal loss to the modality fusion module. This helps eliminate the gradient interference between the modality encoder and modality fusion module while ensuring their respective optimization processes. Finally, extensive experiments on multiple types of modalities, tasks, and frameworks with dense cross-modal interaction demonstrate the effectiveness and versatility of the proposed DGL. Code is available at \href{https://github.com/shicaiwei123/ICCV2025-GDL}{https://github.com/shicaiwei123/ICCV2025-GDL}
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。