Hype versus hope: Deep learning encodes more predictive and robust brain imaging representations than standard machine learning
Anees AbroliD, Zening FuiD, Mustafa S. SalmaniD, Rogers F. SilvaiD, Yuhui DuiD, Sergey PlisiD, Vince D. CalhouniD
Abstract
Previous successes of deep learning (DL) approaches on several complex tasks have hugely inflated expectations of their power to learn subtle properties of complex brain imaging data, and scale to large datasets. Perhaps as a reaction to this inflation, recent critical commentaries unfavorably compare DL with standard machine learning (SML) approaches for the analysis of brain imaging data. Yet, their conclusions are based on pre-engineered features which deprives DL of its main advantage: representation learning. Here we evaluate this and show the importance of representation learning for DL performance on brain imaging data. We report our findings from a large-scale systematic comparison of SML approaches versus DL profiled in a ten-way age and gender-based classification task on 12,314 structural MRI images. Results show that DL methods, if implemented and trained following the prevalent DL practices, have the potential to substantially improve compared to SML approaches. We also show that DL approaches scale particularly well presenting a lower asymptotic complexity in relative computational time, despite being more complex. Our analysis reveals that the performance improvement saturates as the training sample size grows, but shows significantly higher performance throughout. We also show evidence that the superior performance of DL is primarily due to the excellent representation learning capabilities and that SML methods can perform equally well when operating on representations produced by the trained DL models. Finally, we demonstrate that DL embeddings span a comprehensible projection spectrum and that DL consistently localizes discriminative brain biomarkers, providing an example of the robustness of prediction relevance estimates. Our findings highlight the presence of non-linearities in brain imaging data that DL frameworks can exploit to generate superior predictive representations for characterizing the human brain, even with currently available data sizes.
02 /Specification & Classification
Click any tag to browse matching papers- Citations
- 223
- FWCI
- 19.86
- Field percentile
- top 1%
- Organism
- Assay
- Research Focus
- Approach
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- Deep learning for brains?: Different linear and nonlinear scaling in UK Biobank brain images vs. machine-learning datasets
Marc-André Schulz, B.T. Thomas Yeo, … Danilo Bzdok·Nature CommunicationsPeer-reviewed·Aug 25, 2020
- Beyond functional connectivity: deep learning applied to resting-state fMRI time series in the prediction of 58 human traits in the HCP
Bruno Hebling Vieira, Mikkel Schöttner, … Carlos Ernesto Garrido Salmón·bioRxiv·Mar 7, 2024
- Accurate brain age prediction with lightweight deep neural networks
Han Peng, Weikang Gong, … Stephen M. Smith·Medical Image AnalysisPeer-reviewed·Oct 19, 2020
- Deep Learning for Quality Control of Subcortical Brain 3D Shape Models
Dmitry Petrov, Boris A. Gutman, … Paul M. Thompson·bioRxiv·Sep 4, 2018
- Incremental improvements in tractometry-based brain-age modeling with deep learning
Ariel Rokem, Joanna Qiao, … Adam Richie-Halford·bioRxiv·Mar 2, 2023
- The Promise of Deep Learning for BCIs: Classification of Motor Imagery EEG using Convolutional Neural Network
Navneet Tibrewal, Nikki Leeuwis, Maryam Alimardani·bioRxiv·Jun 18, 2021