Visualizing Functional Network Connectivity Differences Using an Explainable Machine-learning Method

Abstract

Functional network connectivity (FNC) estimated from resting-state functional magnetic resonance imaging showed great information about the neural mechanism in different brain disorders. But previous research has mainly focused on standard statistical learning approaches to find FNC features separating patients from control. Although machine learning approaches provide better models separating controls from patients, it is not straightforward for these approaches to provide intuition on the model and the underlying neural process of each disorder. Explainable machine learning offers a solution to this problem by applying machine learning to understand the neural process behind brain disorders. In this study, we introduce a novel framework leveraging SHapley Additive exPlanations (SHAP) to identify crucial Functional Network Connectivity (FNC) features distinguishing between two distinct population classes. Initially, we validate our approach using synthetic data. Subsequently, applying our framework, we ascertain FNC biomarkers distinguishing between, controls and schizophrenia patients with accuracy of 81.04% as well as middle aged adults and old aged adults with accuracy 71.38%, respectively, employing Random Forest (RF), XGBoost, and CATBoost models. Our analysis underscores the pivotal role of the cognitive control network (CCN), subcortical network (SCN), and somatomotor network (SMN) in discerning individuals with schizophrenia from controls. In addition, our platform found CCN and SCN as the most important networks separating young adults from older.