TDF-Net: Trusted Dynamic Feature Fusion Network for Breast Cancer Diagnosis Using Incomplete Multimodal Ultrasound

Pengfei Yan,Wushuang Gong,Minglei Li,Jiusi Zhang,Xiang Li,Yuchen Jiang,Hao Luo,Hang Zhou
DOI: https://doi.org/10.1016/j.inffus.2024.102592
IF: 18.6
2024-01-01
Information Fusion
Abstract:Ultrasound is a critical imaging technique for diagnosing breast cancer. However, the multimodal breast ultrasound diagnostic process is time-consuming and labor-intensive, heavily dependent on the physician’s extensive expertise. Therefore, developing a computer-aided diagnosis system for breast cancer is essential. Existing diagnostic systems fail to consider the varying impacts of different ultrasound modalities on diagnostic results and struggle to address the issue of missing modalities in clinical practice. Consequently, this paper proposes the Trusted Dynamic Feature Fusion Network (TDF-Net) for diagnosing breast cancer using incomplete multimodal ultrasound data. Initially, this method introduces a dual-branch feature extraction module to capture modality-specific information. Meanwhile, a contrastive clustering loss is designed to enforce the consistency constraint, ensuring the coherence of different modal features within the semantic space for each sample. Additionally, an invertible neural network-based recovery method is suggested to establish mappings between different modalities, enabling the recovery of missing modalities. Finally, a trusted dynamic feature fusion module based on the Dirichlet distribution is proposed to quantify each modality’s contribution to the diagnostic result by considering uncertainty, thereby achieving the dynamic fusion of each modality’s features across different samples. The proposed method is validated on an established multimodal breast ultrasound dataset, demonstrating superior diagnostic performance compared to existing methods, with an average AUC of 98.31%, a 95% confidence interval of [96.98%, 99.64%], and a p-value < 0.05. A pilot study is planned to assess the effectiveness and usability of TDF-Net in clinical settings.
What problem does this paper attempt to address?