opacus

收录理由

Opacus 为 PyTorch 项目接入差分隐私提供了一条务实的路径。你不需要重写训练代码，只需把现有的模型、优化器和数据加载器交给 PrivacyEngine，它会自动完成梯度裁剪和加噪，DP-SGD 的训练循环因此只多出几行代码。库在训练过程中持续统计累计隐私预算，让你能清楚回答监管方和用户最关心的问题：模型上线前这次运行消耗了多少隐私。配合快速逐样本梯度计算以及 Fast Gradient Clipping、Ghost Clipping 等裁剪方案，内存开销远低于朴素的微批次方法。模块化内部结构方便研究者自由改动，而教程则为工程人员提供了图像分类、文本分类和 LSTM 模型的可靠起点。