DpsgdOptimizer

class paddle.fluid.optimizer.DpsgdOptimizer(learning_rate=0.001, clip=0.9, batch_size=0.999, sigma=1e-8)[源代码]

Dpsgd优化器是参考CCS16论文 《Deep Learning with Differential Privacy》 相关内容实现的。

其参数更新的计算公式如下:

DpsgdOptimizer - 图1

DpsgdOptimizer - 图2

DpsgdOptimizer - 图3

参数

  • learning_rate (float|Variable,可选) - 学习率,用于参数更新的计算。可以是一个浮点型值或者一个值为浮点型的Variable,默认值为0.001
  • clip (float, 可选) - 裁剪梯度的L2正则项值的阈值下界,若梯度L2正则项值小于clip,则取clip作为梯度L2正则项值,默认值为0.9
  • batch_size (float, 可选) - 每个batch训练的样本数,默认值为0.999
  • sigma (float, 可选) - 参数更新时,会在梯度后添加一个满足高斯分布的噪声。此为高斯噪声的方差,默认值为1e-08

注解

目前 DpsgdOptimizer 不支持 Sparse Parameter Optimization(稀疏参数优化)。

代码示例

  1. import paddle.fluid as fluid
  2. import numpy
  3. # First create the Executor.
  4. place = fluid.CPUPlace() # fluid.CUDAPlace(0)
  5. exe = fluid.Executor(place)
  6. train_program = fluid.Program()
  7. startup_program = fluid.Program()
  8. with fluid.program_guard(train_program, startup_program):
  9. data = fluid.layers.data(name='X', shape=[1], dtype='float32')
  10. hidden = fluid.layers.fc(input=data, size=10)
  11. loss = fluid.layers.mean(hidden)
  12. optimizer = fluid.optimizer.Dpsgd(learning_rate=0.01, clip=10.0, batch_size=16.0, sigma=1.0)
  13. optimizer.minimize(loss)
  14. # Run the startup program once and only once.
  15. exe.run(startup_program)
  16. x = numpy.random.random(size=(10, 1)).astype('float32')
  17. outs = exe.run(program=train_program,
  18. feed={'X': x},
  19. fetch_list=[loss.name])

方法

minimize(loss, startup_program=None, parameter_list=None, no_grad_set=None)

为网络添加反向计算过程,并根据反向计算所得的梯度,更新parameter_list中的Parameters,最小化网络损失值loss。

参数

  • loss (Variable) – 需要最小化的损失值变量
  • startup_program (Program, 可选) – 用于初始化parameter_list中参数的 Program , 默认值为None,此时将使用 default_startup_program
  • parameter_list (list, 可选) – 待更新的Parameter或者Parameter.name组成的列表, 默认值为None,此时将更新所有的Parameter
  • no_grad_set (set, 可选) – 不需要更新的Parameter或者Parameter.name组成集合,默认值为None

返回

tuple(optimize_ops, params_grads),其中optimize_ops为参数优化OP列表;param_grads为由(param, param_grad)组成的列表,其中param和param_grad分别为参数和参数的梯度。该返回值可以加入到 Executor.run() 接口的 fetch_list 参数中,若加入,则会重写 use_prune 参数为True,并根据 feedfetch_list 进行剪枝,详见 Executor 的文档。

代码示例

  1. import numpy
  2. import paddle.fluid as fluid
  3. data = fluid.layers.data(name='X', shape=[1], dtype='float32')
  4. hidden = fluid.layers.fc(input=data, size=10)
  5. loss = fluid.layers.mean(hidden)
  6. adam = fluid.optimizer.Dpsgd(learning_rate=0.2)
  7. adam.minimize(loss)
  8. place = fluid.CPUPlace() # fluid.CUDAPlace(0)
  9. exe = fluid.Executor(place)
  10. x = numpy.random.random(size=(10, 1)).astype('float32')
  11. exe.run(fluid.default_startup_program())
  12. outs = exe.run(program=fluid.default_main_program(),
  13. feed={'X': x},
  14. fetch_list=[loss.name])