Generative multi-adversarial network for striking the right balance in abdominal image segmentation
- Purpose: The identification of abnormalities that are relatively rare within otherwise normal anatomy is a major challenge for deep learning in the semantic segmentation of medical images. The small number of samples of the minority classes in the training data makes the learning of optimal classification challenging, while the more frequently occurring samples of the majority class hamper the generalization of the classification boundary between infrequently occurring target objects and classes. In this paper, we developed a novel generative multi-adversarial network, called Ensemble-GAN, for mitigating this class imbalance problem in the semantic segmentation of abdominal images. Method: The Ensemble-GAN framework is composed of a single-generator and a multi-discriminator variant for handling the class imbalance problem to provide a better generalization than existing approaches. The ensemble model aggregates the estimates of multiple models by training from different initializations and losses from various subsets of the trainingPurpose: The identification of abnormalities that are relatively rare within otherwise normal anatomy is a major challenge for deep learning in the semantic segmentation of medical images. The small number of samples of the minority classes in the training data makes the learning of optimal classification challenging, while the more frequently occurring samples of the majority class hamper the generalization of the classification boundary between infrequently occurring target objects and classes. In this paper, we developed a novel generative multi-adversarial network, called Ensemble-GAN, for mitigating this class imbalance problem in the semantic segmentation of abdominal images. Method: The Ensemble-GAN framework is composed of a single-generator and a multi-discriminator variant for handling the class imbalance problem to provide a better generalization than existing approaches. The ensemble model aggregates the estimates of multiple models by training from different initializations and losses from various subsets of the training data. The single generator network analyzes the input image as a condition to predict a corresponding semantic segmentation image by use of feedback from the ensemble of discriminator networks. To evaluate the framework, we trained our framework on two public datasets, with different imbalance ratios and imaging modalities: the Chaos 2019 and the LiTS 2017. Result: In terms of the F1 score, the accuracies of the semantic segmentation of healthy spleen, liver, and left and right kidneys were 0.93, 0.96, 0.90 and 0.94, respectively. The overall F1 scores for simultaneous segmentation of the lesions and liver were 0.83 and 0.94, respectively. Conclusion: The proposed Ensemble-GAN framework demonstrated outstanding performance in the semantic segmentation of medical images in comparison with other approaches on popular abdominal imaging benchmarks. The Ensemble-GAN has the potential to segment abdominal images more accurately than human experts.…
Author details: | Mina RezaeiORCiDGND, Janne J. NäppiORCiD, Christoph LippertORCiDGND, Christoph MeinelORCiDGND, Hiroyuki Yoshida |
---|---|
DOI: | https://doi.org/10.1007/s11548-020-02254-4 |
ISSN: | 1861-6410 |
ISSN: | 1861-6429 |
Pubmed ID: | https://pubmed.ncbi.nlm.nih.gov/32897490 |
Title of parent work (English): | International journal of computer assisted radiology and surgery |
Publisher: | Springer |
Place of publishing: | Berlin |
Publication type: | Article |
Language: | English |
Date of first publication: | 2020/09/08 |
Publication year: | 2020 |
Release date: | 2023/03/24 |
Tag: | abdominal imaging; generative multi-discriminative networks; imbalanced learning; segmentation; semantic |
Volume: | 15 |
Issue: | 11 |
Number of pages: | 12 |
First page: | 1847 |
Last Page: | 1858 |
Funding institution: | Hasso Plattner Institute, Germany; National Institutes of Health,; USAUnited States Department of Health & Human ServicesNational; Institutes of Health (NIH) - USA [R01CA212382, R01EB023942, R21EB024025,; R21EB022747] |
Organizational units: | An-Institute / Hasso-Plattner-Institut für Digital Engineering gGmbH |
DDC classification: | 0 Informatik, Informationswissenschaft, allgemeine Werke |
Peer review: | Referiert |
Publishing method: | Open Access / Hybrid Open-Access |
License (German): | CC-BY - Namensnennung 4.0 International |