免费、开源的 .keras 和 pickle 模型文件扫描器(适用于 .pt, .bin, .joblib, .pkl, .npy 文件),无需加载模型

📌 One-Sentence Summary
keras-model-scan 和 pickle-model-scan 是用于静态分析模型文件的工具,通过读取字节来检测安全漏洞,而无需加载或执行模型,支持 .keras, .pkl, .joblib, .pt 等格式。
📝 Summary
本文介绍了两个开源的静态扫描工具:keras-model-scan 用于 Keras 文件,pickle-model-scan 用于 pickle 格式文件。这些工具通过分析模型文件中的字节来识别安全风险,例如不安全的反序列化、路径遍历和危险的导入,从而为运行模型之前提供一种安全的审查方式。
💡 Main Points
模型安全静态分析
扫描器通过读取模型文件的原始字节来执行静态分析,从而检测潜在的安全漏洞,例如 Keras 中的不安全反序列化或 pickle 文件中的危险导入,所有这些检查都在不执行模型代码的情况下完成。
支持多种格式
这些工具支持广泛的模型文件格式,包括 .keras, .pkl, .joblib, .pt 和 .npy,这意味着可以对不同框架的模型进行全面的安全检查。
集成与报告
扫描器可以通过 GitHub Actions 集成到 CI/CD 管道中,并以 JSON 和 SARIF 等标准格式输出结果,从而便于跟踪和报告安全问题。
💬 Key Quotes
这些工具都不会加载、导入、反序列化或执行任何东西。它们只读取字节。
pickle 不是一种数据格式。它是一个小型的、基于栈的程序,由反序列化器运行。
扫描器可以通过 GitHub Actions 集成到 CI/CD 管道中,并以 JSON 和 SARIF 等标准格式输出结果。
📊 Article Meta
AI Screening: 88
Source: DEV Community: machinelearning
Author: munimv2
Category: 人工智能
Language: 英文
Read Time: 3 min
Word Count: 720
Tags:
AI 与智能应用 , 安全 , AI 工程 , AI & 智能应用 , 模型训练 & 推理
暂无评论,快来抢沙发~