Galaxy Morphology Classification via Deep Semi-Supervised Learning with Limited Labeled Data
Galaxy morphology classification plays a crucial role in understanding the structure and evolution of the universe. With galaxy observation data growing exponentially, machine learning has become a core technology for this classification task. However, traditional machine learning methods predominantly rely on supervised learning frameworks, and their dependence on large of labeled samples limits practical applications. To address this challenge, we propose an innovative hybrid semi-supervised model, GC-SWGAN, designed to tackle galaxy morphology classification under conditions of limited labeled data. This model integrates semi-supervised generative adversarial networks (SGAN) with Wasserstein GAN with gradient penalty (WGAN-GP), establishing a multi-task learning framework. Within this framework, the discriminator and classifier are designed independently while sharing part of the architecture. By collaborating with the generator, the model significantly enhances both classification performance and sample generation capabilities, while also improving convergence and stability during training. Experimental results demonstrate that, on the Galaxy10 DECals dataset, GC-SWGAN achieves comparable or even superior classification accuracy (exceeding 75%) using only one-fifth of the labeled samples typically required by conventional fully supervised methods. Under identical labeled conditions, the model displays excellent generalization performance, attaining approximately 84% classification accuracy. Notably, in extreme scenarios where only 10\% of the data is labeled, GC-SWGAN still achieves high classification accuracy (over 68%), fully demonstrating its stability and effectiveness in low-labeled data environments. Furthermore, galaxy images generated by GC-SWGAN are visually similar to real samples.