Defense¶

Adding New Methods¶

1. Choose the Correct Category¶

Place defense methods under models/defenses/
All methods should inherit from base classes in models/base/

2. Implementation Guidelines¶

Create a new file for your method (e.g., models/defense/your_method.py)
Inherit from BaseDefense as appropriate
Implement required interface methods:
- For defenses: protect(), verify()

3. Code Integration¶

Add your model class to __init__.py in the respective directory
Add any new metrics to utils/metrics.py
Add any new neural network architectures to utils/models.py
If needed, extend dataset functionality in datasets/datasets.py
It would be better if you could package your code into one class and add to defense.py
Add a parameter or an entry in order to import one attack model from other papers instead of training your own attack model

Example Structure¶

# models/defense/your_method.py
from ..base.defense import BaseDefense

class YourDefense(BaseDefense):
    def __init__(self, args):
        super().__init__()
        self.args = args

    def train(self, victim_model, training_data):
        # Implementation
        pass

    def extract(self, query_data):
        # Implementation
        pass