ruby/tensorflow_privacy
1
0
Fork 0
forked from 626_privacy/tensorflow_privacy
Code Pull requests Activity

Add a TODO comment for replacing noise saving with seed saving for tree aggregation.

Browse source 
PiperOrigin-RevId: 382338346
This commit is contained in:
Zheng Xu 2021-06-30 10:45:50 -07:00 committed by A. Unique TensorFlower
parent 3055f4ad52
commit c192a4166b
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tensorflow_privacy/privacy/dp_query/tree_aggregation.py
View file

@ -179,6 +179,11 @@ def get_step_idx(state: TreeState) -> tf.Tensor:
return step_idx return step_idx
# TODO(b/192464750): instead of saving noise `TreeState.level_buffer`, we can
# save the random seed and online generate noise. This could save some memory
# and disk.
class TreeAggregator(): class TreeAggregator():
"""Tree aggregator to compute accumulated noise in private algorithms. """Tree aggregator to compute accumulated noise in private algorithms.