Graph-Based Disease Prediction in Neuroimaging: Investigating the Impact of Feature Selection

Abstract

In biomedical machine learning, data often appear in the form of graphs. Biological systems such as protein interactions and ecological or brain networks are instances of applications that benefit from graph representations. Geometric deep learning is an arising field of techniques that has extended deep neural networks to non-Euclidean domains such as graphs. In particular, graph convolutional neural networks have achieved advanced performance in semi-supervised learning in those domains. Over the last years, these methods have gained traction in neuroscience as they could be the key to a deeper understanding in clinical diagnosis at the systems or network level (for an individual brain but also for across a cohort of subjects). As a proof-of-principle, we study and validate a previous implementation of graph-based semi-supervised classification using a ridge classifier and graph convolutional neural networks. The models are trained on population graphs that integrate imaging and phenotypic information. Our analysis employs neuroimaging data of structural and functional connectivity for prediction of neurodevelopmental and neurodegenerative disorders. Here, we particularly study the effect of different strategies to reduce the dimensionality of the neuroimaging features on the graph nodes on the classification performance.