arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2310.01035cs.CVcs.LG

用于缺失模态多模态学习的可学习跨模态知识蒸馏

Learnable Cross-modal Knowledge Distillation for Multi-modal Learning with Missing Modality

The University of Adelaide(阿德莱德大学);Alibaba DAMO Academy(阿里巴巴达摩院);University of Surrey(萨里大学)

Hu Wang, Congbo Ma, Jianpeng Zhang, Yuan Zhang, Jodie Avery, Louise Hull, Gustavo Carneiro

AI总结:

针对多模态学习中重要模态缺失导致性能下降的问题,本文提出LCKD,通过教师选举和跨模态知识蒸馏让可用模态继承重要模态知识,并在BraTS2018上显著提升分割性能。

摘要:

缺失模态问题在多模态模型中既关键又难以处理。在多模态任务中,某些模态通常比其他模态贡献更大,而如果这些重要模态缺失,模型性能会显著下降。当前的多模态方法尚未探究这一事实:它们通过特征重建或盲目聚合其他模态的特征来恢复缺失模态的表示,而不是从表现最佳的模态中提取有用信息。本文提出一种可学习跨模态知识蒸馏(LCKD)模型,从跨模态视角自适应识别重要模态并从中蒸馏知识,以帮助其他模态解决缺失模态问题。该方法引入教师选举程序,依据各模态在特定任务上的单模态表现选择最“合格”的教师。随后,针对每项任务在教师模态与学生模态之间执行跨模态知识蒸馏,将模型参数推向对所有任务都有利的位置。因此,即使某些任务的教师模态在测试时缺失,可用的学生模态也能基于从自动选举出的教师模态中学到的知识足够好地完成任务。在2018年脑肿瘤分割数据集(BraTS2018)上的实验表明,LCKD以相当大的优势优于其他方法;以分割Dice分数衡量,其将增强肿瘤、肿瘤核心和全肿瘤的当前最优性能分别提升了3.61%、5.99%和3.76%。

英文摘要:

The problem of missing modalities is both critical and non-trivial to be handled in multi-modal models. It is common for multi-modal tasks that certain modalities contribute more compared to other modalities, and if those important modalities are missing, the model performance drops significantly. Such fact remains unexplored by current multi-modal approaches that recover the representation from missing modalities by feature reconstruction or blind feature aggregation from other modalities, instead of extracting useful information from the best performing modalities. In this paper, we propose a Learnable Cross-modal Knowledge Distillation (LCKD) model to adaptively identify important modalities and distil knowledge from them to help other modalities from the cross-modal perspective for solving the missing modality issue. Our approach introduces a teacher election procedure to select the most ``qualified'' teachers based on their single modality performance on certain tasks. Then, cross-modal knowledge distillation is performed between teacher and student modalities for each task to push the model parameters to a point that is beneficial for all tasks. Hence, even if the teacher modalities for certain tasks are missing during testing, the available student modalities can accomplish the task well enough based on the learned knowledge from their automatically elected teacher modalities. Experiments on the Brain Tumour Segmentation Dataset 2018 (BraTS2018) shows that LCKD outperforms other methods by a considerable margin, improving the state-of-the-art performance by 3.61% for enhancing tumour, 5.99% for tumour core, and 3.76% for whole tumour in terms of segmentation Dice score.

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

↑