foolbox

GitHub
3k 439 简单 1 次阅读 3天前MIT开发框架
AI 解读 由 AI 自动生成,仅供参考

Foolbox 是一款专为评估机器学习模型鲁棒性而设计的 Python 开源库,能够帮助用户轻松生成“对抗样本”,即那些经过细微修改却能误导神经网络做出错误判断的数据。它主要解决了深度学习模型在面对恶意攻击时安全性难以量化测试的痛点,让研究人员和开发者能够便捷地验证模型在 PyTorch、TensorFlow 和 JAX 等主流框架下的防御能力。

这款工具特别适合人工智能领域的研究人员、算法工程师以及关注模型安全性的开发者使用。Foolbox 的核心亮点在于其基于 EagerPy 重构的架构,实现了真正的原生性能支持。这意味着它无需在不同框架间进行繁琐的代码转换或重复开发,即可直接利用各框架的原生计算能力(包括真实的批量处理),从而大幅提升攻击算法的运行效率。此外,Foolbox 集成了大量业界领先的梯度基于和决策基于的攻击算法,并提供了完善的类型检查机制,帮助用户在代码运行前发现潜在错误。无论是进行学术研究还是工业级的模型压力测试,Foolbox 都能提供高效、统一且可靠的解决方案。

使用场景

某自动驾驶初创公司的算法团队正在对自研的交通标志识别模型进行安全审计,急需验证其在对抗样本攻击下的鲁棒性。

没有 foolbox 时

  • 框架适配成本高:团队混合使用了 PyTorch 和 TensorFlow,手动为不同框架重写攻击代码导致大量重复劳动,维护极其困难。
  • 攻击实现复杂:复现前沿的梯度攻击算法(如 PGD)需要深入推导数学公式并处理复杂的张量运算,极易引入隐蔽的 Bug。
  • 评估效率低下:缺乏原生批处理支持,只能单张图片串行生成对抗样本,耗时数天才能完成一次完整的鲁棒性基准测试。
  • 调试困难:由于缺乏类型检查,张量维度不匹配等错误往往在运行很久后才爆发,排查问题耗费大量精力。

使用 foolbox 后

  • 统一代码底座:借助 EagerPy 底层支持,同一套攻击代码可直接无缝运行于 PyTorch、TensorFlow 和 JAX 模型,彻底消除框架隔阂。
  • 开箱即用算法:直接调用内置的 LinfPGD 等状态-of-the-art 攻击接口,几行代码即可发起高强度攻击,无需关注底层数学实现。
  • 原生高性能加速:利用真实的批处理(Batch)支持并行生成对抗样本,将原本数天的测试周期缩短至小时级,大幅迭代速度。
  • 开发更稳健:得益于完善的类型注解,能在编码阶段提前捕获维度错误,显著降低运行时崩溃风险,让工程师更专注于策略分析。

foolbox 通过提供统一、高效且原生的对抗攻击基准测试能力,帮助团队以最小成本构建了可靠的模型防御防线。

运行环境要求

GPU

未说明(取决于所选用的深度学习框架 PyTorch/TensorFlow/JAX 及其对应的 CUDA 版本)

内存

未说明

依赖
notesFoolbox 本身不将 PyTorch、TensorFlow 或 JAX 列为强制依赖,用户需根据实际需求单独安装其中一个或多个框架。不同框架对系统架构和 CUDA 版本有特定要求。当前测试使用的框架版本为:PyTorch 1.10.1, TensorFlow 2.6.3, JAX 0.2.517, NumPy 1.18.1。
python3.8+ (测试环境为 3.8,推测支持 3.6-3.8)
eagerpy
PyTorch (可选,需单独安装)
TensorFlow (可选,需单独安装)
JAX (可选,需单独安装)
NumPy
foolbox hero image

快速开始

.. raw:: html

.. image:: https://badge.fury.io/py/foolbox.svg :target: https://badge.fury.io/py/foolbox

.. image:: https://readthedocs.org/projects/foolbox/badge/?version=latest :target: https://foolbox.readthedocs.io/en/latest/

