A Galaxy Image Augmentation Method Based on Few-shot Learning and Generative Adversarial Networks

Yiqi Yao,Jinqu Zhang,Ping Du,Shuyu Dong,Xiangru Li
DOI: https://doi.org/10.1088/1674-4527/ad2128
2024-01-22
Research in Astronomy and Astrophysics
Abstract:Abstract Galaxy morphology classification based on machine learning are a typical technique to handle enormous amounts of astronomical observation data, but the key challenge is how to provide enough training data for the machine learning models. Therefor this article proposes an image data augmentation method that combines few-shot learning and generative adversarial networks. The Galaxy10 DECaLs dataset is selected for the experiments with consistency, variance, and augmentation effects being evaluated. Three popular networks, including AlexNet, VGG, and ResNet, are used as examples to study the effectiveness of different augmentation methods on galaxy morphology classifications. Experiment results show that the proposed method can generates galaxy images and can be used for expanding the classification model's training set. According to comparative studies, the best enhancement effect on model performance is obtained when the generated dataset is between 0.5 and 1 times the original dataset. Meanwhile, different augmentation strategies have considerably varied effects on different types of galaxies. FSL-GAN achieved the best classification performance on the ResNet network for In between Round Smooth Galaxies and Unbarred Loose Spiral Galaxies, with F1 Scores of 89.54% and 63.18%, respectively. Experimental comparison reveals that various data augmentation techniques have varied effects on different categories of galaxy morphology and machine learning models. Finally, the best augmentation strategies for each galaxy category are suggested.
astronomy & astrophysics
What problem does this paper attempt to address?