Aug, 2023

一种 L2 正则化空间注意力网络用于准确快速分类 2D T1 加权 CE-MRI 图像中的脑肿瘤

TL;DR我们提出了一个准确快速的分类网络,用于 MRI 图像中的脑肿瘤分类,其在准确性方面优于所有轻量级方法。我们在包含三种类型脑肿瘤(脑膜瘤、胶质瘤和垂体瘤)的具有挑战性的 2D T1 加权 CE-MRI 数据集上测试了我们的模型。我们引入了一个 l2 归一化的空间注意力机制,在训练过程中充当了对抗过拟合的正则化器。我们将我们的结果与该数据集上的最新技术进行了比较,并显示通过将 l2 归一化的空间注意力集成到基线网络中,我们实现了 1.79 个百分点的性能提升。通过将我们的模型与预训练的 VGG16 集成在一起,可以实现更好的准确性,但会牺牲执行速度。我们的代码可在此 https URL 上公开访问。