《计算机应用》唯一官方网站

• •    下一篇

轻量级多尺度卷积网络的功能磁共振成像脑龄预测模型

沈嫣然1,温昕1,张瑾昊1,张帅1,曹锐2,高保禄3   

  1. 1. 太原理工大学 软件学院
    2. 太原理工大学软件学院
    3. 太原理工大学
  • 收稿日期:2023-12-20 修回日期:2024-04-03 发布日期:2024-04-28 出版日期:2024-04-28
  • 通讯作者: 高保禄
  • 基金资助:
    基于脑老化的多中心功能磁共振分析方法研究;神经影像大数据功能指纹挖掘及模型可解释性研究

fMRI brain age prediction model with lightweight multi-scale convolutional network

  • Received:2023-12-20 Revised:2024-04-03 Online:2024-04-28 Published:2024-04-28

摘要: 摘 要: 针对功能磁共振成像(fMRI)脑龄预测精度较低、以及该问题与深度学习结合研究较少的现状,提出了一种轻量级多尺度卷积网络的功能磁共振成像脑龄预测模型(LMCN)。首先,通过计算fMRI中感兴趣区域(ROI)的皮尔逊相关系数得到ROI的功能连接矩阵(FC),作为输入;其次,提升FC通道数保证特征数量的同时缩小特征图尺寸,采用具有人类视觉注意力特点的多尺度空洞卷积模块(RFB)对年龄特征进行提取;最后,通过全连接层输出预测脑龄,并且计算各脑区的消融预测结果,探索对脑龄预测结果产生影响的关键脑区。在E-NKI和Cam-CAN两个公开数据集上进行评估,LMCN参数所需内存为2.30MB,比MobileNetV3、ShuffleNetV2减少60.3%、43.5%。预测结果上,在E-NKI数据集,平均绝对误差(MAE)为5.16,皮尔逊相关系数(R)为0.947,均方根误差(RMSE)为6.40,与基于网络的特征选择结合最小角回归模型相比,MAE减少1.34,R增加0.037;在Cam-CAN数据集,MAE为5.97,R为0.904,RMSE为7.93,与基于连接组的机器学习模型相比,R提升0.019,RMSE减少0.64。结果表明,LMCN在参数量较小易于部署的同时,能够有效提高fMRI脑龄预测精度,为评估健康成人脑部状态提供线索。

关键词: 关键词: 功能磁共振成像, 功能连接, 轻量级神经网络, 脑龄预测, 深度学习

Abstract: Abstract: In view of the low accuracy of functional Magnetic Resonance Imaging (fMRI) brain age prediction and the lack of research on the combination of this problem and deep learning, a fMRI brain age prediction model with Lightweight Multi-scale Convolutional Network (LMCN) was proposed. First, the Pearson correlation coefficient of the region of interest (ROI) in fMRI was calculated to obtain the functional connectivity matrix (FC) of the ROI as input; secondly, the number of FC channels was increased to ensure the number of features while reducing the size of the feature map and the multi-scale dilated convolution module (RFB) with the characteristics of human visual attention was used to extract age features; finally, the brain age was predicted through the output of the fully connected layer, and the ablation prediction results of each brain area were calculated to explore the key brain regions affected by the generation of brain age prediction results. Evaluated on two public data sets, E-NKI and Cam-CAN, the memory required for LMCN parameters is 2.30MB, which is 60.3% and 43.5% less than MobileNetV3 and ShuffleNetV2. In terms of prediction results, in the E-NKI data set, the Mean Absolute Error (MAE) is 5.16, the Pearson correlation coefficient (R) is 0.947, and the Root Mean Square Error (RMSE) is 6.40, which is comparable to network-based feature selection combined with the minimum angle regression model. Compared with the method based on the connectome, MAE decreased by 1.34 and R increased by 0.037; in the Cam-CAN data set, MAE was 5.97, R was 0.904, and RMSE was 7.93. Compared with the connectome-based model, R increased by 0.019 and RMSE decreased by 0.64. The results show that while the LMCN has small parameters and is easy to deploy, it can effectively improve the accuracy of fMRI brain age prediction and provide clues for assessing the brain status of healthy adults.

Key words: Keywords: functional Magnetic Resonance Imaging(fMRI), functional connectivity, lightweight neural network, brain age prediction, deep learning

中图分类号: