arXiv:2412.01248cs.CV2024-12被引 30

提出双注意力融合框架,提升多模态医学影像诊断的泛化能力

Multimodal Fusion Learning with Dual Attention for Medical Imaging

  • 设计双注意力模块,分别处理模态内与跨模态信息融合
  • 在五个数据集上显著优于现有方法,提升诊断准确率
  • 适合医疗影像分析、多模态学习研究者使用

多模态融合学习在皮肤癌、脑瘤等疾病分类中展现出巨大潜力,但现有方法存在三大局限:难以泛化到其他诊断任务;未能充分挖掘多种健康记录间的互补信息;通常仅依赖单一注意力机制。为此,本文提出双鲁棒信息融合注意力机制(DRIFA),包含多分支融合注意力和多模态信息融合注意力两个模块。DRIFA可集成于任意深度神经网络,形成名为DRIFA-Net的多模态融合学习框架。实验表明,多分支融合注意力提升了皮肤镜、宫颈涂片、MRI、CT等模态的表征能力,而多模态融合注意力则学习到更精细的共享表示,增强了模型在多任务上的泛化性与整体性能。此外,通过集成蒙特卡洛丢弃策略评估预测不确定性。在五个公开数据集上的大量实验验证了该方法持续优于当前最优方法。代码已开源。

原文摘要 · Abstract (English)

Multimodal fusion learning has shown significant promise in classifying various diseases such as skin cancer and brain tumors. However, existing methods face three key limitations. First, they often lack generalizability to other diagnosis tasks due to their focus on a particular disease. Second, they do not fully leverage multiple health records from diverse modalities to learn robust complementary information. And finally, they typically rely on a single attention mechanism, missing the benefits of multiple attention strategies within and across various modalities. To address these issues, this paper proposes a dual robust information fusion attention mechanism (DRIFA) that leverages two attention modules, i.e. multi-branch fusion attention module and the multimodal information fusion attention module. DRIFA can be integrated with any deep neural network, forming a multimodal fusion learning framework denoted as DRIFA-Net. We show that the multi-branch fusion attention of DRIFA learns enhanced representations for each modality, such as dermoscopy, pap smear, MRI, and CT-scan, whereas multimodal information fusion attention module learns more refined multimodal shared representations, improving the network's generalization across multiple tasks and enhancing overall performance. Additionally, to estimate the uncertainty of DRIFA-Net predictions, we have employed an ensemble Monte Carlo dropout strategy. Extensive experiments on five publicly available datasets with diverse modalities demonstrate that our approach consistently outperforms state-of-the-art methods. The code is available at https://github.com/misti1203/DRIFA-Net.

多模态融合医学影像注意力机制诊断系统

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。