Current browse context:
eess.IV
Change to browse by:
References & Citations
Electrical Engineering and Systems Science > Image and Video Processing
Title: Enhancing MR Image Segmentation with Realistic Adversarial Data Augmentation
(Submitted on 7 Aug 2021 (v1), last revised 19 Jun 2022 (this version, v3))
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.
Submission history
From: Chen Chen [view email][v1] Sat, 7 Aug 2021 11:32:37 GMT (6154kb,D)
[v2] Sat, 23 Apr 2022 21:53:20 GMT (6936kb,D)
[v3] Sun, 19 Jun 2022 20:59:11 GMT (7701kb,D)
Link back to: arXiv, form interface, contact.