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.
@article{arxiv.2206.15131,
title = {Radio Galaxy Classification with wGAN-Supported Augmentation},
author = {Janis Kummer and Lennart Rustige and Florian Griese and Kerstin Borras and Marcus Brüggen and Patrick L. S. Connor and Frank Gaede and Gregor Kasieczka and Peter Schleper},
journal= {arXiv preprint arXiv:2206.15131},
year = {2022}
}
Comments
10 pages, 6 figures; accepted to ml.astro; v2: matches published version