Radio Galaxy Classification with wGAN-Supported Augmentation

Janis Kummer,Lennart Rustige,Florian Griese,Kerstin Borras,Marcus Brüggen,Patrick L. S. Connor,Frank Gaede,Gregor Kasieczka,Peter Schleper
DOI: https://doi.org/10.48550/arXiv.2206.15131
2022-06-30
Instrumentation and Methods for Astrophysics
Abstract:Novel techniques are indispensable to process the flood of data from the new generation of radio telescopes. In particular, the classification of astronomical sources in images is challenging. Morphological classification of radio galaxies could be automated with deep learning models that require large sets of labelled training data. Here, we demonstrate the use of generative models, specifically Wasserstein GANs (wGAN), to generate artificial data for different classes of radio galaxies. Subsequently, we augment the training data with images from our wGAN. We find that a simple fully-connected neural network for classification can be improved significantly by including generated images into the training set.
What problem does this paper attempt to address?