1. Introduction
Autism spectrum disorder (ASD) is a group of neurodevelopmental disorders characterized by difficulties in social interactions, language delay, and repetitive behaviors. The World Health Organization (WHO) reports that ASD affects 1 in 160 children worldwide [
1]. Moreover, the severe deficits associated with ASD place a significant health and financial burden on the global community [
2]. Given the increasing prevalence of ASD, it is important to further develop ASD diagnostic tools to reduce the impact of this burdens and to better manage ASD subjects. In the Diagnostic and Statistical Manual of Mental Disorders (DSM), the American Psychological Association (APA) has classified ASD into three subtypes based on impairment symptoms: classic autistic disorder (ASD), Asperger syndrome (APD) and pervasive developmental disorder—not otherwise specified (PDD-NOS) [
3]. The different impairment ratings for ASD subtypes are based on repetitive behaviors, verbal skills, social interaction and communication. Diagnosis of autism is a difficult task because there is no standard medical test for accurate diagnosis [
4]. Current clinical practice uses various questionnaires based on cognitive characteristics and behavioral observations for diagnosis of ASD [
5,
6,
7,
8]. The disadvantages of these clinical assessments, which require direct interaction between the child and the clinician, are that they are time-consuming and costly. Because symptom-based diagnostic criteria depend on observational techniques and subjective decisions, autism researchers point out that diagnoses of ASD types vary widely from clinic to clinic, even when they are all based on the same standard tests [
8]. Because clinical results are imprecise, there is a need to find accurate biological markers to automate the ASD diagnostic process [
9]. Automated solutions based on artificial intelligence (AI) can enable rapid ASD classification, hence increasing the reliability and accuracy of diagnostic results [
10,
11]. Experts have recently developed many AI algorithms for ASD diagnostic models based on functional magnetic resonance imaging (fMRI). Previous works have shown that resting-state functional magnetic resonance imaging (rs-fMRI) data play an essential role in diagnosing ASD. Because rs-fMRI has better spatial resolution, it enables the functional analysis of deep brain structures [
11]. In particular, rs-fMRI is commonly used to study functional connectivity (FC) of the brain at rest by detecting fluctuations in blood oxygenation-dependent (BOLD) signals. Essentially, FC identifies the spatio–temporal correlations between brain regions based on BOLD signals [
12]. The different patterns of FC in ASD are mainly used with AI algorithms to create ASD classifiers that distinguish ASD from normal cases (NC) [
13,
14,
15].
2. Related Work
In the last 10 years, rs-fMRI techniques have been used in the study of brain activity for diagnosis of ASD using machine learning and deep learning algorithms [
16,
17,
18].
Table A1 in the
Appendix A presents the latest AI technologies in ASD classification using rs-fMRI data from the Autism Brain Imaging Data Exchange (ABIDE). As shown in
Table A1, support vectors machines (SVMs) are a traditional machine learning classifier that has been widely used in previous studies. Chen et al. [
19] proposed a discriminative model using an SVM to classify the selected FC features based on the F-score method. With 240 ASD and 128 NC subjects, the data were collected from six different sites of ABIDE, producing 79.17% classification accuracy based on 10-fold cross validation (CV). Bernas et al. [
20] also used SVM for classification of ASD, taking in-phase synchronization features of the FC network extracted from 30 subjects from one of the ABIDE sites: the LEUVEN dataset. Their classification technique achieved an average accuracy of 86.7% using 30-fold CV. Recently, Ma et al. [
21] extracted the phase synchrony of the FC network and used principal component analysis (PCA) to reduce the dimensionality of the FC network by selecting the best FC features as the feature vector for the SVM classifier. Using a dataset of 90 subjects, the classifier achieved 78.9% accuracy in ASD classification from NC run on a 10-fold CV framework.
As shown above, the selection of discriminative FC features in fMRI data is crucial for good performance of discriminative models for ASD detection. On the other hand, deep learning (DL) algorithms based on FC networks of rs-fMRI data have been used to classify ASD. Heinsfeld et al. [
16] concluded that DL algorithms should use unsupervised methods to extract relevant features while minimizing human intervention. They transferred 19,900 features from the FC network to a deep neural network (DNN) using two stacked denoising auto-encoders and achieved an average accuracy of 70%. Wang et al. [
22] achieved 93.2% accuracy by using SVM with recursive feature elimination (RFE) to select the top-ranked FC features from an FC network; their results were based on 10-fold CV among the full ABIDE dataset. Sherkatghanad et al. [
17] attempted to improve the automated ASD classifier by converting the vector of FC features to two-dimensional (2D) matrices and using the images as input to a convolutional neural network (CNN). Their proposed model achieved an average accuracy of 70.2% based on 10-fold CV. In the same context, Hunag et al. [
18] achieved 76.4% average classification accuracy using 2D images of selected FC features as input to a deep belief network (DBN). Huang et al. [
18] first filtered the FC network using a heuristic graph-based feature selection method that considered both external and internal FC network measurements. Subah et al. [
23] achieved 87.9% ASD classification accuracy by training a DNN using the FC features of rs-fMRI data from ABIDE. Based on the quality assessment of the fMRI data, the authors selected 866 patients from a total number of 1035, including 402 ASD and 464 NC subjects.
It is worth mentioning that the brain network pattern FC plays a key role in the performance of ASD classification models. As can be seen from
Table A1, there are two main patterns of FC: static and dynamic. To determine the static FC network, the Pearson correlation between the BOLD signals of the brain nodes is used. Most studies that have used rs-fMRI data to classify ASD utilized static FC [
16,
17,
18]. On the other hand, dynamic connections can be obtained by representing the FC in the time–frequency domain, resulting in more informative connectivity features.
For example, Chen et al. [
19] examined the resting-state FC in ASD over two frequency bands: the slow-4 (0.01–0.027 Hz) and slow-5 (0.027–0.073 Hz). Bernas et al. [
20] attempted to study FC networks based on in-phase synchronization of coherence between signals, and Ma et al. [
21] determined the dynamic correlation between brain nodes based on phase synchrony coefficients. Exciting new work has shown that viewing the brain FC as dynamic over time and frequency can successfully reveal the disruptions of the normal human brain in a disordered state [
24,
25].
At present, the accuracy of ASD classification models based on multiple ABIDE training datasets ranges from 70–93% over the full ABIDE database. However, despite the increasing number of automated classification models, it remains a challenge to find discriminative models that provide superior accuracy with low false prediction in FC-based ASD diagnosis. Clearly, inaccurate predictions can have a negative impact on a patient’s life and even cause financial costs for healthcare institutions [
26,
27]. For example, if a model misclassifies an ASD patient as a normal case or misidentifies ASD subtypes, the disorder may go untreated and may worsen the patient’s impairment symptoms. In this case, there can be serious consequences for the reputation and performance of health facilities.
On the premise of ensuring a robust and efficient ASD diagnostic model, this study proposes a new dynamic FC as an ASD biomarker for more accurate classification of ASD from NC. Evidence suggests that the temporal dynamics of FC is a key feature in identifying brain disorders [
20,
21].
Therefore, we proposed a new metric called wavelet coherence fluctuations (WCF), which represents the amplitude of coherence between brain regions during low-frequency fluctuations using the wavelet coherence transform. The functional brain network is constructed based on WCF, and biological ASD markers are identified by using a variance analysis-support vector machine (ANOVA-SVM) method. Then, the wavelet coherence plus singular value decomposition is implemented to generate a 2D matrix representing the coherence of the ASD biomarkers based on the pure time–frequency components. This matrix is known as the principal wavelet coherence (PWC) connectivity. By using SVD, the useful properties of the WC matrix in classifying ASD vs. NC can be extracted. Then, the PWC matrix is converted into a 2D image. Finally, a three-layer CNN (3L-CNN) is used as an AI algorithm to examine the performance of the proposed framework in identifying ASD patients using PWC images.
Three main sections of the paper are presented as follows: the methodology in
Section 3 explains the complete ASD classification methods based on the principal subspace of dynamic functional connectivity, including the WCF calculation, the ANOVA-SVM algorithm, the PWC generation and the proposed CNN models.
Section 4 presents the results and discussion, and lastly,
Section 5 provides the conclusion and future works.
5. Conclusions and Future Works
In this paper, an ASD classification framework comprising principal wavelet coherence (PWC) and a 3L-CNN is developed and thoroughly evaluated. In essence, PWC connectivity provides the neuronal biomarkers that are input to the 3L-CNN, which extracts and classifies features. The advantage of PWC to represent dynamic connectivity lies in the fact that it provides both time and frequency domain features that carry critical characteristic for discriminating ASD from NC subjects. Further, PWC is fundamentally the most discriminative feature of WC for ASD classification, as determined by a three-stage process: First, by ANOVA-SVM, providing 212 FCs, which were further distilled using correlation analysis between WCF and SRS scores, resulting in 11 FC. Lastly, application of SVD on the 11 FCs extracted the first principal subspace that sensed the maximal orientation energy of the PWC images. In other words, the input images of the 3L-CNN carry the most significant time–frequency information of the PWC and can be used as the biomarker for diagnosis of ASD. Evaluation of the proposed framework was conducted using the full ABIDE dataset, which includes 1035 individuals. PWC + 3LCNN achieved classification with the highest accuracy of 95.2% with a low error rate (4.8%). Comparison with state-of-the-art ASD classification models show the PWC method performs better than recently developed approaches.
Although PWC is promising for using fMRI features for ASD subtype diagnosis, it still has space for improvement. Furthermore, there is evidence that ASD and ADHD have overlapping symptoms, such as social withdrawal and communication difficulties [
25,
49]. ADHD is defined by a chronic pattern of inattention and/or hyperactivity and impulsivity that interferes with development. The overlap of symptoms in different brain disorders makes clinical diagnosis challenging, and classification models with multiple classes are needed for future studies of neurodevelopmental disorders. Future research on PWC images using CNN models could lead to a method for diagnosing ASD individuals using rs-fMRI data. To improve performance in multi-class classification, the use of different brain atlases, such as Craddock (CC200, CC400), that extract more information from the BOLD signals could be considered. Moreover, PWC images can also be trained and tested on other CNN architectures such as residual or inception blocks to achieve better classification of ASD subtypes. In addition, the proposed technique sheds light on computer-aided diagnosis of other psychological problems such as depression, schizophrenia and early Alzheimer’s disease.