Morphological classification of radio galaxies with Wasserstein generative adversarial network-supported augmentation

Machine learning techniques that perform morphological classification of astronomical sources often suffer from a scarcity of labelled training data. Here, we focus on the case of supervised deep learning models for the morphological classification of radio galaxies, which is particularly topical fo...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Veröffentlicht in:RAS techniques and instruments 2023-01, Vol.2 (1), p.264-277
Hauptverfasser: Rustige, Lennart, Kummer, Janis, Griese, Florian, Borras, Kerstin, Bruggen, Marcus, Connor, Patrick L.S, Gaede, Frank, Kasieczka, Gregor, Knopp, Tobias, Schleper, Peter
Format: Artikel
Sprache:eng
Schlagworte:
Online-Zugang:Volltext
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:Machine learning techniques that perform morphological classification of astronomical sources often suffer from a scarcity of labelled training data. Here, we focus on the case of supervised deep learning models for the morphological classification of radio galaxies, which is particularly topical for the forthcoming large radio surveys. We demonstrate the use of generative models, specifically Wasserstein generative adversarial networks (wGANs), to generate data for different classes of radio galaxies. Further, we study the impact of augmenting the training data with images from our wGAN on three different classification architectures. We find that this technique makes it possible to improve models for the morphological classification of radio galaxies. A simple fully connected neural network benefits most from including generated images into the training set, with a considerable improvement of its classification accuracy. In addition, we find it is more difficult to improve complex classifiers. The classification performance of a convolutional neural network can be improved slightly. However, this is not the case for a vision transformer.
ISSN:2752-8200
2752-8200
DOI:10.1093/rasti/rzad016