City Research Online

Enhancing MR image segmentation with realistic adversarial data augmentation

Chen, C., Qin, C., Ouyang, C. , Li, Z., Wang, S., Qiu, H., Chen, L., Tarroni, G. ORCID: 0000-0002-0341-6138, Bai, W. & Rueckert, D. (2022). Enhancing MR image segmentation with realistic adversarial data augmentation. Medical Image Analysis, 82, article number 102597. doi: 10.1016/j.media.2022.102597

Abstract

The success of neural networks on medical image segmentation tasks typically relies on large labeled datasets for model training. However, acquiring and manually labeling a large medical image set is resource-intensive, expensive, and sometimes impractical due to data sharing and privacy issues. To address this challenge, we propose AdvChain, a generic adversarial data augmentation framework, aiming at improving both the diversity and effectiveness of training data for medical image segmentation tasks. AdvChain augments data with dynamic data augmentation, generating randomly chained photo-metric and geometric transformations to resemble realistic yet challenging imaging variations to expand training data. By jointly optimizing the data augmentation model and a segmentation network during training, challenging examples are generated to enhance network generalizability for the downstream task. The proposed adversarial data augmentation does not rely on generative networks and can be used as a plug-in module in general segmentation networks. It is computationally efficient and applicable for both low-shot supervised and semi-supervised learning. We analyze and evaluate the method on two MR image segmentation tasks: cardiac segmentation and prostate segmentation with limited labeled data. Results show that the proposed approach can alleviate the need for labeled data while improving model generalization ability, indicating its practical value in medical imaging applications.

Publication Type: Article
Additional Information: 2022 The Author(s). Published by Elsevier B.V. This is an open access article under the CC BY license (http://creativecommons.org/licenses/by/4.0/)
Publisher Keywords: MR image segmentation, Adversarial training, Data augmentation, Adversarial data augmentation, Model generalization
Subjects: Q Science > QA Mathematics > QA75 Electronic computers. Computer science
R Medicine > R Medicine (General)
Departments: School of Science & Technology > Computer Science
SWORD Depositor:
[thumbnail of Chen et al. - 2022 - Enhancing MR image segmentation with realistic adversarial data augmentation(2).pdf]
Preview
Text - Published Version
Available under License Creative Commons Attribution.

Download (3MB) | Preview

Export

Add to AnyAdd to TwitterAdd to FacebookAdd to LinkedinAdd to PinterestAdd to Email

Downloads

Downloads per month over past year

View more statistics

Actions (login required)

Admin Login Admin Login