Transformer models and deep learning for medical image analysis in the early detection of neurological diseases
Abstract
Early detection of Alzheimer's Disease (AD) remains a major challenge in neuroimaging, as structural changes in the brain may be subtle and complex during the early stages of the disease. Although Convolutional Neural Networks (CNNs) have been widely applied to medical image analysis, their ability to model long-range spatial dependencies in complex brain images is inherently limited by their predominantly local receptive fields. This study investigates the application of the Vision Transformer (ViT) architecture to Structural Magnetic Resonance Imaging (sMRI) for AD classification. A transformer-based framework was developed to analyze sMRI images. By leveraging the self-attention mechanism, the proposed approach enables the modeling of relationships between spatially distant image regions and the extraction of spatial and textural features. Particular attention was given to identifying disease-associated structural alterations, including changes in the hippocampus and cerebral cortical regions. The performance of the proposed approach was compared with that of conventional CNN-based methods. The results indicated that the ViT-based deep learning approach achieved better performance than the comparison methods in terms of the evaluation metrics considered, including classification accuracy and sensitivity. Furthermore, performance evaluation under noisy conditions demonstrated the model's ability to maintain its performance across sMRI images of varying quality. These findings suggest that transformer-based architectures, through their self-attention mechanisms, offer a promising approach to extracting structural features from sMRI images and classifying individuals with AD. Further validation on independent datasets is required to establish their generalizability and clinical applicability.
Keywords
Medical imaging Alzheimer's disease Vision transformers Deep learning NeuroimagingReferences (15)
- [1] Scheltens, P., De Strooper, B., Kivipelto, M., Holstege, H., Chételat, G., Teunissen, C. E., ... & Van der Flier, W. M. (2021). Alzheimer’s disease. The Lancet, 397(10284), 1577–1590. https://doi.org/10.1016/S0140-6736(20)32205-4
- [2] Gloss, D. (2017). Practice guideline update summary: Mild cognitive impairment: Report of the Guideline Development, Dissemination, and Implementation Subcommittee of the American Academy of Neurology. Neurology, 90(3), 126–135. https://doi.org/10.1212/WNL.0000000000004826
- [3] Litjens, G., Kooi, T., Bejnordi, B. E., Setio, A. A. A., Ciompi, F., Ghafoorian, M., ... & Sánchez, C. I. (2017). A survey on deep learning in medical image analysis. Medical Image Analysis, 42, 60–88. https://doi.org/10.1016/j.media.2017.07.005
- [4] LeCun, Y., Bengio, Y., & Hinton, G. (2015). Deep learning. Nature, 521(7553), 436–444. https://doi.org/10.1038/nature14539
- [5] Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., … & Houlsby, N. (2020). An image is worth 16x16 words: transformers for image recognition at scale [presentation]. International conference on learning representations (ICLR). https://welzin.ai/papers/vit.pdf
- [6] Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A. N., ... & Polosukhin, I. (2017). Attention is all you need [presentation]. Advances in neural information processing systems (Vol. 30, pp. 5998–6008). https://proceedings.neurips.cc/paper_files/paper/2017/hash/3f5ee243547dee91fbd053c1c4a845aa-Abstract.html
- [7] Ronneberger, O., Fischer, P., & Brox, T. (2015). U-net: Convolutional networks for biomedical image segmentation. International conference on medical image computing and computer-assisted intervention (pp. 234–241). Springer. https://doi.org/10.1007/978-3-319-24574-4_28
- [8] Wen, J., Thibeau-Sutre, E., Diaz-Melo, M., Samper-González, J., Routier, A., … & Colliot, O. (2020). Convolutional neural networks for classification of Alzheimer’s disease: Overview and reproducible evaluation. Medical Image Analysis, 63, 101694. https://doi.org/10.1016/j.media.2020.101694
- [9] Liu, Z., Lin, Y., Cao, Y., Hu, H., Wei, Y., Zhang, Z., ... & Guo, B. (2021). Swin transformer: hierarchical vision transformer using shifted windows. 2021 IEEE/cvf international conference on computer vision (ICCV) (pp. 9992–10002). IEEE. https://doi.org/10.1109/ICCV48922.2021.00986
- [10] Jack Jr, C. R., Bernstein, M. A., Fox, N. C., Thompson, P., Alexander, G., Harvey, D., ... & Weiner, M. W. (2008). The Alzheimer’s disease neuroimaging initiative (ADNI): MRI methods. Journal of Magnetic Resonance Imaging: an Official Journal of the International Society for Magnetic Resonance in Medicine, 27(4), 685–691. https://doi.org/10.1002/jmri.21049
- [11] Tustison, N. J., Avants, B. B., Cook, P. A., Zheng, Y., Egan, A., Yushkevich, P. A., & Gee, J. C. (2010). N4ITK: Improved N3 bias correction. IEEE Transactions on Medical Imaging, 29(6), 1310–1320. https://doi.org/10.1109/TMI.2010.2046908
- [12] He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep residual learning for image recognition. 2016 IEEE conference on computer vision and pattern recognition (CVPR) (pp. 770–778). IEEE https://doi.org/10.1109/CVPR.2016.90
- [13] Frisoni, G. B., Fox, N. C., Jack Jr, C. R., Scheltens, P., & Thompson, P. M. (2010). The clinical use of structural MRI in Alzheimer disease. Nature Reviews Neurology, 6(2), 67–77. https://doi.org/10.1038/nrneurol.2009.215?utm_source=chatgpt.com
- [14] Basaia, S., Agosta, F., Wagner, L., Canu, E., Magnani, G., Santangelo, R., ... & Filippi, M. (2019). Automated classification of Alzheimer’s disease and mild cognitive impairment using a single MRI and deep neural networks. NeuroImage: Clinical, 21, 101645. https://doi.org/10.1016/j.nicl.2018.101645
- [15] Liu, M., Zhang, J., Adeli, E., & Shen, D. (2018). Landmark-based deep multi-instance learning for brain disease diagnosis. Medical Image Analysis, 43, 157–168. https://doi.org/10.1016/j.media.2017.10.005
