用机器学习提升小行星危险预警准确率,随机森林和梯度提升表现最优。
Enhanced Predictive Modeling for Hazardous Near-Earth Object Detection: A Comparative Analysis of Advanced Resampling Strategies and Machine Learning Algorithms in Planetary Risk Assessment
- 对比六种算法,随机森林与梯度提升在数据预处理后表现最佳。
- 随机森林与梯度提升的F2-score分别达0.987和0.986,误报漏报极少。
- 适合关注行星风险评估与高精度分类的科研及航天机构参考。
本研究通过二分类框架评估多种机器学习模型对近地小行星(NEOs)危险性的预测性能,涵盖数据标准化、幂变换及交叉验证。比较了六种分类器:随机森林(RFC)、梯度提升(GBC)、支持向量机(SVC)、线性判别分析(LDA)、逻辑回归(LR)和K近邻(KNN)。RFC与GBC表现最佳,F2-score分别为0.987和0.986,波动极小;SVC次之,得分为0.896;LDA与LR中等,得分约为0.749和0.748;KNN因难以捕捉复杂数据模式,表现较差,得分仅0.691。两者混淆矩阵显示极低的假阳性与假阴性,准确率分别达到99.7%与99.6%。结果表明集成方法在高精度与高召回率任务中优势显著,并强调模型选择应结合数据特征与评估指标。未来可优化超参数并引入先进特征工程以进一步提升模型在小行星危险预测中的准确性与鲁棒性。
原文摘要 · Abstract (English)
This study evaluates the performance of several machine learning models for predicting hazardous near-Earth objects (NEOs) through a binary classification framework, including data scaling, power transformation, and cross-validation. Six classifiers were compared, namely Random Forest Classifier (RFC), Gradient Boosting Classifier (GBC), Support Vector Classifier (SVC), Linear Discriminant Analysis (LDA), Logistic Regression (LR), and K-Nearest Neighbors (KNN). RFC and GBC performed the best, both with an impressive F2-score of 0.987 and 0.986, respectively, with very small variability. SVC followed, with a lower but reasonable score of 0.896. LDA and LR had a moderate performance with scores of around 0.749 and 0.748, respectively, while KNN had a poor performance with a score of 0.691 due to difficulty in handling complex data patterns. RFC and GBC also presented great confusion matrices with a negligible number of false positives and false negatives, which resulted in outstanding accuracy rates of 99.7% and 99.6%, respectively. These findings highlight the power of ensemble methods for high precision and recall and further point out the importance of tailored model selection with regard to dataset characteristics and chosen evaluation metrics. Future research could focus on the optimization of hyperparameters with advanced features engineering to further the accuracy and robustness of the model on NEO hazard predictions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。