[Feature] Add serveral differential privacy algorithms

federatedscope/core/auxiliaries/model_builder.py

@@ -149,6 +149,12 @@ def get_model(model_config, local_data=None, backend='torch'):
     elif model_config.type.lower() in ['vmfnet', 'hmfnet']:
         from federatedscope.mf.model.model_builder import get_mfnet
         model = get_mfnet(model_config, input_shape)
+    elif model_config.type.lower() == 'fmlinearregression':
+        from federatedscope.differential_privacy.model.fm_linear_regression import FMLinearRegression
+        model = FMLinearRegression(in_channels=input_shape[-1], epsilon=0.5)
+    elif model_config.type.lower() == 'fmlogisticregression':
+        from federatedscope.differential_privacy.model.fm_logistic_regression import FMLogisticRegression
+        model = FMLogisticRegression(in_channels=input_shape[-1], epsilon=0.5)
     else:
         raise ValueError('Model {} is not provided'.format(model_config.type))
 

federatedscope/core/auxiliaries/optimizer_builder.py

@@ -3,6 +3,9 @@
 except ImportError:
     torch = None
 
+from federatedscope.differential_privacy.optimizers import *
+from federatedscope.register import optimizer_dict
+
 import copy
 
 
@@ -25,6 +28,12 @@ def get_optimizer(model, type, lr, **kwargs):
             else:
                 return getattr(torch.optim, type)(model, lr, **tmp_kwargs)
         else:
+            # registered optimizers
+            for func in optimizer_dict.values():
+                optimizer = func(type)
+                if optimizer is not None:
+                    return optimizer(model.parameters(), lr, **tmp_kwargs)
+
             raise NotImplementedError(
                 'Optimizer {} not implement'.format(type))
     else:

federatedscope/core/auxiliaries/regularizer_builder.py

@@ -1,5 +1,5 @@
 from federatedscope.register import regularizer_dict
-from federatedscope.core.regularizer.proximal_regularizer import *
+from federatedscope.core.regularizer import *
 try:
     from torch.nn import Module
 except ImportError:

federatedscope/core/auxiliaries/trainer_builder.py

@@ -25,6 +25,7 @@
     "fedvattrainer": "FedVATTrainer",
     "fedfocaltrainer": "FedFocalTrainer",
     "mftrainer": "MFTrainer",
+    "fmtrainer": "FMTrainer"
 }
 
 
