Using High-Performance Computing to Scale Generative Adversarial Networks
Name
Flores-floresd-meng-eecs-2021-thesis.pdf
Description
Thesis PDF
Size
1.56 MB
Format
Adobe PDF
Checksum (MD5)
94241905e641067a37d5b58ddd3b4689
Author(s)
Flores, Diana J.
Advisor(s)
Hemberg, Erik
Toutouh, Jamal
O’Reilly, Una-May
Date Issued
June 2021
Publisher
Massachusetts Institute of Technology
Abstract
Generative adversarial networks(GANs) are methods that can be used for data augmentation, which helps in creating better detection models for rare or imbalanced datasets. They can be difficult to train due to issues such as mode collapse. We aim to improve the performance and accuracy of the Lipizzaner GAN framework by taking advantage of its distributed nature and running it at very large scales. Lipizzaner was implemented for robustness, but has not been tested at scale in high performance computing(HPC) systems. We believe that by utilizing HPC technologies, we can scale up Lipizzaner and observe performance enhancements. This thesis achieves this scale up, using Oak Ridge National Labs’ Summit Supercomputer. We observed improvements in the performance of Lipizzaner, especially when run with poorer network architectures, which implies Lipizzaner is able to overcome network limitations through scale.
MIT Department
Massachusetts Institute of Technology. Department of Electrical Engineering and Computer Science
Terms of Use
In Copyright - Educational Use Permitted
Copyright MIT
Persistent DSpace Link