Skip to content

Deep Leakage from Gradients show that it is possible to obtain the private training data from the publicly shared gradients, during machine learning operations. This project was developed in completion of CSCM38 Advanced Topics: Artificial Intelligence and Cyber Security Coursework 2.

Notifications You must be signed in to change notification settings

joshtowell/deep-leakage-from-gradients

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

Deep Leakage from Gradients

A project developed in completion of CSCM38 Advanced Topics: Artificial Intelligence and Cyber Security Coursework 2.

Cited source: https://dlg.mit.edu/

Background

"Exchanging gradients is a widely used method in modern multi-node machine learning system (e.g., distributed training, collaborative learning). For a long time, people believed that gradients are safe to share: i.e., the training data will not be leaked by gradient exchange.

However, we show that it is possible to obtain the private training data from the publicly shared gradients. We name this leakage as Deep Leakage from Gradient and empirically validate the effectiveness on both computer vision and natural language processing tasks. Experimental results show that our attack is much stronger than previous approaches: the recovery is pixel-wise accurate for images and token-wise matching for texts.

We want to raise people's awareness to rethink the gradient's safety. Finally, we discuss several possible strategies to prevent such deep leakage. The most effective defense method is gradient pruning."

More information to come (time dependent).

About

Deep Leakage from Gradients show that it is possible to obtain the private training data from the publicly shared gradients, during machine learning operations. This project was developed in completion of CSCM38 Advanced Topics: Artificial Intelligence and Cyber Security Coursework 2.

Topics

Resources

Stars

Watchers

Forks