Classification and Diagnosis of Autism Spectrum Disorder Using Swin Transformer.

Heqian Zhang,Zhaohui Wang,Yuefu Zhan
DOI: https://doi.org/10.1109/isbi53787.2023.10230792
2023-01-01
Abstract:Autism spectrum disorder (ASD) is a complex neurodevelopmental disorder. Failure to receive timely treatment may seriously affect a child's physical and mental development. Early diagnosis of autism is crucial for treatment and recovery of normal development. Currently, MRI-assisted diagnosis methods for autism are increasingly available. However, due to the complexity of ASD lesions, it is still difficult to establish robust models. In CNN-based models, the detailed information may be lost due to the operations such as convolution and down-sampling. The potential of Transformer for image recognition makes it possible to overcome these limitations. In this paper, a multi-scale Transformer-based model for ASD classification is proposed. Specifically, a strong baseline based on Transformer is first constructed, and then two modules are carefully designed to enhance robust feature extraction from fMRI. The context-enhanced module is proposed to aggregate multi-scale feature information and enhance feature differentiation. The cross-patch module is introduced to rearrange patches via cross-shifting, and to further capture fine-grained local features. The model was trained and tested by fMRI from Autism Brain Imaging Data Exchange dataset. The result of a 5-fold cross-validation shows that the accuracy reaches 78%. Our method can achieve a better classification performance in the assisted diagnosis of autism than conventional deep learning models.
What problem does this paper attempt to address?