@@ -80,6 +81,8 @@ def get_trainer(model=None,
             dict_path = "federatedscope.gfl.flitplus.trainer"
         elif config.trainer.type.lower() in ['mftrainer']:
             dict_path = "federatedscope.mf.trainer.trainer"
+        elif config.trainer.type.lower() == 'fmtrainer':
+            dict_path = "federatedscope.differential_privacy.trainers.fmtrainer"
         else:
             raise ValueError
 

federatedscope/core/regularizer/__init__.py

@@ -1 +1,4 @@
-from federatedscope.core.regularizer.proximal_regularizer import *
+from federatedscope.core.regularizer.proximal_regularizer import ProximalRegularizer
+from federatedscope.core.regularizer.l2_regularizer import L2Regularizer
+
+__all__ = ['ProximalRegularizer', 'L2Regularizer']

federatedscope/core/regularizer/l2_regularizer.py

@@ -0,0 +1,36 @@
+from federatedscope.register import register_regularizer
+from torch.nn import Module
+
+import torch
+
+REGULARIZER_NAME = "l2_regularizer"
+
+
+class L2Regularizer(Module):
+    """Returns the l2 norm of weight
+
+        Arguments:
+            p (int): The order of norm.
+            tensor_before: The original matrix or vector
+            tensor_after: The updated matrix or vector
+
+        Returns:
+            Tensor: the norm of the given udpate.
+    """
+    def __init__(self):
+        super(L2Regularizer, self).__init__()
+
+    def forward(self, ctx):
+        l2_norm = 0.
+        for param in ctx.model.parameters():
+            l2_norm += torch.sum(param**2)
+        return l2_norm
+
+
+def call_l2_regularizer(type):
+    if type == REGULARIZER_NAME:
+        regularizer = L2Regularizer
+        return regularizer
+
+
+register_regularizer(REGULARIZER_NAME, call_l2_regularizer)

federatedscope/differential_privacy/baseline/dpsgd_gaussian_on_femnist.yaml

@@ -0,0 +1,39 @@
+use_gpu: True
+device: 0
+early_stop:
+  patience: 5
+seed: 12345
+federate:
+  mode: standalone
+  total_round_num: 300
+  sample_client_rate: 0.2
+data:
+  root: data/
+  type: femnist
+  splits: [0.6,0.2,0.2]
+  batch_size: 10
+  subsample: 0.05
+  num_workers: 0
+  transform: [['ToTensor'], ['Normalize', {'mean': [0.1307], 'std': [0.3081]}]]
+model:
+  type: convnet2
+  hidden: 2048
+  out_channels: 62
+  dropout: 0.0
+train:
+  local_update_steps: 1
+  batch_or_epoch: epoch
+  optimizer:
+    type: DPGaussianSGD
+    lr: 0.01
+    l2_norm_sensitivity: 0.1
+    noise_multiplier: 0.1
+grad:
+  grad_clip: 5.0
+criterion:
+  type: CrossEntropyLoss
+trainer:
+  type: cvtrainer
+eval:
+  freq: 10
+  metrics: ['acc', 'correct']

federatedscope/differential_privacy/baseline/fm_linear_regression_on_toy.yaml

@@ -0,0 +1,27 @@
+use_gpu: True
+federate:
+  mode: 'standalone'
+  total_round_num: 500
+  client_num: 10
+seed: 12345
+trainer:
+  type: 'FMTrainer'
+train:
+  local_update_steps: 10
+  batch_or_epoch: 'batch'
+  optimizer:
+    type: 'SGD'
+    lr: 0.01
+eval:
+  freq: 20
+  metrics: ['loss_regular']
+  count_flops: False
+model:
+  type: 'FMLinearRegression'
+data:
+  type: 'toy'
+criterion:
+  type: MSELoss
+regularizer:
+  type: 'l2_regularizer'
+  mu: 0.01

federatedscope/differential_privacy/composition/compositor.py

@@ -0,0 +1,27 @@
+
+import numpy as np
+
+# For a client
+class AdvancedComposition(object):
+    pass
+
+
+class PrivacyAccountantComposition(object):
+    pass
+
+
+class RenyiComposition(object):
+    def __init__(self, sample_rate):
+        self.orders = [1.5, 1.75, 2, 2.5, 3, 4, 5, 6, 8, 16, 32, 64]
+
+        self.budgets = np.zeros_like(self.orders)
+
+        self.epsilon = 0
+
+        # sampling rate
+        # alpha rate
+
+    def compose(self, scale):
+        for i, order in enumerate(self.orders):
+            epsilon = order / scale ** 2
+            self.budgets[i] += epsilon

federatedscope/differential_privacy/model/__init__.py

@@ -0,0 +1,4 @@
+from federatedscope.differential_privacy.model.fm_linear_regression import FMLinearRegression
+from federatedscope.differential_privacy.model.fm_logistic_regression import FMLogisticRegression
+
+__all__ = ['FMLinearRegression', 'FMLogisticRegression']

federatedscope/differential_privacy/model/fm_linear_regression.py

@@ -0,0 +1,52 @@
+from torch.nn import Parameter
+from torch.nn import Module
+from torch.nn.init import kaiming_normal_
+
+import numpy as np
+
+import torch
+import math
+
+
+class FMLinearRegression(Module):
+    """Implementation of Functional Mechanism for linear regression refer to
+    `Functional Mechanism: Regression Analysis under Differential Privacy`
+    [Jun Wang, et al.](https://arxiv.org/abs/1208.0219)
+
+    Args:
+        in_channels (int): the number of dimensions
+        epsilon (int): the epsilon bound for differential privacy
+
+    Note:
+        The forward function returns the average loss directly, so that we
+        don't need the criterion function for fm linear regression.
+    """
+    def __init__(self, in_channels, epsilon):
+        super(FMLinearRegression, self).__init__()
+        self.w = Parameter(torch.empty(in_channels, 1))
+        kaiming_normal_(self.w, a=math.sqrt(5))
+
+        sensitivity = float(2*(1+2*in_channels+in_channels**2))
+
+        self.laplace = torch.distributions.laplace.Laplace(loc=0, scale=sensitivity / epsilon * np.sqrt(2))
+
+    def forward(self, x, y):
+        # J=0
+        lambda0 = torch.matmul(y.t(), y)
+        lambda0 += self.laplace.sample(sample_shape=lambda0.size()).to(lambda0.device)
+        # J=1
+        lambda1 = -2 * torch.matmul(y.t(), x)
+        lambda1 += self.laplace.sample(sample_shape=lambda1.size()).to(lambda1.device)
+        # J=2
+        lambda2 = torch.matmul(x.t(), x)
+        lambda2 += self.laplace.sample(sample_shape=lambda2.size()).to(lambda2.device)
+        w2 = torch.matmul(self.w, self.w.t())
+
+        loss_total = lambda0 + torch.sum(lambda1.t() * self.w) + torch.sum(lambda2*w2)
+
+        pred = torch.matmul(x, self.w)
+
+        return pred, loss_total / x.size(0)
+
+
+

federatedscope/differential_privacy/model/fm_logistic_regression.py

@@ -0,0 +1,49 @@
+from torch.nn import Parameter
+from torch.nn import Module
+from torch.nn.init import kaiming_normal_
+
+import numpy as np
+
+import torch
+
+
+class FMLogisticRegression(Module):
+    """Implementation of Functional Mechanism for logistic regression refer to
+        `Functional Mechanism: Regression Analysis under Differential Privacy`
+        [Jun Wang, et al.](https://arxiv.org/abs/1208.0219)
+
+        Args:
+            in_channels (int): the number of dimensions
+            epsilon (int): the epsilon bound for differential privacy
+
+        Note:
+            The forward function returns the average loss directly, so that we
+            don't need the criterion function for fm logistic regression.
+    """
+    def __init__(self, in_channels, epsilon):
+        super(FMLogisticRegression, self).__init__()
+        self.w = Parameter(torch.empty(in_channels, 1))
+        kaiming_normal_(self.w)
+
+        sensitivity = 0.25 * in_channels ** 2 + 3 * in_channels
+        self.laplace = torch.distributions.laplace.Laplace(loc=0, scale=sensitivity / epsilon * np.sqrt(2))
+
+    def forward(self, x, y):
+        if len(y.size()) == 1:
+            y = torch.unsqueeze(y, dim=-1)
+        # J=0
+        lambda0 = np.log(2)
+        lambda0 += self.laplace.sample(sample_shape=[1]).to(x.device)
+        # J=1
+        lambda1 = 0.5 * x - y * x
+        lambda1 += self.laplace.sample(sample_shape=lambda1.size()).to(lambda1.device)
+        # J=2
+        lambda2 = torch.matmul(x.t(), x)
+        lambda2 += self.laplace.sample(sample_shape=lambda2.size()).to(lambda2.device)
+        w2 = torch.matmul(self.w, self.w.t())
+
+        loss_total = lambda0 * x.size(0) + torch.sum(lambda1.t() * self.w) + 0.125 * torch.sum(lambda2 * w2)
+
+        pred = torch.matmul(x, self.w)
+
+        return pred, loss_total / x.size(0)

federatedscope/differential_privacy/optimizers/__init__.py

@@ -0,0 +1,3 @@
+from federatedscope.differential_privacy.optimizers.dp_optimizer import DPGaussianSGD, DPLaplaceSGD
+
+__all__ = ['DPGaussianSGD', 'DPLaplaceSGD']