Joint Optimization of Class-Specific Training- and Test-Time Data Augmentation in Segmentation

This paper presents an effective and general data augmentation framework for medical image segmentation. We adopt a computationally efficient and data-efficient gradient-based meta-learning scheme to explicitly align the distribution of training and validation data which is used as a proxy for unsee...

Full description

Saved in:
Bibliographic Details
Published in:IEEE transactions on medical imaging 2023-11, Vol.PP (11), p.1-1
Main Authors: Li, Zeju, Kamnitsas, Konstantinos, Dou, Qi, Qin, Chen, Glocker, Ben
Format: Article
Language:English
Subjects:
Biomedical imaging
Data augmentation
Data models
Heterogeneity
Image processing
Image segmentation
Medical imaging
meta-learning
Optimization
Task analysis
Testing time
Training
Training data
Citations: Items that this one cites
Items that cite this one
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:This paper presents an effective and general data augmentation framework for medical image segmentation. We adopt a computationally efficient and data-efficient gradient-based meta-learning scheme to explicitly align the distribution of training and validation data which is used as a proxy for unseen test data. We improve the current data augmentation strategies with two core designs. First, we learn class-specific training-time data augmentation (TRA) effectively increasing the heterogeneity within the training subsets and tackling the class imbalance common in segmentation. Second, we jointly optimize TRA and test-time data augmentation (TEA), which are closely connected as both aim to align the training and test data distribution but were so far considered separately in previous works. We demonstrate the effectiveness of our method on four medical image segmentation tasks across different scenarios with two state-of-the-art segmentation models, DeepMedic and nnU-Net. Extensive experimentation shows that the proposed data augmentation framework can significantly and consistently improve the segmentation performance when compared to existing solutions. Code is publicly available 1 .
ISSN:0278-0062
1558-254X
DOI:10.1109/TMI.2023.3282728