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

Institutions

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
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

GreenBERT: A lightweight green transformer for automated prediction of software vulnerability scores

Ali Tafti, Andrea Pugliese, Seyedeh Leili Mirtaheri, Hamid Heidari Soureshjani
Array
Information and Cyber Security
article

GreenBERT: A lightweight green transformer for automated prediction of software vulnerability scores

Ali Tafti, Andrea Pugliese, Seyedeh Leili Mirtaheri, Hamid Heidari Soureshjani
article en

Abstract

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.

ArrayVol. 28
Kharazmi University (IR), University of Calabria (IT)
Openalex Percentile: Top 50%
Information and Cyber Security
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

GreenBERT: A lightweight green transformer for automated prediction of software vulnerability scores — Ali Tafti, Andrea Pugliese, et al. · Array (2025) | TGRS Research Map | TGRS