Opacus
  • 简介
  • 常见问题
  • 教程
  • API 参考
  • GitHub
支持乌克兰 🇺🇦 帮助向乌克兰提供人道主义援助。

使用差分隐私训练 PyTorch 模型

简介
开始使用
教程

核心特性

Scalable logo

可扩展性

向量化逐样本梯度计算,比微批次(microbatching)快 10 倍

PyTorch logo

基于 PyTorch 构建

支持大多数类型的 PyTorch 模型,只需对原始神经网络进行极小的修改即可使用。

Extensible logo

可扩展性

开源、模块化的差分隐私研究 API。欢迎所有人贡献代码。

开始使用

  1. 安装 Opacus

    通过 pip
    pip install opacus
    
    通过 conda
    conda install -c conda-forge opacus
    
    从源码安装
    
    git clone https://github.com/pytorch/opacus.git
    cd opacus
    pip install -e .
                    
    
  2. 快速入门

    使用差分隐私进行训练就像实例化一个 PrivacyEngine 一样简单
    # define your components as usual
    model = Net()
    optimizer = SGD(model.parameters(), lr=0.05)
    data_loader = torch.utils.data.DataLoader(dataset, batch_size=1024)
    
    # enter PrivacyEngine
    privacy_engine = PrivacyEngine()
    model, optimizer, data_loader = privacy_engine.make_private(
        module=model,
        optimizer=optimizer,
        data_loader=data_loader,
        noise_multiplier=1.1,
        max_grad_norm=1.0,
    )
    # Now it's business as usual
    
Opacus
文档
简介常见问题教程API 参考
Github
opacus
法律
隐私政策条款
Meta Open Source
Copyright © 2025 Meta Platforms, Inc.