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 |
|