Downloads · 30 days
0
vulnberta-cweml-roberta-base
fusier/vulnberta-cweml-roberta-base
vulnberta-cweml-roberta-base is a machine learning model from fusier. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This is a tiered model for CWE classification based on vulnerability description. The models fine-tuned from RoBERTa-base with vulnerabilities from NIST feeds spanning from 2002 to early 2022.
Access
Public
Updated Feb 27, 2025
Repo size
7.5 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.ckpt7.5 GB · 100%
At a glance
- License
- mit
- Access
- Public
- Created
- Feb 27, 2025
- Updated
- Feb 27, 2025
- SHA
- 9cbd9ed7
Model card
From the Hugging Face model README
VulnBERTa
This is a tiered model for CWE classification based on vulnerability description. The models fine-tuned from RoBERTa-base with vulnerabilities from NIST feeds spanning from 2002 to early 2022.
References
@INPROCEEDINGS{10628784, author={Turtiainen, Hannu and Costin, Andrei}, booktitle={2024 IEEE European Symposium on Security and Privacy Workshops (EuroS&PW)}, title={VulnBERTa: On Automating CWE Weakness Assignment and Improving the Quality of Cybersecurity CVE Vulnerabilities Through ML/NLP}, year={2024}, volume={}, number={}, pages={618-625}, keywords={Systematics;Databases;Terminology;Organizations;Machine learning;NIST;Transformers;CWE;CVE;ML;AI;BERT;RoBERTa;Vulnerabilities;Vulnerability management;VulnBERTa;CWEml}, doi={10.1109/EuroSPW61312.2024.00075}}
- License
- mit
- Created
- Feb 27, 2025
- Updated
- Feb 27, 2025