Application of TBSS-based machine learning models in the diagnosis of pediatric autism

Abstract

ObjectiveTo explore the microstructural changes of white matter in children with pediatric autism by using diffusion kurtosis imaging (DKI), and evaluate whether the combination of tract-based spatial statistics (TBSS) and back-propagation neural network (BPNN)/support vector machine (SVM)/logistic regression (LR) was feasible for the classification of pediatric autism.MethodsDKI data were retrospectively collected from 32 children with autism and 27 healthy controls (HCs). Kurtosis fractional anisotropy (FAK), mean kurtosis (MK), axial kurtosis (KA), radial kurtosis (RK), fractional anisotropy (FA), axial diffusivity (DA), mean diffusivity (MD) and Radial diffusivity (DR) were generated by iQuant workstation. TBSS was used to detect the regions of parameters values abnormalities and for the comparison between these two groups. In addition, we also introduced the lateralization indices (LI) to study brain lateralization in children with pediatric autism, using TBSS for additional analysis. The parameters values of the differentiated regions from TBSS were then calculated for each participant and used as the features in SVM/BPNN/LR. All models were trained and tested with leave-one-out cross validation (LOOCV).ResultsCompared to the HCs group, the FAK, DA, and KA values of multi-fibers [such as the bilateral superior longitudinal fasciculus (SLF), corticospinal tract (CST) and anterior thalamic radiation (ATR)] were lower in pediatric autism group (p < 0.05, TFCE corrected). And we also found DA lateralization abnormality in Superior longitudinal fasciculus (SLF) (the LI in HCs group was higher than that in pediatric autism group). However, there were no significant differences in FA, MD, MK, DR, and KR values between HCs and pediatric autism group (P > 0.05, TFCE corrected). After performing LOOCV to train and test three model (SVM/BPNN/LR), we found the accuracy of BPNN (accuracy = 86.44%) was higher than that of LR (accuracy = 76.27%), but no different from SVM (RBF, accuracy = 81.36%; linear, accuracy = 84.75%).ConclusionOur proposed method combining TBSS findings with machine learning (LR/SVM/BPNN), was applicable in the classification of pediatric autism with high accuracy. Furthermore, the FAK, DA, and KA values and Lateralization index (LI) value could be used as neuroimaging biomarkers to discriminate the children with pediatric autism or not.