arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.06360cs.IRcs.CV
收藏

面向多媒体推荐的模态平衡学习

Modality-Balanced Learning for Multimedia Recommendation

Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所);School of Artificial Intelligence, University of Chinese Academy of Sciences(中国科学院大学人工智能学院)

Jinghao Zhang, Guofan Liu, Qiang Liu, Shu Wu, Liang Wang

AI总结:

针对多模态推荐中的模态失衡及全模态优化不足问题,提出一种基于反事实知识蒸馏的模态平衡学习方法,可即插即用适配不同融合骨干网络,大幅提升推荐性能。

摘要:

已有诸多推荐模型致力于研究如何将多模态内容信息有效融入传统协同过滤框架。多模态信息的应用有望提供更全面的信息,带来更优的性能表现。然而,多模态的融合往往会遭遇模态失衡问题:由于不同模态的信息存在不均衡性,对所有模态优化同一目标会导致弱模态出现优化不足的问题,表现为收敛速度更慢或性能更低。更严重的是,我们发现在多模态推荐模型中,所有模态都存在优化不充分的问题。为解决这些问题,我们提出了一种反事实知识蒸馏方法,该方法能够解决失衡问题并充分利用所有模态的信息。通过模态专属知识蒸馏,它可以引导多模态模型从单模态教师模型中学习模态专属知识。我们还设计了一种新颖的通用-专属蒸馏损失,用于引导多模态学生模型从教师模型中学习更广泛、更深入的知识。此外,为了在训练过程中自适应地重新调整多模态模型对弱模态的关注程度,我们利用反事实推理技术估计每个模态对训练目标的因果效应,借此可以确定弱模态、量化失衡程度,并相应地对蒸馏损失进行重新加权。我们的方法可以作为即插即用模块,适用于后融合和早融合两种骨干网络。在六种骨干网络上开展的大量实验表明,我们提出的方法能够大幅提升性能。源代码将发布于https://github.com/CRIPAC-DIG/Balanced-Multimodal-Rec。

英文摘要:

Many recommender models have been proposed to investigate how to incorporate multimodal content information into traditional collaborative filtering framework effectively. The use of multimodal information is expected to provide more comprehensive information and lead to superior performance. However, the integration of multiple modalities often encounters the modal imbalance problem: since the information in different modalities is unbalanced, optimizing the same objective across all modalities leads to the under-optimization problem of the weak modalities with a slower convergence rate or lower performance. Even worse, we find that in multimodal recommendation models, all modalities suffer from the problem of insufficient optimization. To address these issues, we propose a Counterfactual Knowledge Distillation method that could solve the imbalance problem and make the best use of all modalities. Through modality-specific knowledge distillation, it could guide the multimodal model to learn modality-specific knowledge from uni-modal teachers. We also design a novel generic-and-specific distillation loss to guide the multimodal student to learn wider-and-deeper knowledge from teachers. Additionally, to adaptively recalibrate the focus of the multimodal model towards weaker modalities during training, we estimate the causal effect of each modality on the training objective using counterfactual inference techniques, through which we could determine the weak modalities, quantify the imbalance degree and re-weight the distillation loss accordingly. Our method could serve as a plug-and-play module for both late-fusion and early-fusion backbones. Extensive experiments on six backbones show that our proposed method can improve the performance by a large margin. The source code will be released at \url{https://github.com/CRIPAC-DIG/Balanced-Multimodal-Rec}

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

↑