School of Information Science and Engineering, East China University of Science and Technology, Shanghai 200237, China.
School of Information Science and Engineering, East China University of Science and Technology, Shanghai 200237, China.
Comput Biol Med. 2024 Dec;183:109331. doi: 10.1016/j.compbiomed.2024.109331. Epub 2024 Nov 5.
Brain gliomas are a leading cause of cancer mortality worldwide. Existing glioma segmentation approaches using multi-modal inputs often rely on a simplistic approach of stacking images from all modalities, disregarding modality-specific features that could optimize diagnostic outcomes. This paper introduces STE-Net, a spatial reinforcement hybrid Transformer-based tri-branch multi-modal evidential fusion network designed for conflict-free brain tumor segmentation. STE-Net features two independent encoder-decoder branches that process distinct modality sets, along with an additional branch that integrates features through a cross-modal channel-wise fusion (CMCF) module. The encoder employs a spatial reinforcement hybrid Transformer (SRHT), which combines a Swin Transformer block and a modified convolution block to capture richer spatial information. At the output level, a conflict-free evidential fusion mechanism (CEFM) is developed, leveraging the Dempster-Shafer (D-S) evidence theory and a conflict-solving strategy within a complex network framework. This mechanism ensures balanced reliability among the three output heads and mitigates potential conflicts. Each output is treated as a node in the complex network, and its importance is reassessed through the computation of direct and indirect weights to prevent potential mutual conflicts. We evaluate STE-Net on three public datasets: BraTS2018, BraTS2019, and BraTS2021. Both qualitative and quantitative results demonstrate that STE-Net outperforms several state-of-the-art methods. Statistical analysis further confirms the strong correlation between predicted tumors and ground truth. The code for this project is available at https://github.com/whotwin/STE-Net.
脑胶质瘤是全球癌症死亡的主要原因。现有的基于多模态输入的脑胶质瘤分割方法往往依赖于一种简单的方法,即堆叠来自所有模态的图像,而忽略了可能优化诊断结果的模态特定特征。本文介绍了 STE-Net,这是一种基于空间强化混合 Transformer 的三分支多模态证据融合网络,专为无冲突的脑肿瘤分割而设计。STE-Net 具有两个独立的编码器-解码器分支,分别处理不同的模态集,以及一个通过跨模态通道融合 (CMCF) 模块集成特征的附加分支。编码器采用空间强化混合 Transformer (SRHT),它结合了 Swin Transformer 块和一个修改后的卷积块来捕获更丰富的空间信息。在输出级别,开发了一种无冲突的证据融合机制 (CEFM),利用 Dempster-Shafer (D-S) 证据理论和复杂网络框架内的冲突解决策略。该机制确保三个输出头之间的可靠性平衡,并减轻潜在冲突。每个输出都被视为复杂网络中的一个节点,通过计算直接和间接权重来重新评估其重要性,以防止潜在的相互冲突。我们在三个公共数据集上评估了 STE-Net:BraTS2018、BraTS2019 和 BraTS2021。定性和定量结果都表明,STE-Net 优于几种最先进的方法。统计分析进一步证实了预测肿瘤与地面实况之间的强相关性。该项目的代码可在 https://github.com/whotwin/STE-Net 上获得。