ml-security

仓库创建 2026年7月1日最近提交 23 天前SkillHot 收录 22 天前
▸ 精选理由

强调安全模型格式与溯源检查,适合 ML 工程与审计场景。

这个 Skill 做什么

涵盖模型加载、数据/模型投毒、PII 与模型血缘的安全建议。

帮你检查和防护模型相关的安全问题:怎么安全地加载模型(优先 safetensors、避免任意 .pkl),识别模型或数据被投毒、训练数据里泄露的 PII,以及笔记本里藏的密钥。会在拉第三方 checkpoint、做微调或把模型放到生产/Notebook 运行时用上。特点是给出具体可操作的加载与验证办法(如 weights_only、校验溯源/签名),把易被忽视的风险点说清楚。

▸ 展开 SKILL.md 英文原文

Model artifact loading (pickle vs safetensors), model & data poisoning, PII in training data, secrets in notebooks, model provenance / lineage

开发编程模型安全数据投毒供应链通用
2
Stars
0
Forks
40
仓库内 Skill
+0
7 日增星
安装 / 使用
给你的 Agent 一句话(通用)
帮我安装这个 skill:https://raw.githubusercontent.com/ShieldNet-360/secure-vibe/main/skills/ml-security/SKILL.md
或 curl 直取 SKILL.md
curl -fsSL "https://raw.githubusercontent.com/ShieldNet-360/secure-vibe/main/skills/ml-security/SKILL.md"
SKILL.MD 节选查看完整文件 ↗
# ML Model Security

## Rules (for AI agents)

### ALWAYS
- When loading models, use **safetensors** for PyTorch and Hugging Face; use
  `weights_only=True` with `torch.load` on PyTorch 2.4+; never load arbitrary
  `.pkl` / `.pt` files from untrusted sources.
- Verify **provenance / lineage** of any third-party or externally fine-tuned
  model — known author, signed or hashed checkpoint, recorded source — before
  loading it.
- Pin and hash **model + dataset versions** and record them, so a poisoned
  artifact can be traced and rolled back.
- Scrub PII, credentials, and secrets from training / fine-tuning data — at the
  source (ingestion), at storage (encryption + access control), and in an
via SKILL·HOT · 数据来自 GitHub 公开信息 · 原文版权归作者所有