基于3D注意力卷积与自监督学习的脑疾病分类方法

    Brain Disease Classification Method Based on 3D Attention Convolutional Neural Networks and Self-supervised Learning

    • 摘要: 为了提升现有脑疾病分类方法提取三维空间特征的能力,提出一种融合3D注意力卷积与自监督学习的分类模型。首先,提出一种基于残差结构的3D注意力卷积神经网络来提取空间特征,利用3D注意力机制区分体素数据中不同空间位置的重要性;其次,利用空间特征构建一个基于自监督学习的多任务学习框架,通过基于空间连续性的自监督辅助任务来进一步挖掘体素的空间依赖关系;最后,通过辅助任务与目标分类任务的联合训练优化神经网络参数,进而提升分类模型的性能。在ABIDE-Ⅰ和ABIDE-Ⅱ数据集上的实验结果表明,所提方法具有优异的分类性能,分类结果也具备良好的可解释性。

       

      Abstract: To improve the 3D spatial feature extraction ability of current brain disease classification methods, a novel classification method based on 3D attention convolutions and self-supervised learning was proposed. First, a 3D attention convolutional neural network based on residual connections was proposed to extract spatial features, aiming to distinguish the importance of different spatial positions in voxel data using 3D attention mechanism. Second, a multi-task learning framework based on self-supervised learning was constructed by using the learned spatial features, and a self-supervised auxiliary task based on spatial continuity was used to further mine the spatial dependence of voxels. Finally, the model parameters were optimized by jointly training the target classification task and self-supervised auxiliary task, and then the classification performance can be largely boosted. Experimental results on ABIDE-Ⅰ and ABIDE-Ⅱ show that the proposed method has superior classification performance and the results also demonstrate good interpretability.

       

    /

    返回文章
    返回