Different scaling of linear models and deep learning in UKBiobank brain images versus machine-learning datasets

Research output: Contribution to journalResearch articleContributedpeer-review

Contributors

  • Marc-Andre Schulz - , University Hospital Aachen (Author)
  • B T Thomas Yeo - , Nanyang Technological University (Author)
  • Joshua T Vogelstein - , Johns Hopkins University (Author)
  • Janaina Mourao-Miranada - , University College London (Author)
  • Jakob N Kather - , Else Kröner Fresenius Center for Digital Health, University Hospital Aachen, German Cancer Research Center (DKFZ), German Cancer Consortium (DKTK) Partner Site Dresden (Author)
  • Konrad Kording - , University of Pennsylvania (Author)
  • Blake Richards - , McGill University (Author)
  • Danilo Bzdok - , Mila - Quebec Artificial Intelligence Institute (Author)

Abstract

Recently, deep learning has unlocked unprecedented success in various domains, especially using images, text, and speech. However, deep learning is only beneficial if the data have nonlinear relationships and if they are exploitable at available sample sizes. We systematically profiled the performance of deep, kernel, and linear models as a function of sample size on UKBiobank brain images against established machine learning references. On MNIST and Zalando Fashion, prediction accuracy consistently improves when escalating from linear models to shallow-nonlinear models, and further improves with deep-nonlinear models. In contrast, using structural or functional brain scans, simple linear models perform on par with more complex, highly parameterized models in age/sex prediction across increasing sample sizes. In sum, linear models keep improving as the sample size approaches ~10,000 subjects. Yet, nonlinearities for predicting common phenotypes from typical brain scans remain largely inaccessible to the examined kernel and deep learning methods.

Details

Original languageEnglish
Article number4238
JournalNature communications
Volume11
Publication statusPublished - 25 Aug 2020
Peer-reviewedYes

External IDs

PubMedCentral PMC7447816
Scopus 85089774011

Keywords

Keywords

  • Biological Specimen Banks, Brain/diagnostic imaging, Deep Learning, Humans, Linear Models, Machine Learning, Neuroimaging/methods, Phenotype, Sample Size, United Kingdom