626_privacy/tensorflow_privacy
2
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

Revert to dp-accounting 0.3.0 API.

Browse source 
PiperOrigin-RevId: 515432485
This commit is contained in:
Galen Andrew 2023-03-09 13:55:47 -08:00 committed by A. Unique TensorFlower
parent 61dfbcc1f5
commit 701a585e1a
1 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
tensorflow_privacy/privacy/analysis/compute_dp_sgd_privacy_lib.py
View file

@ -213,11 +213,15 @@ def _compute_dp_sgd_example_privacy(
else: else:
count = int(math.ceil(num_epochs)) count = int(math.ceil(num_epochs))
event_ = dp_accounting.SelfComposedDpEvent(count=count, event=event_) event_ = dp_accounting.SelfComposedDpEvent(count=count, event=event_)
return (
dp_accounting.rdp.RdpAccountant() # TODO(b/271341062) rdp_orders = (
.compose(event_) [1 + x / 10.0 for x in range(1, 100)]
.get_epsilon(example_delta) + list(range(11, 64))
+ [128, 256, 512, 1024]
) )
accountant = dp_accounting.rdp.RdpAccountant(rdp_orders) # TODO(b/271341062)
accountant.compose(event_)
return accountant.get_epsilon(example_delta)
def compute_dp_sgd_privacy(n, batch_size, noise_multiplier, epochs, delta): def compute_dp_sgd_privacy(n, batch_size, noise_multiplier, epochs, delta):
@ -269,4 +273,5 @@ def compute_dp_sgd_privacy(n, batch_size, noise_multiplier, epochs, delta):
steps, steps,
) )
return accountant.compose(event).get_epsilon_and_optimal_order(delta) accountant.compose(event)
return accountant.get_epsilon_and_optimal_order(delta)