Deep granular feature-label distribution learning for neuroimaging-based infant age prediction

for UNC/UMN Baby Connectome Project Consortium

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Neuroimaging-based infant age prediction is important for brain development analysis but often suffers insufficient data. To address this challenge, we introduce label distribution learning (LDL), a popular machine learning paradigm focusing on the small sample problem, for infant age prediction. As directly applying LDL yields dramatically increased number of day-to-day age labels and also extremely scarce data describing each label, we propose a new strategy, called granular label distribution (GLD). Particularly, by assembling the adjacent labels to granules and designing granular distributions, GLD makes each brain MRI contribute to not only its own age but also its neighboring ages at a granule scale, which effectively keeps the information augmentation superiority of LDL and reduces the number of labels. Furthermore, to extremely augment the information supplied by the small data, we propose a novel method named granular feature distribution (GFD). GFD leverages the variability of the brain images at the same age, thus significantly increases the learning effectiveness. Moreover, deep neural network is exploited to approximate the GLD. These strategies constitute a new model: deep granular feature-label distribution learning (DGFLDL). By taking 8 types of cortical morphometric features from structural MRI as predictors, the proposed DGFLDL is validated on infant age prediction using 384 brain MRI scans from 35 to 848 days after birth. Our proposed method, approaching the mean absolute error as 36.1 days, significantly outperforms the baseline methods. Besides, the permutation importance analysis of features based on our method reveals important biomarkers of infant brain development.

Original languageEnglish
Title of host publicationMedical Image Computing and Computer Assisted Intervention – MICCAI 2019 - 22nd International Conference, Proceedings
EditorsDinggang Shen, Pew-Thian Yap, Tianming Liu, Terry M. Peters, Ali Khan, Lawrence H. Staib, Caroline Essert, Sean Zhou
PublisherSpringer
Pages149-157
Number of pages9
ISBN (Print)9783030322502
DOIs
Publication statusPublished - 2019
Event22nd International Conference on Medical Image Computing and Computer-Assisted Intervention, MICCAI 2019 - Shenzhen, China
Duration: 2019 Oct 132019 Oct 17

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume11767 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference22nd International Conference on Medical Image Computing and Computer-Assisted Intervention, MICCAI 2019
CountryChina
CityShenzhen
Period19/10/1319/10/17

Keywords

  • Cortical features
  • Infant age prediction
  • Label distribution learning

ASJC Scopus subject areas

  • Theoretical Computer Science
  • Computer Science(all)

Fingerprint Dive into the research topics of 'Deep granular feature-label distribution learning for neuroimaging-based infant age prediction'. Together they form a unique fingerprint.

Cite this