File size: 901 Bytes
c1697e9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | ---
license: mit
tags:
- vulnerability-detection
- linux-kernel
- security
- pytorch
datasets:
- pebblebed/kernel-vuln-dataset
metrics:
- auc
- recall
- precision
pipeline_tag: text-classification
---
# VulnBERT v8
Vulnerability detection model for Linux kernel commits.
## Results
| Metric | Value |
|--------|-------|
| AUC | 0.987 |
| Recall | 91.4% |
| Precision | 88.4% |
| F1 | 0.899 |
| FPR | 5.9% |
## Usage
```python
import torch
checkpoint = torch.load("pytorch_model.pt", map_location="cpu")
model.load_state_dict(checkpoint["model_state_dict"])
```
Full code: [github.com/quguanni/vulnbert](https://github.com/quguanni/vulnbert)
## Training
- Dataset: [pebblebed/kernel-vuln-dataset](https://huggingface.co/datasets/pebblebed/kernel-vuln-dataset) (650K commits)
- Architecture: CodeBERT + 118 handcrafted features
- Time: ~7 hours on NVIDIA GH200
## License
MIT
|