RepBoTNet-CESA:An Alzheimer’s Disease Computer Aided Diagnosis Method Using Structural Reparameterization BoTNet and Cubic Embedding Self Attention  

作　　者：Xiabin Zhang Zhongyi Hu Lei Xiao Hui Huang 

机构地区：[1]College of Computer Science and Artificial Intelligence,Wenzhou University,Wenzhou,325035,China [2]Key Laboratory of Intelligence Image Processing and Analysis,Wenzhou University,Wenzhou,325035,China

出　　处：《Computers, Materials & Continua》2024年第5期2879-2905,共27页计算机、材料和连续体（英文）

基　　金：the Key Project of Zhejiang Provincial Natural Science Foundation under Grants LD21F020001,Z20F020022;the National Natural Science Foundation of China under Grants 62072340,62076185;the Major Project of Wenzhou Natural Science Foundation under Grants 2021HZSY0071,ZS2022001.

摘　　要：Various deep learning models have been proposed for the accurate assisted diagnosis of early-stage Alzheimer’s disease(AD).Most studies predominantly employ Convolutional Neural Networks(CNNs),which focus solely on local features,thus encountering difficulties in handling global features.In contrast to natural images,Structural Magnetic Resonance Imaging(sMRI)images exhibit a higher number of channel dimensions.However,during the Position Embedding stage ofMulti Head Self Attention(MHSA),the coded information related to the channel dimension is disregarded.To tackle these issues,we propose theRepBoTNet-CESA network,an advanced AD-aided diagnostic model that is capable of learning local and global features simultaneously.It combines the advantages of CNN networks in capturing local information and Transformer networks in integrating global information,reducing computational costs while achieving excellent classification performance.Moreover,it uses the Cubic Embedding Self Attention(CESA)proposed in this paper to incorporate the channel code information,enhancing the classification performance within the Transformer structure.Finally,the RepBoTNet-CESA performs well in various AD-aided diagnosis tasks,with an accuracy of 96.58%,precision of 97.26%,and recall of 96.23%in the AD/NC task;an accuracy of 92.75%,precision of 92.84%,and recall of 93.18%in the EMCI/NC task;and an accuracy of 80.97%,precision of 83.86%,and recall of 80.91%in the AD/EMCI/LMCI/NC task.This demonstrates that RepBoTNet-CESA delivers outstanding outcomes in various AD-aided diagnostic tasks.Furthermore,our study has shown that MHSA exhibits superior performance compared to conventional attention mechanisms in enhancing ResNet performance.Besides,the Deeper RepBoTNet-CESA network fails to make further progress in AD-aided diagnostic tasks.

关 键 词：Alzheimer CNN structural reparameterization multi head self attention computer aided diagnosis 

分 类 号：R749.16[医药卫生—神经病学与精神病学]