Trans-ResNet: Integrating Transformers and CNNs for Alzheimer’s Disease Classification

Chao Li,Yue Cui,Na Luo,Yong Liu,Pierrick Bourgeat,Jurgen Fripp,Tianzi Jiang
DOI: https://doi.org/10.1109/isbi52829.2022.9761549
2022-01-01
Abstract:Convolutional neural networks (CNNs) have demonstrated excellent performance for brain disease classification from MRI data. However, CNNs lack the ability to capture global dependencies. The recently proposed architecture called Transformer uses attention mechanisms to match or even outperform CNNs on various vision tasks. Transformer’s performance is dependent on access to large training datasets, but sample sizes for most brain MRI datasets are relatively small. To overcome this limitation, we propose Trans-ResNet, a novel architecture which integrates the advantages of both CNNs and Transformers. In addition, we pre-trained our Trans-ResNet on a large-scale dataset on the task of brain age estimation for higher performance. Using three neuroimaging cohorts (UK Biobank, AIBL, ADNI), we demonstrated that our Trans-ResNet achieved higher classification accuracy on Alzheimer disease prediction compared to other state-of-the-art CNN-based methods.
What problem does this paper attempt to address?