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.
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
Extracted from the database of Vulnerability-Lookup with the VulnTrain project. Dumps of the data are available here.
Splits
Fields
Usage
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
- CIRCL/vulnerability-severity-classification-roberta-base — RoBERTa severity classifier
- CIRCL/vulnerability-severity-classification-distilbert-base-uncased — DistilBERT severity classifier
References
- Vulnerability-Lookup — the vulnerability data source
- VulnTrain — training pipeline
- ML-Gateway — inference API
- VLAI paper — Bonhomme, C., Dulaunoy, A. (2025)
