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