VLAI: A RoBERTa-Based Model for Automated Vulnerability Severity Classification
Cédric Bonhomme
Alexandre Dulaunoy
Main:7 Pages
2 Figures
Bibliography:2 Pages
Abstract
This paper presents VLAI, a transformer-based model that predicts software vulnerability severity levels directly from text descriptions. Built on RoBERTa, VLAI is fine-tuned on over 600,000 real-world vulnerabilities and achieves over 82% accuracy in predicting severity categories, enabling faster and more consistent triage ahead of manual CVSS scoring. The model and dataset are open-source and integrated into the Vulnerability-Lookup service.
View on arXivComments on this paper
