Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Multimodal Fusion Learning with Dual Attention for Medical Imaging

About

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.

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

Related benchmarks

TaskDatasetResultRank
Medical Image SegmentationISIC 2018 (test)
Dice Score90.6
57
SegmentationBraTS 2020
Dice0.8992
36
9-class classificationPathMNIST
Accuracy93.1
32
ClassificationRetinaMNIST
ACC69.38
24
ClassificationPneumoniaMNIST
Accuracy91.83
24
SegmentationLiTS
Dice Score94.5
20
ClassificationOrganAMNIST
Accuracy96.45
14
Medical Image ClassificationIQ-OTHNCCD D2 (test)
Acc99.7
12
ClassificationBreastMNIST
Accuracy88.95
12
ClassificationTissueMNIST
Accuracy73.75
12
Showing 10 of 22 rows

Other info

Follow for update