Combining Radiomics and Machine Learning Approaches for Objective ASD Diagnosis: Verifying White Matter Associations with ASD

Junlin Song,Yuzhuo Chen,Yuan Yao,Zetong Chen,Renhao Guo,Lida Yang,Xinyi Sui,Qihang Wang,Xijiao Li,Aihua Cao,Wei Li
2024-05-25
Abstract:utism Spectrum Disorder is a condition characterized by a typical brain development leading to impairments in social skills, communication abilities, repetitive behaviors, and sensory processing. There have been many studies combining brain MRI images with machine learning algorithms to achieve objective diagnosis of autism, but the correlation between white matter and autism has not been fully utilized. To address this gap, we develop a computer-aided diagnostic model focusing on white matter regions in brain MRI by employing radiomics and machine learning methods. This study introduced a MultiUNet model for segmenting white matter, leveraging the UNet architecture and utilizing manually segmented MRI images as the training data. Subsequently, we extracted white matter features using the Pyradiomics toolkit and applied different machine learning models such as Support Vector Machine, Random Forest, Logistic Regression, and K-Nearest Neighbors to predict autism. The prediction sets all exceeded 80% accuracy. Additionally, we employed Convolutional Neural Network to analyze segmented white matter images, achieving a prediction accuracy of 86.84%. Notably, Support Vector Machine demonstrated the highest prediction accuracy at 89.47%. These findings not only underscore the efficacy of the models but also establish a link between white matter abnormalities and autism. Our study contributes to a comprehensive evaluation of various diagnostic models for autism and introduces a computer-aided diagnostic algorithm for early and objective autism diagnosis based on MRI white matter regions.
Image and Video Processing,Computer Vision and Pattern Recognition,Machine Learning,Quantitative Methods
What problem does this paper attempt to address?
### Problems the Paper Aims to Solve This paper aims to achieve objective diagnosis of Autism Spectrum Disorder (ASD) by combining radiomics and machine learning methods, and to validate the association between white matter and ASD. #### Research Background - **Autism Spectrum Disorder (ASD)**: A disorder characterized by deficits in social skills, communication difficulties, repetitive behaviors, and sensory processing issues. - **Limitations of Traditional Diagnostic Methods**: Currently, diagnosis mainly relies on the qualitative assessment based on the Diagnostic and Statistical Manual of Mental Disorders, Fifth Edition (DSM-5) criteria, which may not fully capture the complexity of ASD and could miss the optimal treatment window. #### Research Objectives - **Develop a Computer-Aided Diagnostic Model**: Utilize segmentation and feature extraction of white matter regions in MRI images, combined with various machine learning algorithms (such as Support Vector Machine, Random Forest, Logistic Regression, K-Nearest Neighbors, etc.) for ASD prediction. - **Validate the Relationship Between White Matter Abnormalities and ASD**: Explore the impact of white matter abnormalities on ASD diagnosis by training different classification models and select the optimal model for early and objective ASD diagnosis. #### Method Overview - **Data Acquisition and Preprocessing**: Collected MRI data from 62 healthy children and 85 ASD patients from Qilu Hospital, and performed preprocessing steps such as noise removal, image registration, grayscale normalization, and histogram equalization. - **Segmentation Model Establishment**: Proposed the MultiUNet model, composed of three UNet models, each responsible for segmenting different parts of thick-slice MRI images, thereby improving segmentation accuracy. - **Feature Extraction and Classification Model Training**: Used the Pyradiomics toolkit to extract white matter features from MRI images and input these features into various machine learning models such as SVM, RF, KNN, LR for training and testing. - **Results Evaluation**: Evaluated the performance of different models using metrics such as cross-validation and ROC curves. The results showed that the SVM model performed best in predicting ASD, with an AUC of 0.89 and an accuracy of 89.47%. #### Main Contributions - **Proposed a New Segmentation Model**: The MultiUNet model performed excellently in segmenting white matter regions in thick-slice MRI images, achieving an IoU value of 0.92. - **Validated the Correlation Between White Matter and ASD**: Confirmed the effectiveness of white matter features for ASD diagnosis through comparisons of various machine learning models. - **Provided an Objective Diagnostic Tool**: Developed a computer-aided diagnostic system based on MRI white matter features, which helps improve the accuracy of early ASD diagnosis.