arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.02610cs.LGstat.ML
收藏

iDLG: Improved Deep Leakage from Gradients

School of Informatics(信息学院);The University of Edinburgh(爱丁堡大学)

Bo Zhao, Konda Reddy Mopuri, Hakan Bilen

英文摘要:

It is widely believed that sharing gradients will not leak private training data in distributed learning systems such as Collaborative Learning and Federated Learning, etc. Recently, Zhu et al. presented an approach which shows the possibility to obtain private training data from the publicly shared gradients. In their Deep Leakage from Gradient (DLG) method, they synthesize the dummy data and corresponding labels with the supervision of shared gradients. However, DLG has difficulty in convergence and discovering the ground-truth labels consistently. In this paper, we find that sharing gradients definitely leaks the ground-truth labels. We propose a simple but reliable approach to extract accurate data from the gradients. Particularly, our approach can certainly extract the ground-truth labels as opposed to DLG, hence we name it Improved DLG (iDLG). Our approach is valid for any differentiable model trained with cross-entropy loss over one-hot labels. We mathematically illustrate how our method can extract ground-truth labels from the gradients and empirically demonstrate the advantages over DLG.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