Machine learning models are known to memorize private data to reduce their training loss, which can be inadvertently exploited by privacy attacks such as model inversion and membership inference. To protect against these attacks, differential privacy (DP) has become the de facto standard for privacy-preserving machine learning, particularly those popular training algorithms using stochastic gradient descent, such as DPSGD. Nonetheless, DPSGD still suffers from severe utility loss due to its slow convergence. This is partially caused by the random sampling, which brings bias and variance to the gradient, and partially by the Gaussian noise, which leads to fluctuation of gradient updates. Our key idea to address these issues is to apply selective updates to the model training, while discarding those useless or even harmful updates. Motivated by this, this paper proposes DPSUR, a Differentially Private training framework based on Selective Updates and Release, where the gradient from each iteration is evaluated based on a validation test, and only those updates leading to convergence are applied to the model. As such, DPSUR ensures the training in the right direction and thus can achieve faster convergence than DPSGD. The main challenges lie in two aspects -- privacy concerns arising from gradient evaluation, and gradient selection strategy for model update. To address the challenges, DPSUR introduces a clipping strategy for update randomization and a threshold mechanism for gradient selection. Experiments conducted on MNIST, FMNIST, CIFAR-10, and IMDB datasets show that DPSUR significantly outperforms previous works in terms of convergence speed and model utility.
翻译:机器学习模型已知会记忆私有数据以降低训练损失,这种特性可能被模型反转、成员推断等隐私攻击无意利用。为抵御此类攻击,差分隐私已成为隐私保护机器学习的事实标准,尤其适用于采用随机梯度下降的热门训练算法(如DPSGD)。然而,DPSGD仍因收敛缓慢而遭受严重的效用损失,这部分源于随机采样带来的梯度偏差与方差,部分源于高斯噪声导致的梯度更新波动。针对这些问题,我们的核心思想是对模型训练施加选择性更新,同时舍弃那些无用甚至有害的更新。基于此动机,本文提出DPSUR——一种基于选择性更新与发布的差分隐私训练框架,其中每次迭代的梯度通过验证测试评估,只有那些促进收敛的更新被应用于模型。由此,DPSUR确保训练沿正确方向进行,从而比DPSGD实现更快的收敛速度。主要挑战在于两个方面:梯度评估引发的隐私问题,以及模型更新的梯度选择策略。为应对这些挑战,DPSUR引入了用于更新随机化的裁剪策略和用于梯度选择的阈值机制。在MNIST、FMNIST、CIFAR-10和IMDB数据集上的实验表明,DPSUR在收敛速度和模型效用上显著优于以往工作。