APT-Malware-Classification

KO | EN

gitlite — search

#python #go #golang #android #deep-learning #docker #macos #java #kubernetes #awesome-list #vue #ios #openai #open-source

APT-Malware-Classification

(★ 10)

自动判定恶意样本所属 APT 组织

APT-Malware-Classification 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

```
 ╔═╗ ╔═╗ ╔╦╗      ╔╦╗ ╔═╗ ╦   ╦ ╦ ╔═╗ ╦═╗ ╔═╗      ╔═╗ ╦   ╔═╗ ╔═╗ ╔═╗ ╦ ╔═╗ ╦ ╔═╗ ╔═╗ ╔╦╗ ╦ ╔═╗ ╔╗╔
 ╠═╣ ╠═╝  ║       ║║║ ╠═╣ ║   ║║║ ╠═╣ ╠╦╝ ║╣       ║   ║   ╠═╣ ╚═╗ ╚═╗ ║ ╠╣  ║ ║   ╠═╣  ║  ║ ║ ║ ║║║
 ╩ ╩ ╩    ╩       ╩ ╩ ╩ ╩ ╩═╝ ╚╩╝ ╩ ╩ ╩╚═ ╚═╝      ╚═╝ ╩═╝ ╩ ╩ ╚═╝ ╚═╝ ╩ ╚   ╩ ╚═╝ ╩ ╩  ╩  ╩ ╚═╝ ╝╚╝
```

# 前言

利用 [tf-idf](https://zh.m.wikipedia.org/zh-cn/Tf-idf) 和 [xgboost](https://zh.wikipedia.org/zh-cn/XGBoost) 对恶意样本的 API 调用序列进行数据分析，形成机器学习模型，自动判定恶意样本所属 APT 组织。

# 数据集

全部数据集都在 Cuckoo Report.zip 里面，以各 APT 组织分类，各 APT 组织的分析报告从 [VirusTotal](https://www.virustotal.com) 获取。由于  VirusTotal api 有限制，所以考虑我的另外一个项目 [基于 sha256 使用 selenium 爬取 Virustotal 获得文件类型与行为特征文件](https://github.com/yym68686/VirusTotalSpider) 。

# 项目目录说明

```
APT_MALWARE_CLASSIFICATION
│  All_Windows_API.json   # 所有 windows api
│  Cuckoo Report.zip      # 所有从 VirusTotal 爬取的行为特征文件
│  feature_engineering.py # 将行为特征文件转化为 pkl 文件
│  overview.csv           # 恶意样本对应的 sha256
│  README.md
│  test.py
│  xgboost_model.py       # 训练程序
│
└─API Sequence            # 所有 pkl 文件
```

# 训练步骤

将数据集转化为 pkl 文件。

```bash
python feature_engineering.py
```

运行后将自动创建 API Sequence 文件夹，生成 pkl 文件。开始训练：

```bash
python xgboost_model.py
```

采用五折交叉验证方法对模型进行评估，对于部分 APT 组织的聚类结果如表所示。

| APT Group    | precision | recall | F1-score | 报告数 |
| ------------ | --------- | ------ | -------- | ------ |
| APT1         | 0.87      | 0.90   | 0.89     | 252    |
| APT29        | 0.97      | 0.80   | 0.88     | 128    |
| APT30        | 0.90      | 0.88   | 0.89     | 149    |
| Dark Hotel   | 0.84      | 0.86   | 0.85     | 196    |
| Gorgon Group | 0.93      | 0.97   | 0.95     | 208    |
| Winnti       | 0.92      | 0.91   | 0.91     | 162    |

APT-Malware-Classification

APT-Malware-Classification 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

APT-Malware-Classification

APT-Malware-Classification 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README