Generalizing Deep Whole Brain Segmentation for Pediatric and Post-Contrast MRI with Augmented Transfer Learning

by   Camilo Bermudez, et al.

Generalizability is an important problem in deep neural networks, especially in the context of the variability of data acquisition in clinical magnetic resonance imaging (MRI). Recently, the Spatially Localized Atlas Network Tiles (SLANT) approach has been shown to effectively segment whole brain non-contrast T1w MRI with 132 volumetric labels. Enhancing generalizability of SLANT would enable broader application of volumetric assessment in multi-site studies. Transfer learning (TL) is commonly used to update the neural network weights for local factors; yet, it is commonly recognized to risk degradation of performance on the original validation/test cohorts. Here, we explore TL by data augmentation to address these concerns in the context of adapting SLANT to anatomical variation and scanning protocol. We consider two datasets: First, we optimize for age with 30 T1w MRI of young children with manually corrected volumetric labels, and accuracy of automated segmentation defined relative to the manually provided truth. Second, we optimize for acquisition with 36 paired datasets of pre- and post-contrast clinically acquired T1w MRI, and accuracy of the post-contrast segmentations assessed relative to the pre-contrast automated assessment. For both studies, we augment the original TL step of SLANT with either only the new data or with both original and new data. Over baseline SLANT, both approaches yielded significantly improved performance (signed rank tests; pediatric: 0.89 vs. 0.82 DSC, p<0.001; contrast: 0.80 vs 0.76, p<0.001). The performance on the original test set decreased with the new-data only transfer learning approach, so data augmentation was superior to strict transfer learning.



page 2

page 3

page 5

page 6


Multi-center, multi-vendor automated segmentation of left ventricular anatomy in contrast-enhanced MRI

Accurate delineation of the left ventricular boundaries in late gadolini...

Transfer Learning in 4D for Breast Cancer Diagnosis using Dynamic Contrast-Enhanced Magnetic Resonance Imaging

Deep transfer learning using dynamic contrast-enhanced magnetic resonanc...

Deep Learning-based Type Identification of Volumetric MRI Sequences

The analysis of Magnetic Resonance Imaging (MRI) sequences enables clini...

The role of MRI physics in brain segmentation CNNs: achieving acquisition invariance and instructive uncertainties

Being able to adequately process and combine data arising from different...

A Comparative Study of Existing and New Deep Learning Methods for Detecting Knee Injuries using the MRNet Dataset

This work presents a comparative study of existing and new techniques to...

Inter-Scanner Harmonization of High Angular Resolution DW-MRI using Null Space Deep Learning

Diffusion-weighted magnetic resonance imaging (DW-MRI) allows for non-in...

An augmentation strategy to mimic multi-scanner variability in MRI

Most publicly available brain MRI datasets are very homogeneous in terms...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.