Abstract
Purpose To investigate multivariate regional patterns for schizophrenia (SZ) classification, sex differences, and brain age by utilizing structural MRI, demographics, and explainable artificial intelligence (AI).
Methods Various AI models were employed, and the outperforming model was identified for SZ classification, sex differences, and brain age predictions. For the SZ and sex classification tasks, support vector classifier (SVC), k-nearest neighbor (KNN), and deep learning neural network (DL) models were compared. In the case of regression-based brain age prediction, Lasso regression (LR), Ridge regression (RR), support vector regression (SVR), and DL models were compared. For each regression or classification task, the optimal model was further integrated with the Shapley additive explanations (SHAP) and the significant multivariate brain regional patterns were identified.
Results Our results demonstrated that the DL model outperformed other models in SZ classification, sex differences, and brain age predictions. We then integrated outperforming DL model with SHAP, and this integrated DL-SHAP was used to identify the individualized multivariate regional patterns associated with each prediction. Using DL-SHAP approach, we found that individuals with SZ had anatomical changes particularly in left pallidum, left posterior insula, left hippocampus, and left putamen regions, and such changes associated with SZ were different between female and male patients. Finally, we further applied DL-SHAP method to brain age prediction and suggested important brain regions related to aging in health controls (HC) and SZ processes.
Conclusion This study systematically utilized predictive modeling and novel explainable AI approaches and identified the complex multivariate brain regions involved with SZ classification, sex differences, and brain aging and built a deeper understanding of neurobiological mechanisms involved in the disease, offering new insights to future SZ diagnosis and treatments and laying the foundation of the development of precision medicine.
Competing Interest Statement
The authors have declared no competing interest.
Funding Statement
Funding GBC is supported by the Mallinckrodt Institute of Radiology from Washington University in St. Louis as well as by the National Institutes of Health K01AG083230.
Author Declarations
I confirm all relevant ethical guidelines have been followed, and any necessary IRB and/or ethics committee approvals have been obtained.
Yes
The details of the IRB/oversight body that provided approval or exemption for the research described are given below:
The study used (or will use) ONLY openly available human data that were originally located at: Schizconnect, including data from following projects: Center for Biomedical Research Excellence (COBRE), Neuromorphometry by Computer Algorithm Chicago (NMorphCH), and function Biomedical Informatics Research Network (fBIRN) PhaseII__0010. Schizconnect data collection and sharing for this project was funded by NIMH cooperative agreement 1U01 MH097435. Cobre data was downloaded from the COllaborative Informatics and Neuroimaging Suite Data Exchange tool (COINS; http://coins.mrn.org/dx) and data collection was performed at the Mind Research Network and funded by a Center of Biomedical Research Excellence (COBRE) grant 5P20RR021938/P20GM103472 from the NIH to Dr. Vince Calhoun. The fBIRN data used for this study were downloaded from the Function BIRN Data Repository (http://fbirnbdr.birncommunity.org:8080/BDR/), supported by grants to the Function BIRN (U24-RR021992) Testbed funded by the National Center for Research Resources at the National Institutes of Health, U.S.A. NMorhCH data used in preparation of this article were obtained from the Neuromorphometry by Computer Algorithm Chicago (NMorphCH) dataset (http://nunda.northwestern.edu/nunda/data/projects/NMorphCH). As such, the investigators within NMorphCH contributed to the design and implementation of NMorphCH and/or provided data but did not participate in analysis or writing of this report and data collection and sharing for this project was funded by NIMH grant R01 MH056584.
I confirm that all necessary patient/participant consent has been obtained and the appropriate institutional forms have been archived, and that any patient/participant/sample identifiers included were not known to anyone (e.g., hospital staff, patients or participants themselves) outside the research group so cannot be used to identify individuals.
Yes
I understand that all clinical trials and any other prospective interventional studies must be registered with an ICMJE-approved registry, such as ClinicalTrials.gov. I confirm that any such study reported in the manuscript has been registered and the trial registration ID is provided (note: if posting a prospective study registered retrospectively, please provide a statement in the trial ID field explaining why the study was not registered in advance).
Yes
I have followed all appropriate research reporting guidelines, such as any relevant EQUATOR Network research reporting checklist(s) and other pertinent material, if applicable.
Yes
Footnotes
List of abbreviations
- (SZ)
- Schizophrenia
- (AI)
- Artificial intelligence
- (SVC)
- Support vector classifier
- (KNN)
- K-nearest neighbor
- (DL)
- Deep learning neural network
- (LR)
- Lasso regression
- (RR)
- Ridge regression
- (SVR)
- Support vector regression
- (SHAP)
- Shapley additive explanations
- (HC)
- Healthy controls
- (ML)
- Machine learning
- (MRI)
- Magnetic resonance imaging
- (MUSE)
- MUlti-atlas region Segmentation utilizing Ensembles of registration algorithms and parameters and locally optimal atlas selection
- (ROIs)
- Regions of interest
- (RBF)
- Radial basis function
- (ReLU)
- Rectified linear unit
- (CV)
- Cross-validation
- (BCE)
- Binary cross entropy
- (MAE)
- Mean absolute error.
The Chan Zuckerberg Initiative, Cold Spring Harbor Laboratory, the Sergey Brin Family Foundation, California Institute of Technology, Centre National de la Recherche Scientifique, Fred Hutchinson Cancer Center, Imperial College London, Massachusetts Institute of Technology, Stanford University, University of Washington, and Vrije Universiteit Amsterdam.