Logo image
Rewiring transformers for exploit likelihood prediction of cyber vulnerabilities
Conference proceeding   Peer reviewed

Rewiring transformers for exploit likelihood prediction of cyber vulnerabilities

Pranav Kumar, Harshit Agarwal, Vandana Sharma, Iqra Husain, Pradeep Hewage and Celestine Iwendi
AIP conference proceedings, Vol.3410(1)
APPLIED DATA SCIENCE AND SMART SYSTEMS (Rajpura, India, 13/12/2024–14/12/2024)
21/05/2026

Abstract

This paper addresses the challenge of predicting the exploit likelihood of cyber vulnerabilities, a critical aspect of vulnerability management in the world of evolving global cyber threats. Traditional vulnerability scoring systems like CVSS and machine learning models struggle to accurately capture exploitability dynamics due to their resource-intensive nature and reliance on static datasets. Our proposed solution involves fine-tuning advanced transformer models, specifically DistilBERT and GPT2-Large, to predict exploitability vulnerabilities by integrating textual descriptions and numerical CVSS scores. This method demonstrates significant improvements in predicting accuracy and efficiency using the datasets from the National Vulnerability Database (NVD). The DistilBERT model outperformed the GPT2 model in this task. The study underscores the potential of transformer-based architectures to revolutionize vulnerability management processes, enabling organizations to allocate resources more effectively for risk mitigation.
url
Link to published versionView
Published (Version of record) Publisher sites may require purchase to access content Restricted In Copyright All Rights Reserved  — This item is protected by copyright and/or related rights. You are free to use this Item in any way that is permitted by the copyright and related rights legislation (such as any Fair Dealing allowances allowed by copyright law). For other uses you need to obtain permission from the rights-holder(s).

Metrics

1 Record Views

Details

Logo image

Usage Policy