GreenBERT: A lightweight green transformer for automated prediction of software vulnerability scores
Timely assessment of software vulnerabilities is critical for effective patch prioritization, yet manual Common Vulnerability Scoring System (CVSS) scoring remains slow and resource-intensive. While transformer-based models such as BERT have advanced automated scoring, their substantial computational demands conflict with sustainable, green computing objectives. This paper introduces GreenBERT , a tailored ensemble of lightweight student Transformers, each specialized on individual CVSS metrics through a targeted multi-head knowledge distillation framework. By jointly optimizing alignment with ground-truth labels and softened outputs from a fine-tuned BERT teacher, GreenBERT efficiently captures complex vulnerability patterns while significantly reducing computational overhead. Extensive experiments on the National Vulnerability Database (NVD) and a more challenging COMBINED dataset demonstrate that GreenBERT achieves an average F1-score improvement exceeding 6% over the BERT baseline, while simultaneously reducing inference time by approximately 80% and cutting energy usage and CO 2 emissions by about 70%. These results position GreenBERT as a robust, scalable, and environmentally conscious solution for high-performance vulnerability scoring, effectively reconciling the traditionally conflicting goals of predictive accuracy and sustainable AI.
Authors
- Ali Tafti (ORCID: https://orcid.org/0000-0002-4450-3668)
- Andrea Pugliese (ORCID: https://orcid.org/0000-0003-4385-958X)
- Seyedeh Leili Mirtaheri (ORCID: https://orcid.org/0000-0002-0744-5876)
- Hamid Heidari Soureshjani
Institutions
- Kharazmi University (IR)
- University of Calabria (IT)
Publication Details
- Journal
- Array
- Published
- 2025-11-26
- DOI
- https://doi.org/10.1016/j.array.2025.100536
- Primary Topic
- Information and Cyber Security
- Type
- article
- Field-Weighted Citation Impact
- 0.00