CoolFace
Datasetpublic

CIRCL/vulnerability-scores

vulnerability-scores This dataset comprises 798,886 real-world vulnerabilities used to train and evaluate VLAI, a transformer-based model designed to predict software vulnerability severity levels directly from text descriptions, enabling faster and more consistent triage. The dataset is presented in the paper VLAI: A RoBERTa-Based Model for Automated Vulnerability Severity Classification. Sources Source Label Entries Share cvelistv5 CVE Program… See the full description on the dataset page: https://huggingface.co/datasets/CIRCL/vulnerability-scores.

sourceHugging Facecc-by-4.0updated 18h agoView on Hugging Face
11likes303downloads
Dataset Card

vulnerability-scores

This dataset comprises 798,886 real-world vulnerabilities used to train and evaluate VLAI, a transformer-based model designed to predict software vulnerability severity levels directly from text descriptions, enabling faster and more consistent triage.

The dataset is presented in the paper VLAI: A RoBERTa-Based Model for Automated Vulnerability Severity Classification.

Sources

SourceLabelEntriesShare
cvelistv5CVE Program (enriched with vulnrichment and Fraunhofer FKIE)377,27847.2%
githubGitHub Security Advisories375,21547.0%
csaf_redhatCSAF Red Hat28,8593.6%
pysecPySec advisories7,5250.9%
csaf_cisaCSAF CISA6,0660.8%
csaf_ciscoCSAF Cisco3,9430.5%

Extracted from the database of Vulnerability-Lookup with the VulnTrain project. Dumps of the data are available here.

Splits

SplitExamples
train718,997
test79,889

Fields

FieldTypeDescription
idstringVulnerability identifier (e.g., CVE-2024-1234, GHSA-xxxx, PYSEC-2024-xxx)
titlestringVulnerability title
descriptionstringVulnerability description in English
cpeslist[string]Common Platform Enumeration identifiers
cvss_v4_0floatCVSS v4.0 score
cvss_v3_1floatCVSS v3.1 score
cvss_v3_0floatCVSS v3.0 score
cvss_v2_0floatCVSS v2.0 score
patch_commit_urlstringURL to the patch commit on GitHub, if available
sourcestringData source identifier

Usage

python
import json
from datasets import load_dataset

dataset = load_dataset("CIRCL/vulnerability-scores")

vulnerabilities = ["CVE-2012-2339", "RHSA-2023:5964", "GHSA-7chm-34j8-4f22", "PYSEC-2024-225"]

filtered_entries = dataset.filter(lambda elem: elem["id"] in vulnerabilities)

for entry in filtered_entries["train"]:
    print(json.dumps(entry, indent=4))

Related models

References