=====================================================
This project aims to certify the robustness of machine learning models against adaptive online data poisoning attacks.
To install the dependencies, run the following command: pip install -r requirements.txt
The generalized_certificate.py module provides an abstract class for computing certificates and using them to meta-learn a robust learning algorithm. This class serves as a foundation for implementing specific certification methods.
MeanEstimation/mean_estimation_toy.py: Provides a concrete implementation of the certificate abstract class for the mean estimation task. Classification/classification_toy.py: Provides a concrete implementation of the certificate abstract class for the classification task.
To perform mean estimation, follow these steps:
- Navigate to the
MeanEstimationdirectory: cd MeanEstimation - Run the following script: python3 mean_estimation.py
To perform classification, follow these steps:
- Download the MNIST dataset (or any corresponding dataset) and store the dataset in the
datafolder. - Navigate to the
Classificationdirectory: cd Classification - Run the following script: python3 mnist.py