OpenMined / PyDP

The Python Differential Privacy Library. Built on top of: https://github.com/google/differential-privacy
Apache License 2.0
497 stars 136 forks source link

Gaussian Mechanism Local DP for SKLearn Pipeline #387

Open grilhami opened 2 years ago

grilhami commented 2 years ago

Feature Description

Enabling the use of Gaussian Mechanism for Local DP in SKLean Pipeline. This should be in the form of an "operator" that should be inserted between the layers in SKLearn's Pipeline class.

Additional Context

Current development for the SKLean Pipeline is in branch feature/machine-learning-1.

Preferably, the name of the "operator" imported from PyDP should be called GaussianMechanism. The use should be as seamless and convenient as possible in SKLearn's Pipeline class. For example:

pipe = Pipeline([
    ('scaler', StandardScaler()),
    ('laplace', GaussianMechanism()), 
    ('nb', GaussianNB())
])

For more examples, please have look at the notebook example of Laplace Mechanism's implementation.

As starting guidance, please refer to the source code for LaplaceMechanism in here.

Note: the Gaussian Mechanism would most likely need different parameters to be initialized compare to the Laplace Mechanism.

duynht commented 12 months ago

Hi @grilhami. I'd like to work on this.