论文标题

基于大脑MRI的3D卷积神经网络用于精神分裂症和对照的分类

Brain MRI-based 3D Convolutional Neural Networks for Classification of Schizophrenia and Controls

论文作者

Hu, Mengjiao, Sim, Kang, Zhou, Juan Helen, Jiang, Xudong, Guan, Cuntai

论文摘要

卷积神经网络(CNN)已成功应用于自然图像和医学图像的分类,但尚未应用于将精神分裂症与健康对照组区分开来。鉴于精神分裂症的微妙,混合和稀疏分布的大脑萎缩模式,自动特征学习的能力使CNN成为从控制中分类精神分裂症的强大工具,因为它可以消除选择相关空间特征的主观性。为了检查基于结构磁共振成像(MRI)将CNN应用于精神分裂症和对照组的可行性,我们构建了具有不同体系结构的3D CNN模型,并使用基于手工特征的机器学习方法将其性能进行了比较。支持向量机(SVM)用作分类器,基于体素的形态计(VBM)用作基于手工特征的机器学习的功能。具有连续体系结构,Inception模块和残留模块的3D CNN模型从头开始训练。与基于手动功能的机器学习相比,CNN模型达到的交叉验证精度更高。此外,在独立数据集上进行测试,3D CNN型号大大优于基于手工的机器学习。这项研究强调了CNN使用3D脑MR图像鉴定精神分裂症患者的潜力,并为基于成像的个人水平诊断和精神疾病的预后铺平了道路。

Convolutional Neural Network (CNN) has been successfully applied on classification of both natural images and medical images but not yet been applied to differentiating patients with schizophrenia from healthy controls. Given the subtle, mixed, and sparsely distributed brain atrophy patterns of schizophrenia, the capability of automatic feature learning makes CNN a powerful tool for classifying schizophrenia from controls as it removes the subjectivity in selecting relevant spatial features. To examine the feasibility of applying CNN to classification of schizophrenia and controls based on structural Magnetic Resonance Imaging (MRI), we built 3D CNN models with different architectures and compared their performance with a handcrafted feature-based machine learning approach. Support vector machine (SVM) was used as classifier and Voxel-based Morphometry (VBM) was used as feature for handcrafted feature-based machine learning. 3D CNN models with sequential architecture, inception module and residual module were trained from scratch. CNN models achieved higher cross-validation accuracy than handcrafted feature-based machine learning. Moreover, testing on an independent dataset, 3D CNN models greatly outperformed handcrafted feature-based machine learning. This study underscored the potential of CNN for identifying patients with schizophrenia using 3D brain MR images and paved the way for imaging-based individual-level diagnosis and prognosis in psychiatric disorders.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源