Multimodal active subspace analysis for computing assessment oriented subspaces from neuroimaging data.

ABSTRACT

BACKGROUND: For successful biomarker discovery, it is essential to develop computational frameworks that summarize high-dimensional neuroimaging data in terms of involved sub-systems of the brain, while also revealing underlying heterogeneous functional and structural changes covarying with specific cognitive and biological traits. However, unsupervised decompositions do not inculcate clinical assessment information, while supervised approaches extract only individual feature importance, thereby impeding qualitative interpretation at the level of subspaces. NEW METHOD: We present a novel framework to extract robust multimodal brain subspaces associated with changes in a given cognitive or biological trait. Our approach involves active subspace learning on the gradients of a trained machine learning model followed by clustering to extract and summarize the most salient and consistent subspaces associated with the target variable. RESULTS: Through a rigorous cross-validation procedure on an Alzheimer's disease (AD) dataset, our framework successfully extracts multimodal subspaces specific to a given clinical assessment (e.g., memory and other cognitive skills), and also retains predictive performance in standard machine learning algorithms. We also show that the salient active subspace directions occur consistently across randomly sub-sampled repetitions of the analysis. COMPARISON WITH EXISTING METHOD(S) Compared to existing unsupervised decompositions based on principle component analysis, the subspace components in our framework retain higher predictive information. CONCLUSIONS: As an important step towards biomarker discovery, our framework not only uncovers AD-related brain regions in the associated brain subspaces, but also enables automated identification of multiple underlying structural and functional sub-systems of the brain that collectively characterize changes in memory and proficiency in cognitive skills related to brain disorders like AD.