File size: 1,265 Bytes
fd0bbda | 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 51 52 53 54 55 56 57 58 59 60 61 | ---
license: apache-2.0
---
---
language:
- zh
- en
tags:
- vulnerability-detection
- security
- code-analysis
- mindspore
- t5
license: mit
---
# BreakingGod - CVE Bug Pattern 检测模型
基于 LLM 的 CVE 漏洞模式学习与检测模型。
## 📖 模型介绍
本模型用于从代码中抽象安全漏洞特征，并生成 Joern 查询语句用于漏洞检测。
### 模型文件
| 文件 | 说明 | 用途 |
|------|------|------|
| checkpoints_abstract_v2.zip | 抽象模型 | 从代码中提取漏洞特征 |
| checkpoints_generate_joern.zip | Joern 生成模型 | 生成 Joern 查询语句 |
## 🛠️ 技术细节
- **基础模型**: Mengzi/t5-base
- **框架**: MindSpore 2.0.0 + MindNLP 0.4.0
- **训练数据**: 2000+ 样本（来自 libpng, openssl, zlib）
## 🎯 支持检测
- Use After Free
- Buffer Overflow
- Integer Overflow
## 💻 使用方法
1. 下载模型文件并解压
2. 安装依赖：
```bash
pip install mindspore==2.0.0 mindnlp==0.4.0
```
3. 运行推理脚本
## 📂 GitHub 仓库
完整代码和使用说明：
👉 [https://github.com/BreakingBad6/BreakingGod-ai-vulnerability-pattern-learning](https://github.com/BreakingBad6/BreakingGod-ai-vulnerability-pattern-learning)
## 📄 License
MIT License |