Reassessing CVSS Severity Classification with LightGBM and Operating System Metadata
DOI:
https://doi.org/10.65455/451jd869关键词:
LightGBM, CVSS Severity, Operating System Metadata, Data Leakage, TreeSHAP, Vulnerability Classification摘要
CVSS severity and observed exploitation are different prediction targets. This study reassesses a LightGBM pipeline using an available 1,314-record vulnerability dataset and explicitly defines its label as a CVSS score greater than 9. The original 98.48% accuracy is reproducible, but the pipeline encodes complete score-defining vectors and fits text features before splitting. We therefore remove identifiers and complete vectors, fit preprocessing only on training records, and compare five classifiers using parsed attack vector, TF-IDF text and two operating-system metadata features. After description deduplication, 882 records with CVE identifier years up to 2023 are used for training and 390 records from identifier year 2024 for testing. This is an identifier-year robustness check, not a disclosure-time evaluation. LightGBM achieves 95.64% accuracy, 41.67% recall, 0.8412 ROC-AUC and 0.4821 average precision; the majority baseline already achieves 93.85% accuracy. Adding operating-system metadata changes average precision by −0.0303 (paired bootstrap 95% interval −0.1077 to 0.0409), providing no consistent improvement. TreeSHAP dependence plots and error cases reveal strong reliance on attack vector and lexical associations. The results support a limited severity-classification case study and demonstrate why these data cannot establish real-world exploitability prediction.
参考
[1]FIRST. Common Vulnerability Scoring System v3.1: Specification Document[Z/OL]. https://www.first.org/cvss/v3-1/specification-document. Accessed 19 September 2026.
[2]FIRST. Common Vulnerability Scoring System v3.1: User Guide[Z/OL]. https://www.first.org/cvss/v3-1/user-guide. Accessed 19 September 2026.
[3]JACOBS J, ROMANOSKY S, EDWARDS B, et al. Exploit Prediction Scoring System (EPSS). Digital Threats: Research and Practice, 2021, 2(3): 1-17. DOI: https://doi.org/10.1145/3436242
[4]JACOBS J, ROMANOSKY S, SUCIU O, et al. Enhancing Vulnerability Prioritization: Data-Driven Exploit Predictions with Community-Driven Insights[Z/OL]. arXiv:2302.14172, 2023. DOI: https://doi.org/10.1109/EuroSPW59978.2023.00027
[5]SUCIU O, NELSON C, LYU Z, et al. Expected Exploitability: Predicting the Development of Functional Vulnerability Exploits//31st USENIX Security Symposium. Berkeley, CA: USENIX Association, 2022: 377-394.
[6]KE G, MENG Q, FINLEY T, et al. LightGBM: A Highly Efficient Gradient Boosting Decision Tree. Advances in Neural Information Processing Systems, 2017, 30.
[7]LIU Z. Explainable Machine Learning for Telecom Customer Churn Prediction and Actionable Retention Strategies. Applied Artificial Intelligence Research, 2026, 2(1). DOI: https://doi.org/10.65455/1z4v1627
[8]WANG X. A Case Study of the Influence of Multifarious Factors on Traffic Flow Forecasting. Applied Artificial Intelligence Research, 2026, 2(2). DOI: https://doi.org/10.65455/sbd3t486
[9]CVE Program. Frequently Asked Questions, archived CVE website: CVE identifier year[Z/OL]. https://www.cve.org/Resources/Media/Archives/OldWebsite/about/faqs.html. Accessed 19 September 2026.
[10]LUNDBERG S M, LEE S I. A Unified Approach to Interpreting Model Predictions. Advances in Neural Information Processing Systems, 2017, 30.
##submission.downloads##
已出版
##submission.dataAvailability##
This reanalysis uses a preprocessed five-column CSV snapshot (SHA-256: 872f14ef9010f585b12c706e6553690ca2e83472c9bcb877698678308db006c4). The reproducibility package contains executable scripts, a frozen software environment manifest, held-out predictions, ablation metrics, and TreeSHAP values. All random processes are seeded for reproducibility, and supplementary files list all held-out CVEs to allow numerical validation of reported results. The original raw data collection workflow cannot be fully reconstructed from the shared materials.
##submission.license##
##submission.copyrightStatement##
##submission.license.cc.by4.footer##