A Hybrid Ensemble method for Pulsar Candidate Classification
arXiv:1909.05303 · doi:10.1007/s10509-019-3602-4
Abstract
In this paper, three ensemble methods: Random Forest, XGBoost, and a Hybrid Ensemble method were implemented to classify imbalanced pulsar candidates. To assist these methods, tree models were used to select features among 30 features of pulsar candidates from references. The skewness of the integrated pulse profile, chi-squared value for sine-squared fit to amended profile and best S/N value play important roles in Random Forest, while the skewness of the integrated pulse profile is one of the most significant features in XGBoost. More than 20 features were selected by their relative scores and then applied in three ensemble methods. In the Hybrid Ensemble method, we combined Random Forest and XGBoost with EasyEnsemble. By changing thresholds, we tried to make a trade-off between Recall and Precision to make them approximately equal and as high as possible. Experiments on HTRU 1 and HTRU 2 datasets show that the Hybrid Ensemble method achieves higher Recall than the other two algorithms. In HTRU 1 dataset, Recall, Precision, and F-Score of the Hybrid Ensemble method are , , and , respectively. In HTRU 2 dataset, the three values of that are , , and , respectively.
15 pages, 11 figures, accepted by Astrophysics and Space Science
References in corpus (4)
- The Five-Hundred-Meter Aperture Spherical Radio Telescope (FAST) Project
- The Green Bank Northern Celestial Cap Pulsar Survey - I: Survey Description, Data Analysis, and Initial Results
- Discovery of 28 pulsars using new techniques for sorting pulsar candidates
- SPINN: a straightforward machine learning solution to the pulsar candidate selection problem