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

元学习模态加权知识蒸馏用于缺失数据下的鲁棒多模态学习

Meta-Learned Modality-Weighted Knowledge Distillation for Robust Multi-Modal Learning with Missing Data

University of Adelaide, Australia(澳大利亚阿德莱德大学);New York University Abu Dhabi, UAE(阿联酋纽约大学阿布扎克分校);University of Surrey, UK(英国萨里大学);Boai hospital of Zhongshan, China(中国中山博爱医院);University of Central Florida, USA(美国佛罗里达大学)
Hu Wang, Salma Hassan, Yuyuan Liu, Congbo Ma, Yuanhong Chen, Qing Li, Jiahui Geng, Bingjie Wang, Yu Tian, Yutong Xie, Jodie Avery, Louise Hull, Ian Reid, Mohamm… 展开作者

Hu Wang, Salma Hassan, Yuyuan Liu, Congbo Ma, Yuanhong Chen, Qing Li, Jiahui Geng, Bingjie Wang, Yu Tian, Yutong Xie, Jodie Avery, Louise Hull, Ian Reid, Mohammad Yaqub, Gustavo Carneiro

AI总结:

提出MetaKD方法,通过元学习估计模态重要性权重并指导知识蒸馏,在缺失模态下保持多模态模型高精度,适用于分割和分类任务,在五个数据集上大幅超越对比模型。

摘要:

在多模态学习中,某些模态比其他模态更具影响力,它们的缺失会对分类/分割精度产生显著影响。针对这一挑战,我们提出了一种名为元学习模态加权知识蒸馏(MetaKD)的新方法,使多模态模型即使在关键模态缺失时也能保持高精度。MetaKD通过元学习过程自适应地估计每个模态的重要性权重。这些学习到的重要性权重指导成对的模态加权知识蒸馏过程,使高重要性模态能够将知识传递给低重要性模态,从而在输入缺失的情况下仍能实现稳健的性能。与以往该领域通常针对特定任务且需要大量修改的方法不同,我们的方法旨在以最小的适应性适用于多种任务(例如分割和分类)。在五个主流数据集上的实验结果,包括三个脑肿瘤分割数据集(BraTS2018、BraTS2019和BraTS2020)、阿尔茨海默病神经影像学倡议(ADNI)分类数据集和Audiovision-MNIST分类数据集,表明所提出的模型能够大幅超越对比模型。代码可在https://github.com/billhhh/MetaKD获取。

英文摘要:

In multi-modal learning, some modalities are more influential than others, and their absence can have a significant impact on classification/segmentation accuracy. Addressing this challenge, we propose a novel approach called Meta-learned Modality-weighted Knowledge Distillation (MetaKD), which enables multi-modal models to maintain high accuracy even when key modalities are missing. MetaKD adaptively estimates the importance weight of each modality through a meta-learning process. These learned importance weights guide a pairwise modality-weighted knowledge distillation process, allowing high-importance modalities to transfer knowledge to lower-importance ones, resulting in robust performance despite missing inputs. Unlike previous methods in the field, which are often task-specific and require significant modifications, our approach is designed to work in multiple tasks (e.g., segmentation and classification) with minimal adaptation. Experimental results on five prevalent datasets, including three Brain Tumor Segmentation datasets (BraTS2018, BraTS2019 and BraTS2020), the Alzheimer's Disease Neuroimaging Initiative (ADNI) classification dataset and the Audiovision-MNIST classification dataset, demonstrate the proposed model is able to outperform the compared models by a large margin. The code is available at https://github.com/billhhh/MetaKD.

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

↑