From YouTube to the brain: Transfer learning can improve brain-imaging predictions with deep learning.

Nahiyan Malik, Danilo Bzdok
Author Information
  1. Nahiyan Malik: School of Computer Science, McGill University, Montreal, QC, Canada; Mila - Quebec Artificial Intelligence Institute, Montreal, QC, Canada. Electronic address: nahiyan.malik@mail.mcgill.ca.
  2. Danilo Bzdok: School of Computer Science, McGill University, Montreal, QC, Canada; Mila - Quebec Artificial Intelligence Institute, Montreal, QC, Canada; McConnell Brain Imaging Centre, Montreal Neurological Institute (MNI), McGill University, Montreal, QC, Canada; Department of Biomedical Engineering, Faculty of Medicine, McGill University, Montreal, QC, Canada. Electronic address: danilo.bzdok@mcgill.ca.

Abstract

Deep learning has recently achieved best-in-class performance in several fields, including biomedical domains such as X-ray images. Yet, data scarcity poses a strict limit on training successful deep learning systems in many, if not most, biomedical applications, including those involving brain images. In this study, we translate state-of-the-art transfer learning techniques for single-subject prediction of simpler (sex and age) and more complex phenotypes (number of people in household, household income, fluid intelligence and smoking behavior). We fine-tuned 2D and 3D ResNet-18 convolutional neural networks for target phenotype predictions from brain images of ∼40,000 UK Biobank participants, after pretraining on YouTube videos from the Kinetics dataset and natural images from the ImageNet dataset. Transfer learning was effective on several phenotypes, especially sex and age classification. Additionally, transfer learning in particular outperformed deep learning models trained from scratch especially on smaller sample sizes. The out-of-sample performance using transfer learning from previously learned knowledge based on real-world images and videos could unlock the potential in many areas of imaging neuroscience where deep learning solutions are currently infeasible.

Keywords

Grants

  1. MC_PC_17228/Medical Research Council
  2. MC_QA137853/Medical Research Council

MeSH Term

Brain
Deep Learning
Humans
Intelligence
Neural Networks, Computer
Social Media

Word Cloud

Created with Highcharts 10.0.0learningimagesdeeptransferTransferDeepperformanceseveralincludingbiomedicalmanybrainsexagephenotypeshouseholdneuralpredictionsYouTubevideosdatasetespeciallyimagingneurosciencerecentlyachievedbest-in-classfieldsdomainsX-rayYetdatascarcityposesstrictlimittrainingsuccessfulsystemsapplicationsinvolvingstudytranslatestate-of-the-arttechniquessingle-subjectpredictionsimplercomplexnumberpeopleincomefluidintelligencesmokingbehaviorfine-tuned2D3DResNet-18convolutionalnetworkstargetphenotype∼40000UKBiobankparticipantspretrainingKineticsnaturalImageNeteffectiveclassificationAdditionallyparticularoutperformedmodelstrainedscratchsmallersamplesizesout-of-sampleusingpreviouslylearnedknowledgebasedreal-worldunlockpotentialareassolutionscurrentlyinfeasiblebrain:canimprovebrain-imagingBrainConvolutionalnetworkalgorithmsImagingMachine

Similar Articles

Cited By