Breakingbad6 commited on
Commit
fd0bbda
·
verified ·
1 Parent(s): 43f5bc5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +61 -3
README.md CHANGED
@@ -1,3 +1,61 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
4
+ ---
5
+ language:
6
+ - zh
7
+ - en
8
+ tags:
9
+ - vulnerability-detection
10
+ - security
11
+ - code-analysis
12
+ - mindspore
13
+ - t5
14
+ license: mit
15
+ ---
16
+
17
+ # BreakingGod - CVE Bug Pattern 检测模型
18
+
19
+ 基于 LLM 的 CVE 漏洞模式学习与检测模型。
20
+
21
+ ## 📖 模型介绍
22
+
23
+ 本模型用于从代码中抽象安全漏洞特征,并生成 Joern 查询语句用于漏洞检测。
24
+
25
+ ### 模型文件
26
+
27
+ | 文件 | 说明 | 用途 |
28
+ |------|------|------|
29
+ | checkpoints_abstract_v2.zip | 抽象模型 | 从代码中提取漏洞特征 |
30
+ | checkpoints_generate_joern.zip | Joern 生成模型 | 生成 Joern 查询语句 |
31
+
32
+ ## 🛠️ 技术细节
33
+
34
+ - **基础模型**: Mengzi/t5-base
35
+ - **框架**: MindSpore 2.0.0 + MindNLP 0.4.0
36
+ - **训练数据**: 2000+ 样本(来自 libpng, openssl, zlib)
37
+
38
+ ## 🎯 支持检测
39
+
40
+ - Use After Free
41
+ - Buffer Overflow
42
+ - Integer Overflow
43
+
44
+ ## 💻 使用方法
45
+
46
+ 1. 下载模型文件并解压
47
+ 2. 安装依赖:
48
+ ```bash
49
+ pip install mindspore==2.0.0 mindnlp==0.4.0
50
+ ```
51
+ 3. 运行推理脚本
52
+
53
+ ## 📂 GitHub 仓库
54
+
55
+ 完整代码和使用说明:
56
+
57
+ 👉 [https://github.com/BreakingBad6/BreakingGod-ai-vulnerability-pattern-learning](https://github.com/BreakingBad6/BreakingGod-ai-vulnerability-pattern-learning)
58
+
59
+ ## 📄 License
60
+
61
+ MIT License