Skip to main navigation Skip to search Skip to main content

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

  • Marc Andre Schulz
  • , B. T.Thomas Yeo
  • , Joshua T. Vogelstein
  • , Janaina Mourao-Miranada
  • , Jakob N. Kather
  • , Konrad Kording
  • , Blake Richards
  • , Danilo Bzdok

Research output: Contribution to journal › Article › peer-review

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.

Original languageEnglish (US)
Article number4238
JournalNature communications
Volume11
Issue number1
DOIs
StatePublished - Dec 1 2020

ASJC Scopus subject areas

  • General Chemistry
  • General Biochemistry, Genetics and Molecular Biology
  • General Physics and Astronomy

Fingerprint

Dive into the research topics of 'Different scaling of linear models and deep learning in UKBiobank brain images versus machine-learning datasets'. Together they form a unique fingerprint.

Cite this