Patch

README.md

@@ -134,7 +134,7 @@ You can also find more examples in our tutorials and documentation.
 | Attack        | Free-Rider             | [Delta-Weight](https://arxiv.org/pdf/1911.12560.pdf)                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                             |
 | Attack        | Evasion                | [Gradient-Descent Attack](https://arxiv.org/abs/1708.06131)                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                      |
 | Attack        | Membership Inference   | [Shaddow Attack](https://arxiv.org/abs/1610.05820)                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                               |
-| Defense       | Homomorphic Encryption | [Paiilier](https://link.springer.com/chapter/10.1007/3-540-48910-X_16), [CKKS](https://eprint.iacr.org/2016/421.pdf)                                                                                                                                                                                                                                                                                                                                                                                                                                             |
+| Defense       | Homomorphic Encryption | [Paiilier](https://link.springer.com/chapter/10.1007/3-540-48910-X_16)                                                                                                                                                                                                                                                                                                                                                                                                                                             |
 | Defense       | Differential Privacy   | [DPSGD](https://arxiv.org/abs/1607.00133), [AdaDPS](https://arxiv.org/pdf/2202.05963.pdf)                                                                                                                                                                                                                                                                                                                                                                                                                                                                        |
 | Defense       | Anonymization          | [Mondrian](https://ieeexplore.ieee.org/document/1617393)                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                         |
 | Defense       | Debugging              | [Model Assertions](https://cs.stanford.edu/~matei/papers/2019/debugml_model_assertions.pdf) |

docs/source/aijack.defense.debugging.rst

@@ -0,0 +1,29 @@
+aijack.defense.debugging package
+================================
+
+Submodules
+----------
+
+aijack.defense.debugging.assertions module
+------------------------------------------
+
+.. automodule:: aijack.defense.debugging.assertions
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+aijack.defense.debugging.utils module
+-------------------------------------
+
+.. automodule:: aijack.defense.debugging.utils
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+Module contents
+---------------
+
+.. automodule:: aijack.defense.debugging
+   :members:
+   :undoc-members:
+   :show-inheritance:

docs/source/aijack.defense.rst

@@ -7,7 +7,7 @@ Subpackages
 .. toctree::
    :maxdepth: 4
 
-   aijack.defense.ckks
+   aijack.defense.debugging
    aijack.defense.dp
    aijack.defense.foolsgold
    aijack.defense.kanonymity

src/aijack/attack/__init__.py

@@ -16,18 +16,3 @@
 )
 from .membership import ShadowMembershipInferenceAttack  # noqa: F401
 from .poison import Poison_attack_sklearn  # noqa: F401
-
-__all__ = [
-    "BaseAttacker",
-    "Evasion_attack_sklearn",
-    "MI_FACE",
-    "GANAttackClientManager",
-    "Generator_Attack",
-    "GradientInversion_Attack",
-    "GradientInversionAttackServerManager",
-    "attach_ganattack_to_client",
-    "NormAttackSplitNNManager",
-    "attach_normattack_to_splitnn",
-    "Poison_attack_sklearn",
-    "ShadowMembershipInferenceAttack",
-]

src/aijack/defense/__init__.py

@@ -1,6 +1,5 @@
 """Subpackage for defense algorithms for machine learning models.
 """
-from .ckks import CKKSEncoder, CKKSEncrypter, CKKSPlaintext  # noqa: F401
 from .dp import DPSGDManager, GeneralMomentAccountant  # noqa: F401
 from .mid import VIB, KL_between_normals, mib_loss  # noqa:F401
 from .paillier import PaillierGradientClientManager, PaillierKeyGenerator  # noqa: F401