Semi-supervised classification of stars, galaxies and quasars using K-means and random-forest approaches
arXiv:2507.14072 · doi:10.1051/0004-6361/202555620
Abstract
Classifying stars, galaxies, and quasars is essential for understanding cosmic structure and evolution; however, the vast data from modern surveys make manual classification impractical, while supervised learning methods remain constrained by the scarcity of labeled spectroscopic data. We aim to develop a scalable, label-efficient method for astronomical classification by leveraging semi-supervised learning (SSL) to overcome the limitations of fully supervised approaches. We propose a novel SSL framework combining K-means clustering with random forest classification. Our method partitions unlabeled data into 50 clusters, propagates labels from spectroscopically confirmed centroids to 95% of cluster members, and trains a random forest on the expanded pseudo-labeled dataset. We applied this to the CPz catalog, containing multi-survey photometric and spectroscopic data, and compared performance with a fully supervised random forest. Our SSL approach achieves F1 scores of 98.8%, 98.9%, and 92.0% for stars, galaxies, and quasars, respectively, closely matching the supervised method with F1 scores of 99.1%, 99.1%, and 93.1%, while outperforming traditional color-cut techniques. The method demonstrates robustness in high-dimensional feature spaces and superior label efficiency compared to prior work. This work highlights SSL as a scalable solution for astronomical classification when labeled data is limited, though performance may be degraded in lower dimensional settings.
arXiv comment9 pages, 9 figures, 2 tables, Accepted for Publication in A&A
References in corpus (35)
- SMOTE: Synthetic Minority Over-sampling Technique
- The Wide-field Infrared Survey Explorer (WISE): Mission Description and Initial On-orbit Performance
- LSST: from Science Drivers to Reference Design and Anticipated Data Products
- The Eleventh and Twelfth Data Releases of the Sloan Digital Sky Survey: Final Data from SDSS-III
- Overview of the DESI Legacy Imaging Surveys
- A Comparative Study of Efficient Initialization Methods for the K-Means Clustering Algorithm
- The 6dF Galaxy Survey: Final Redshift Release (DR3) and Southern Large-Scale Structures
- Preliminary Results from NEOWISE: An Enhancement to the Wide-field Infrared Survey Explorer for Solar System Science
- Mid-Infrared Selection of AGN with the Wide-Field Infrared Survey Explorer. I. Characterizing WISE-Selected AGN in COSMOS
- Galaxy And Mass Assembly (GAMA): end of survey report and data release 2
- The 6dF Galaxy Survey: Samples, Observational Techniques and the First Data Release
- The VIMOS VLT Deep Survey final data release: a spectroscopic sample of 35016 galaxies and AGN out to z~6.7 selected with 17.5<=i_{AB}<=24.7
- The VISTA Deep Extragalactic Observations (VIDEO) Survey
- The PRIsm MUlti-Object Survey (PRIMUS) I: Survey Overview and Characteristics
- The first and second data releases of the Kilo-Degree Survey
- The VIMOS Public Extragalactic Survey (VIPERS): First Data Release of 57 204 spectroscopic measurements
- Colors of 2625 Quasars at 0<z<5 Measured in the Sloan Digital Sky Survey Photometric System
- The PRIsm MUlti-object Survey (PRIMUS). II. Data Reduction and Redshift Fitting
- SuperRAENN: A Semi-supervised Supernova Photometric Classification Pipeline Trained on Pan-STARRS1 Medium Deep Survey Supernovae
- Galaxy And Mass Assembly (GAMA): Assimilation of KiDS into the GAMA database
- Identifying galaxies, quasars, and stars with machine learning: A new catalogue of classifications for 111 million SDSS sources without spectra
- Unsupervised star, galaxy, qso classification: Application of HDBSCAN
- Photometric classification of emission line galaxies with Machine Learning methods
- CPz: Classification-aided photometric-redshift estimation
- Machine-learning identification of galaxies in the WISExSuperCOSMOS all-sky catalogue
- A review of unsupervised learning in astronomy
- Photometric redshift-aided classification using ensemble learning
- Photometric selection and redshifts for quasars in the Kilo-Degree Survey Data Release 4
- Catalog of quasars from the Kilo-Degree Survey Data Release 3
- Towards automatic classification of all WISE sources
- Photometric identification of compact galaxies, stars and quasars using multiple neural networks
- Wide Area VISTA Extra-galactic Survey (WAVES): Unsupervised star-galaxy separation on the WAVES-Wide photometric input catalogue using UMAP and
- Semi-Supervised Classification and Clustering Analysis for Variable Stars
- J-PLUS DR3: Galaxy-Star-Quasar classification
- Mapping the Similarities of Spectra: Global and Locally-biased Approaches to SDSS Galaxy Data