On the Convergence and Calibration of Deep Learning with Differential Privacy

• Published in: Transactions on machine learning research Vol. 2023
• Main Authors: Bu, Zhiqi, Wang, Hua, Dai, Zongyu, Long, Qi
• Format: Journal Article
• Language: English
• Published: United States 01.06.2023
• ISSN: 2835-8856

Differentially private (DP) training preserves the data privacy usually at the cost of slower convergence (and thus lower accuracy), as well as more severe mis-calibration than its non-private counterpart. To analyze the convergence of DP training, we formulate a continuous time analysis through the lens of neural tangent kernel (NTK), which characterizes the per-sample gradient clipping and the noise addition in DP training, for arbitrary network architectures and loss functions. Interestingly, we show that the noise addition only affects the privacy risk but not the convergence or calibration, whereas the per-sample gradient clipping (under both flat and layerwise clipping styles) only affects the convergence and calibration. Furthermore, we observe that while DP models trained with small clipping norm usually achieve the best accurate, but are poorly calibrated and thus unreliable. In sharp contrast, DP models trained with large clipping norm enjoy the same privacy guarantee and similar accuracy, but are significantly more . Our code can be found at https://github.com/woodyx218/opacus_global_clipping.