.. image:: https://img.shields.io/badge/code%20style-black-000000.svg :target: https://github.com/ambv/black

.. image:: https://joss.theoj.org/papers/10.21105/joss.02607/status.svg :target: https://doi.org/10.21105/joss.02607

=============================================================================================================================== Foolbox:用于在 PyTorch、TensorFlow 和 JAX 中基准机器学习模型鲁棒性的快速对抗攻击工具

Foolbox <https://foolbox.jonasrauber.de>_ 是一个 Python 库,可让您轻松地对深度神经网络等机器学习模型执行对抗攻击。它基于 EagerPy 构建,原生支持 PyTorch <https://pytorch.org>TensorFlow <https://www.tensorflow.org>JAX <https://github.com/google/jax>_ 中的模型。

🔥 设计

Foolbox 3 已经从头重写, 使用 EagerPy <https://github.com/jonasrauber/eagerpy>_ 替代 NumPy, 以实现对 PyTorch、TensorFlow 和 JAX 中开发模型的原生性能, 且仅需一套代码库即可完成,无需重复编写。

  • 原生性能:Foolbox 3 基于 EagerPy 构建,在 PyTorch、TensorFlow 和 JAX 中原生运行,并真正支持批量处理。
  • 最先进攻击:Foolbox 提供大量最先进的基于梯度和基于决策的对抗攻击方法。
  • 类型检查:借助 Foolbox 中丰富的类型注解,在运行代码之前即可捕获错误。

📖 文档

  • 指南:开始使用 Foolbox 的最佳地点是官方 指南 <https://foolbox.jonasrauber.de>_。
  • 教程:如果您正在寻找教程,请查看此 Jupyter 笔记本 <https://github.com/jonasrauber/foolbox-native-tutorial/blob/master/foolbox-native-tutorial.ipynb>_ |colab|。
  • 文档:API 文档可在 ReadTheDocs <https://foolbox.readthedocs.io/en/stable/> 上找到。

.. |colab| image:: https://colab.research.google.com/assets/colab-badge.svg :target: https://colab.research.google.com/github/jonasrauber/foolbox-native-tutorial/blob/master/foolbox-native-tutorial.ipynb

🚀 快速入门

.. code-block:: bash

pip install foolbox

Foolbox 经过 Python 3.8 及更高版本的测试;不过,它很可能也能在 3.6 至 3.8 版本中正常工作。要将其与 PyTorch <https://pytorch.org>TensorFlow <https://www.tensorflow.org>JAX <https://github.com/google/jax>_ 一起使用,需要分别安装相应的框架。这些框架并未被声明为依赖项,因为并非所有人都希望同时使用并安装它们,而且其中一些软件包针对不同的架构和 CUDA 版本有不同的构建版本。除此之外,所有必要的依赖项都会自动安装。

我们当前用于测试的版本可在下方的 兼容性部分 <#-compatibility>_ 中查看,但通常情况下,更新的版本也应能正常工作。

🎉 示例

.. code-block:: python

import foolbox as fb

model = ... fmodel = fb.PyTorchModel(model, bounds=(0, 1))

attack = fb.attacks.LinfPGD() epsilons = [0.0, 0.001, 0.01, 0.03, 0.1, 0.3, 0.5, 1.0] _, advs, success = attack(fmodel, images, labels, epsilons=epsilons)

更多示例可在 examples <./examples/> 文件夹中找到,例如完整的 ResNet-18 示例 <./examples/single_attack_pytorch_resnet18.py>_。

📄 引用

如果您在工作中使用了 Foolbox,请引用我们的 JOSS 关于 Foolbox Native(即 Foolbox 3.0)的文章 <https://doi.org/10.21105/joss.02607>_ 以及我们在 ICML 研讨会发表的关于 Foolbox 的论文 <https://arxiv.org/abs/1707.04131>_,引用格式如下:

.. code-block::

