Advanced Automated Vulnerability Scoring: Improving Performance with a Fine-Tuned BERT-CNN Model

The ever-growing prevalence of software systems necessitates robust and efficient vulnerability assessment methods. An automated vulnerability assessment model can significantly streamline this process, providing accurate identification of vulnerability severity levels based on the given textual descriptions. This paper addresses this critical need by proposing a novel vulnerability score predictor model. Our model uses Bidirectional Encoder Representations from Transformers (BERT) and deep learning to classify vulnerability metrics accurately. We have fine-tuned the BERT tokenizer on NVD’s descriptions and used TextCNN for better performance. This approach represents a significant advancement in the field of automated vulnerability assessment, offering substantial benefits for security professionals by streamlining vulnerability identification and prioritization. The carried out experimental evaluations in comparison to various deep learning models, including LSTM, CNN, and LSTM-Attention, as well as two state-of-the-art automated vulnerability assessment baselines indicate an accuracy score of over $\mathbf{9 7 \%}$ across all vulnerability metrics.

Authors

Institutions

Publication Details

Published
2024-10-09
DOI
https://doi.org/10.1109/ist64061.2024.10843410
Citations
3
Primary Topic
Web Application Security Vulnerabilities
Type
article
Field-Weighted Citation Impact
2.34
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Advanced Automated Vulnerability Scoring: Improving Performance with a Fine-Tuned BERT-CNN Model

Andrea Pugliese, Narges Movahedkor, Seyedeh Leili Mirtaheri, AmirHossein Majd
3 citations
Web Application Security Vulnerabilities
2.34
article

Advanced Automated Vulnerability Scoring: Improving Performance with a Fine-Tuned BERT-CNN Model

Andrea Pugliese, Narges Movahedkor, Seyedeh Leili Mirtaheri, AmirHossein Majd
article en
3 citations

Abstract

The ever-growing prevalence of software systems necessitates robust and efficient vulnerability assessment methods. An automated vulnerability assessment model can significantly streamline this process, providing accurate identification of vulnerability severity levels based on the given textual descriptions. This paper addresses this critical need by proposing a novel vulnerability score predictor model. Our model uses Bidirectional Encoder Representations from Transformers (BERT) and deep learning to classify vulnerability metrics accurately. We have fine-tuned the BERT tokenizer on NVD’s descriptions and used TextCNN for better performance. This approach represents a significant advancement in the field of automated vulnerability assessment, offering substantial benefits for security professionals by streamlining vulnerability identification and prioritization. The carried out experimental evaluations in comparison to various deep learning models, including LSTM, CNN, and LSTM-Attention, as well as two state-of-the-art automated vulnerability assessment baselines indicate an accuracy score of over $\mathbf{9 7 \%}$ across all vulnerability metrics.

Amirkabir University of Technology (IR), University of Calabria (IT)
Climate action
Openalex Percentile: Top 8%
Web Application Security Vulnerabilities
2.34
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.