Liew Andrea, Lee Chun Cheng, Lan Boon Leong, Tan Maxine
Electrical and Computer Systems Engineering Discipline, School of Engineering, Monash University Malaysia, Bandar Sunway, 47500, Malaysia.
Radiology Department, Sunway Medical Centre, Bandar Sunway, 47500, Malaysia.
Comput Biol Med. 2021 Sep;136:104690. doi: 10.1016/j.compbiomed.2021.104690. Epub 2021 Jul 28.
Convolutional neural networks (CNNs) have been used quite successfully for semantic segmentation of brain tumors. However, current CNNs and attention mechanisms are stochastic in nature and neglect the morphological indicators used by radiologists to manually annotate regions of interest. In this paper, we introduce a channel and spatial wise asymmetric attention (CASPIAN) by leveraging the inherent structure of tumors to detect regions of saliency. To demonstrate the efficacy of our proposed layer, we integrate this into a well-established convolutional neural network (CNN) architecture to achieve higher Dice scores, with less GPU resources. Also, we investigate the inclusion of auxiliary multiscale and multiplanar attention branches to increase the spatial context crucial in semantic segmentation tasks. The resulting architecture is the new CASPIANET++, which achieves Dice Scores of 91.19%, 87.6% and 81.03% for whole tumor, tumor core and enhancing tumor respectively. Furthermore, driven by the scarcity of brain tumor data, we investigate the Noisy Student method for segmentation tasks. Our new Noisy Student Curriculum Learning paradigm, which infuses noise incrementally to increase the complexity of the training images exposed to the network, further boosts the enhancing tumor region to 81.53%. Additional validation performed on the BraTS2020 data shows that the Noisy Student Curriculum Learning method works well without any additional training or finetuning.
卷积神经网络(CNN)已非常成功地用于脑肿瘤的语义分割。然而,当前的CNN和注意力机制本质上是随机的,并且忽略了放射科医生用于手动标注感兴趣区域的形态学指标。在本文中,我们通过利用肿瘤的固有结构来检测显著区域,引入了一种通道和空间维度的非对称注意力机制(CASPIAN)。为了证明我们提出的层的有效性,我们将其集成到一个成熟的卷积神经网络(CNN)架构中,以在使用更少GPU资源的情况下获得更高的Dice分数。此外,我们研究了包含辅助多尺度和多平面注意力分支,以增加语义分割任务中至关重要的空间上下文。由此产生的架构是新的CASPIANET++,其在全肿瘤、肿瘤核心和强化肿瘤上分别实现了91.19%、87.6%和81.03%的Dice分数。此外,受脑肿瘤数据稀缺的驱动,我们研究了用于分割任务的Noisy Student方法。我们新的Noisy Student课程学习范式,即逐步注入噪声以增加网络所接触的训练图像的复杂性,进一步将强化肿瘤区域提升至81.53%。在BraTS2020数据上进行的额外验证表明,Noisy Student课程学习方法无需任何额外训练或微调即可良好运行。