Data augmentation using learned transformations for one-shot medical image segmentation

Abstract

Image segmentation is an important task in many medical applications. Methods based on convolutional neural networks attain state-of-the-art accuracy; however, they typically rely on supervised training with large labeled datasets. Labeling medical images requires significant expertise and time, and typical hand-tuned approaches for data augmentation fail to capture the complex variations in such images. We present an automated data augmentation method for synthesizing labeled medical images. We demonstrate our method on the task of segmenting magnetic resonance imaging (MRI) brain scans. Our method requires only a single segmented scan, and leverages other unlabeled scans in a semi-supervised approach. We learn a model of transformations from the images, and use the model along with the labeled example to synthesize additional labeled examples. Each transformation is comprised of a spatial deformation field and an intensity change, enabling the synthesis of complex effects such as variations in anatomy and image acquisition procedures. We show that training a supervised segmenter with these new examples provides significant improvements over state-of-the-art methods for one-shot biomedical image segmentation. Our code is available at https://github.com/xamyzhao/brainstorm.

9 pages, CVPR 2019

References in corpus (9)

• TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems
• Efficient Multi-Scale 3D CNN with Fully Connected CRF for Accurate Brain Lesion Segmentation
• Densely Connected Convolutional Networks
• AutoAugment: Learning Augmentation Policies from Data
• Unsupervised Learning for Fast Probabilistic Diffeomorphic Registration
• Learning to Compose Domain-Specific Transformations for Data Augmentation
• Anatomy-specific classification of medical images using deep convolutional nets
• FusionSeg: Learning to combine motion and appearance for fully automatic segmention of generic objects in videos
• Few-Shot Segmentation Propagation with Guided Networks

Cited by in corpus (8)

• Embracing Imperfect Datasets: A Review of Deep Learning Solutions for Medical Image Segmentation
• Towards Data-Efficient Learning: A Benchmark for COVID-19 CT Lung and Infection Segmentation
• Transfer Learning in Magnetic Resonance Brain Imaging: a Systematic Review
• CovidExpert: A Triplet Siamese Neural Network framework for the detection of COVID-19
• Mine Your Own vieW: Self-Supervised Learning Through Across-Sample Prediction
• Towards Cross-modality Medical Image Segmentation with Online Mutual Knowledge Distillation
• Multi-task learning for joint weakly-supervised segmentation and aortic arch anomaly classification in fetal cardiac MRI
• 2D Multi-Class Model for Gray and White Matter Segmentation of the Cervical Spinal Cord at 7T