The effects of data leakage on connectome-based machine learning models

• Published in: bioRxiv
• Main Authors: Rosenblatt, Matthew, Tejavibulya, Link, Jiang, Rongtao, Noble, Stephanie, Scheinost, Dustin
• Format: Journal Article Paper
• Language: English
• Published: United States Cold Spring Harbor Laboratory Press 28.12.2023; Cold Spring Harbor Laboratory
• Edition: 1.2
• Subjects: Data integrity; Feature selection; Leakage; Learning algorithms; Machine learning; Medical imaging; Neuroimaging; Neuroscience; Phenotypes; Prediction models; Structure-function relationships
• ISSN: 2692-8205; 2692-8205
• DOI: 10.1101/2023.06.09.544383

Predictive modeling has now become a central technique in neuroimaging to identify complex brain-behavior relationships and test their generalizability to unseen data. However, data leakage, which unintentionally breaches the separation between data used to train and test the model, undermines the validity of predictive models. Previous literature suggests that leakage is generally pervasive in machine learning, but few studies have empirically evaluated the effects of leakage in neuroimaging data. Although leakage is always an incorrect practice, understanding the effects of leakage on neuroimaging predictive models provides insight into the extent to which leakage may affect the literature. Here, we investigated the effects of leakage on machine learning models in two common neuroimaging modalities, functional and structural connectomes. Using over 400 different pipelines spanning four large datasets and three phenotypes, we evaluated five forms of leakage fitting into three broad categories: feature selection, covariate correction, and lack of independence between subjects. As expected, leakage via feature selection and repeated subjects drastically inflated prediction performance. Notably, other forms of leakage had only minor effects (e.g., leaky site correction) or even decreased prediction performance (e.g., leaky covariate regression). In some cases, leakage affected not only prediction performance, but also model coefficients, and thus neurobiological interpretations. Finally, we found that predictive models using small datasets were more sensitive to leakage. Overall, our results illustrate the variable effects of leakage on prediction pipelines and underscore the importance of avoiding data leakage to improve the validity and reproducibility of predictive modeling.