Sparsity Preserving DP-SGD in TF Privacy [5 of 5]

Integrate sparsity preserving noise into DP Keras Model with fast gradient clipping.

See https://research.google/blog/sparsity-preserving-differentially-private-training/ for more details on the algorithm.

PiperOrigin-RevId: 648402434

tensorflow_privacy/privacy/fast_gradient_clipping/BUILD

@@ -43,8 +43,11 @@ py_library(
     srcs = ["gradient_clipping_utils.py"],
     srcs_version = "PY3",
     deps = [
+        ":common_manip_utils",
         ":layer_registry",
         ":type_aliases",
+        "//tensorflow_privacy/privacy/sparsity_preserving_noise:layer_registry",
+        "//tensorflow_privacy/privacy/sparsity_preserving_noise:type_aliases",
     ],
 )
 
@@ -54,10 +57,7 @@ py_test(
     python_version = "PY3",
     shard_count = 8,
     srcs_version = "PY3",
-    deps = [
-        ":gradient_clipping_utils",
-        ":type_aliases",
-    ],
+    deps = [":gradient_clipping_utils"],
 )
 
 py_library(
@@ -83,6 +83,12 @@ py_library(
     ],
 )
 
+py_library(
+    name = "noise_utils",
+    srcs = ["noise_utils.py"],
+    deps = ["//tensorflow_privacy/privacy/sparsity_preserving_noise:sparse_noise_utils"],
+)
+
 py_test(
     name = "clip_grads_test",
     srcs = ["clip_grads_test.py"],
@@ -92,7 +98,14 @@ py_test(
     deps = [
         ":clip_grads",
         ":common_test_utils",
+        ":gradient_clipping_utils",
         ":layer_registry",
         ":type_aliases",
     ],
 )
+
+py_test(
+    name = "noise_utils_test",
+    srcs = ["noise_utils_test.py"],
+    deps = [":noise_utils"],
+)