@article{rauber2017foolboxnative, doi = {10.21105/joss.02607}, url = {https://doi.org/10.21105/joss.02607}, year = {2020}, publisher = {The Open Journal}, volume = {5}, number = {53}, pages = {2607}, author = {Jonas Rauber 和 Roland Zimmermann、Matthias Bethge、Wieland Brendel}, title = {Foolbox Native:用于在 PyTorch、TensorFlow 和 JAX 中基准机器学习模型鲁棒性的快速对抗攻击工具}, journal = {Journal of Open Source Software} }

.. code-block::

@inproceedings{rauber2017foolbox, title={Foolbox:用于基准机器学习模型鲁棒性的 Python 工具箱}, author={Rauber, Jonas 和 Brendel, Wieland、Bethge, Matthias}, booktitle={Reliable Machine Learning in the Wild Workshop,第 34 届国际机器学习大会}, year={2017}, url={http://arxiv.org/abs/1707.04131}, }

👍 贡献

我们欢迎各种形式的贡献,请参阅我们的 开发指南 <https://foolbox.jonasrauber.de/guide/development.html>。特别是,欢迎您贡献 新的对抗攻击 <https://foolbox.jonasrauber.de/guide/adding_attacks.html>。如果您想提供帮助,也可以查看标记为 欢迎贡献 <https://github.com/bethgelab/foolbox/issues?q=is%3Aopen+is%3Aissue+label%3A%22contributions+welcome%22>_ 的问题。

💡 有问题吗?

如果您有任何问题或需要帮助,欢迎在 GitHub 上提交一个问题。待 GitHub Discussions 公开可用后,我们将切换到该平台。

💨 性能

Foolbox 3.0 比 Foolbox 1 和 2 快得多。基本的 性能比较_ 可在 performance 文件夹中找到。

🐍 兼容性

我们目前使用的版本如下:

  • PyTorch 1.10.1
  • TensorFlow 2.6.3
  • JAX 0.2.517
  • NumPy 1.18.1

.. _性能比较: performance/README.md

版本历史

v3.3.42024/03/04
v3.3.32022/04/02
v3.3.22022/03/08
v3.3.12021/02/23
v3.3.02021/02/10
v3.2.12020/09/26
v3.2.02020/09/26
v3.1.12020/08/29
v3.1.02020/08/29
v3.0.42020/07/03
v3.0.32020/07/03
v3.0.22020/05/23
v3.0.12020/05/23
v3.0.02020/03/22
v3.0.0b12020/02/16
v3.0.0b02020/02/15
v2.4.02020/02/07
v2.3.02019/11/04
v2.2.02019/10/28
v2.1.02019/10/27

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|2天前
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|3天前
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

144.7k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

107.9k|★★☆☆☆|2天前
开发框架图像Agent

markitdown

MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器

93.4k|★★☆☆☆|昨天
插件开发框架

LLMs-from-scratch

LLMs-from-scratch 是一个基于 PyTorch 的开源教育项目,旨在引导用户从零开始一步步构建一个类似 ChatGPT 的大型语言模型(LLM)。它不仅是同名技术著作的官方代码库,更提供了一套完整的实践方案,涵盖模型开发、预训练及微调的全过程。 该项目主要解决了大模型领域“黑盒化”的学习痛点。许多开发者虽能调用现成模型,却难以深入理解其内部架构与训练机制。通过亲手编写每一行核心代码,用户能够透彻掌握 Transformer 架构、注意力机制等关键原理,从而真正理解大模型是如何“思考”的。此外,项目还包含了加载大型预训练权重进行微调的代码,帮助用户将理论知识延伸至实际应用。 LLMs-from-scratch 特别适合希望深入底层原理的 AI 开发者、研究人员以及计算机专业的学生。对于不满足于仅使用 API,而是渴望探究模型构建细节的技术人员而言,这是极佳的学习资源。其独特的技术亮点在于“循序渐进”的教学设计:将复杂的系统工程拆解为清晰的步骤,配合详细的图表与示例,让构建一个虽小但功能完备的大模型变得触手可及。无论你是想夯实理论基础,还是为未来研发更大规模的模型做准备

90.1k|★★★☆☆|2天前
语言模型图像Agent