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

用于医学成像的双重注意力多模态融合学习

Multimodal Fusion Learning with Dual Attention for Medical Imaging

Indian Institute of Technology Ropar(印度理工学院罗帕尔分校);Deakin University(迪肯大学);Queensland University of Technology(昆士兰科技大学);NIMS University(NIMS大学);Jio Institute(吉奥学院);RMIT University(皇家墨尔本理工大学)

Joy Dhar, Nayyar Zaidi, Maryam Haghighat, Puneet Goyal, Sudipta Roy, Azadeh Alavi, Vikas Kumar

AI总结:

本文提出含多分支融合与多模态信息融合双注意力模块的DRIFA机制,构建可嵌入任意深度网络的DRIFA-Net,并用集成Monte Carlo dropout估计不确定性,在五个多模态医学数据集上取得优于现有方法的表现。

摘要:

多模态融合学习在对皮肤癌和脑肿瘤等多种疾病进行分类方面已展现出显著前景。然而,现有方法面临三个关键局限。首先,由于它们专注于某一特定疾病,往往缺乏对其他诊断任务的泛化能力。其次,它们没有充分利用来自不同模态的多种健康记录来学习鲁棒的互补信息。最后,它们通常依赖单一注意力机制,错失了在各种模态内部及跨模态采用多种注意力策略所带来的益处。为解决这些问题,本文提出了一种双重鲁棒信息融合注意力机制(DRIFA),该机制利用两个注意力模块,即多分支融合注意力模块和多模态信息融合注意力模块。DRIFA可以与任何深度神经网络集成,形成一个称为DRIFA-Net的多模态融合学习框架。我们表明,DRIFA的多分支融合注意力为每种模态(例如皮肤镜检查、宫颈涂片、MRI和CT扫描)学习增强的表示,而多模态信息融合注意力模块则学习更精细的多模态共享表示,从而提高网络跨多个任务的泛化能力并提升整体性能。此外,为了估计DRIFA-Net预测的不确定性,我们采用了集成Monte Carlo dropout策略。在五个具有不同模态的公开可用数据集上进行的大量实验表明,我们的方法始终优于最先进的方法。代码可在https://github.com/misti1203/DRIFA-Net获取。

英文摘要:

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.

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

↑