BMRO14/ML-Powered-AV-with-Image-Signatures

GitHub: BMRO14/ML-Powered-AV-with-Image-Signatures

基于机器学习的PE文件恶意软件检测原型,利用熵驱动的图像签名特征和Random Forest模型实现二分类与家族级多分类检测。

Stars: 0 | Forks: 0

# 基于机器学习的图像签名反病毒系统 一个轻量级的反病毒项目,通过静态特征、基于熵的可视化以及机器学习来分析可移植可执行(PE)文件。 该系统可以将文件分类为**恶意软件**或**良性**,同时支持家族级别的分类。 ## 工作原理 1. 用户选择一个文件。 2. 系统读取 PE 结构并提取静态信息。 3. 计算熵值并将其转换为类 RGB 图像签名。 4. 两个 Random Forest 模型对特征进行分析。 5. 应用程序返回带有置信度的双重预测结果(恶意软件 vs. 良性 / 家族归因)。 ## 主要特性 - PE 文件的静态分析。 - 基于熵的 RGB 图像签名。 - Random Forest 分类。 - 二分类和多分类检测。 - 简洁的反病毒风格界面。 ## 项目图片 在此处添加截图以展示处理流程和界面。 ### 完整工作流 ![项目工作流](https://static.pigsec.cn/wp-content/uploads/repos/cas/d4/d47513290b11c9ca60f1b040c726846e3257d3c5dddf4a05f9b48df205399b65.png) ### 主应用程序 ![主应用程序](https://static.pigsec.cn/wp-content/uploads/repos/cas/22/22c2aa7cc009a780f35f169c9155053a760100593622e67f40f840d2c0659d33.png) ### 扫描示例 ![恶意软件扫描](https://static.pigsec.cn/wp-content/uploads/repos/cas/19/197730f32b6a9e8a102fc95c162b1df6f96b355195e1d5f57df8ba70e5b975c2.png) ![良性扫描](https://static.pigsec.cn/wp-content/uploads/repos/cas/29/29797487cb5aca928d48978f18ffac0cd29861ffa3975e1363962d7deb58c7df.png) ## 数据集 该项目使用了恶意和良性 PE 样本,并进行了组织以供训练和评估。 特征从每个文件的起始部分提取,并被转换为 199 维的表示形式。 ## 结果 论文报告了二分类和多分类任务的强劲表现,该模型还提供了特征重要性的深入见解。 ## 环境要求 - Python - scikit-learn - NumPy - pandas - matplotlib - PE 解析 / 文件分析库 ## 作者 Vlad-Andrei Paraschiv
标签:Apex, Python, Windows PE文件, 云安全监控, 无后门, 机器学习, 逆向工具, 防病毒, 随机森林, 静态分析