用多模态知识蒸馏压缩模型,小网络也能高效处理医学数据。
MIND: Modality-Informed Knowledge Distillation Framework for Multimodal Clinical Prediction Tasks
- 用不同规模的单模态教师网络,向小模型传递多模态知识。
- 在5个临床任务上,小模型性能超越现有最优方法。
- 支持无缺失模态时直接融合,适合医疗场景部署。
多模态融合通过整合多源信息提升特征表示能力,但医疗数据集通常规模较小,且模态增多会带来模型膨胀问题。为应对这一挑战,本文提出模态感知知识蒸馏框架MIND,将多个预训练的异构单模态网络作为教师,指导一个更小的多模态学生模型学习。该框架采用多头联合融合机制,无需对缺失模态进行插补或掩码,可有效利用单模态样本。实验在时间序列与胸片数据上的二分类及多标签临床预测任务中验证,MIND在五项任务上均优于当前主流基线方法,涵盖多种融合策略与架构。此外,在三个非医疗多模态多分类数据集上也展现了良好泛化能力。
原文摘要 · Abstract (English)
Multimodal fusion leverages information across modalities to learn better feature representations with the goal of improving performance in fusion-based tasks. However, multimodal datasets, especially in medical settings, are typically smaller than their unimodal counterparts, which can impede the performance of multimodal models. Additionally, the increase in the number of modalities is often associated with an overall increase in the size of the multimodal network, which may be undesirable in medical use cases. Utilizing smaller unimodal encoders may lead to sub-optimal performance, particularly when dealing with high-dimensional clinical data. In this paper, we propose the Modality-INformed knowledge Distillation (MIND) framework, a multimodal model compression approach based on knowledge distillation that transfers knowledge from ensembles of pre-trained deep neural networks of varying sizes into a smaller multimodal student. The teacher models consist of unimodal networks, allowing the student to learn from diverse representations. MIND employs multi-head joint fusion models, as opposed to single-head models, enabling the use of unimodal encoders in the case of unimodal samples without requiring imputation or masking of absent modalities. As a result, MIND generates an optimized multimodal model, enhancing both multimodal and unimodal representations. It can also be leveraged to balance multimodal learning during training. We evaluate MIND on binary and multilabel clinical prediction tasks using time series data and chest X-ray images. Additionally, we assess the generalizability of the MIND framework on three non-medical multimodal multiclass datasets. Experimental results demonstrate that MIND enhances the performance of the smaller multimodal network across all five tasks, as well as various fusion methods and multimodal architectures, compared to state-of-the-art baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。