Update README.md
Browse files
README.md
CHANGED
@@ -22,11 +22,7 @@ model-index:
|
|
22 |
|
23 |
# VLAI: A RoBERTa-Based Model for Automated Vulnerability Severity Classification
|
24 |
|
25 |
-
This model
|
26 |
-
|
27 |
-
The model was presented in the paper [VLAI: A RoBERTa-Based Model for Automated Vulnerability Severity Classification](https://huggingface.co/papers/2507.03607).
|
28 |
-
|
29 |
-
**Abstract:** VLAI is a transformer-based model that predicts software vulnerability severity levels directly from text descriptions. Built on RoBERTa, VLAI is fine-tuned on over 600,000 real-world vulnerabilities and achieves over 82% accuracy in predicting severity categories, enabling faster and more consistent triage ahead of manual CVSS scoring. The model and dataset are open-source and integrated into the Vulnerability-Lookup service.
|
30 |
|
31 |
For more information, visit the [Vulnerability-Lookup project page](https://vulnerability.circl.lu) or the [ML-Gateway GitHub repository](https://github.com/vulnerability-lookup/ML-Gateway), which demonstrates its usage in a FastAPI server.
|
32 |
|
|
|
22 |
|
23 |
# VLAI: A RoBERTa-Based Model for Automated Vulnerability Severity Classification
|
24 |
|
25 |
+
This model is a fine-tuned version of [hfl/chinese-macbert-base](https://huggingface.co/hfl/chinese-macbert-base) on the dataset [CIRCL/Vulnerability-CNVD](https://huggingface.co/datasets/CIRCL/Vulnerability-CNVD).
|
|
|
|
|
|
|
|
|
26 |
|
27 |
For more information, visit the [Vulnerability-Lookup project page](https://vulnerability.circl.lu) or the [ML-Gateway GitHub repository](https://github.com/vulnerability-lookup/ML-Gateway), which demonstrates its usage in a FastAPI server.
|
28 |
|