Skip to content

Commit

Permalink
More stable loop termination criterion in fraction RDP order computat…
Browse files Browse the repository at this point in the history
…ion.

Previously the loop sometimes terminated too early, resulting in underestimates of the RDP at some orders. Now it will run to convergence in most cases, and in case of too many iterations, it will return an RDP of inf at that order, guaranteeing that the resulting epsilon is a true upper bound.

PiperOrigin-RevId: 668593123
  • Loading branch information
galenmandrew authored and tensorflower-gardener committed Sep 3, 2024
1 parent 66d05a2 commit 427d4ac
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ class ComputeDpSgdPrivacyTest(parameterized.TestCase):
@parameterized.named_parameters(
('Test0', 60000, 150, 1.3, 15, 1e-5, 0.7242234026109595, 19.0),
('Test1', 100000, 100, 1.0, 30, 1e-7, 1.4154988495444842, 13.0),
('Test2', 100000000, 1024, 0.1, 10, 1e-7, 5907982.3113984, 1.25),
('Test2', 100000000, 1024, 0.1, 10, 1e-7, 5907982.3113978775, 1.25),
)
def test_compute_dp_sgd_privacy(self, n, batch_size, noise_multiplier, epochs,
delta, expected_eps, expected_order):
Expand Down

0 comments on commit 427d4ac

Please sign in to comment.