mlsanityData to 7 Oct 2026

A Theoretical Insight into Attack and Defense of Gradient Leakage in Transformer

arXiv 2311.13624 ↗First posted 22 Nov 2023cs.LG cs.CRcited by 6

Not on the map. Why?

Abstract (arXiv)

The Deep Leakage from Gradient (DLG) attack has emerged as a prevalent and highly effective method for extracting sensitive training data by inspecting exchanged gradients. This approach poses a substantial threat to the privacy of individuals and organizations alike. This research presents a comprehensive analysis of the gradient leakage method when applied specifically to transformer-based models. Through meticulous examination, we showcase the capability to accurately recover data solely from gradients and rigorously investigate the conditions under which gradient attacks can be executed, providing compelling evidence. Furthermore, we reevaluate the approach of introducing additional noise on gradients as a protective measure against gradient attacks. To address this, we outline a theoretical proof that analyzes the associated privacy costs within the framework of differential privacy. Additionally, we affirm the convergence of the Stochastic Gradient Descent (SGD) algorithm under perturbed gradients. The primary objective of this study is to augment the understanding of gradient leakage attack and defense strategies while actively contributing to the development of privacy-preserving techniques specifically tailored for transformer-based models. By shedding light on the vulnerabilities and countermeasures associated with gradient leakage, this research aims to foster advancements in safeguarding sensitive data and upholding privacy in the context of transformer-based models.

Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 93 of its 156 references that are in the map. First posted on arXiv 22 Nov 2023.

Builds on

93 of 156 references.

References · 93

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
  4. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  5. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  6. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  7. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  8. 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
  9. 2020Scaling Laws for Neural Language Modelscited by 6,411
  10. 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
  11. 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,530
  12. 2022BLIP: Bootstrapping Language-Image Pre-training for Unified Vision-Language Understanding and Generationcited by 4,310
38 more
  1. 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
  2. 2022Training Compute-Optimal Large Language Modelscited by 2,927
  3. 2023Universal and Transferable Adversarial Attacks on Aligned Language Modelscited by 2,914
  4. 2023Sparks of Artificial General Intelligence: Early experiments with GPT-4cited by 2,579
  5. 2020Reformer: The Efficient Transformercited by 1,658
  6. 2023PaLM 2 Technical Reportcited by 1,154
  7. 2018Federated Learning with Non-IID Datacited by 1,099
  8. 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 784
  9. 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 696
  10. 2023A Watermark for Large Language Modelscited by 667
  11. 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 655
  12. 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 562
  13. 2023XSTest: A Test Suite for Identifying Exaggerated Safety Behaviours in Large Language Modelscited by 477
  14. 2017Differentially Private Federated Learning: A Client Level Perspectivecited by 451
  15. 2023SmoothLLM: Defending Large Language Models Against Jailbreaking Attackscited by 434
  16. 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 408
  17. 2023Trustworthy LLMs: a Survey and Guideline for Evaluating Large Language Models' Alignmentcited by 394
  18. 2020iDLG: Improved Deep Leakage from Gradientscited by 367
  19. 2014Memory Networkscited by 354
  20. 2019Differential Privacy Has Disparate Impact on Model Accuracycited by 316
  21. 2022Deduplicating Training Data Mitigates Privacy Risks in Language Modelscited by 309
  22. 2023Deja Vu: Contextual Sparsity for Efficient LLMs at Inference Timecited by 247
  23. 2021Evaluating Gradient Inversion Attacks and Defenses in Federated Learningcited by 196
  24. 2020Threats to Federated Learning: A Surveycited by 171
  25. 2023Representational Strengths and Limitations of Transformerscited by 132
  26. 2023ProPILE: Probing Privacy Leakage in Large Language Modelscited by 127
  27. 2019Polylogarithmic width suffices for gradient descent to achieve arbitrarily small test error with shallow ReLU networkscited by 120
  28. 2023A Theory for Emergence of Complex Skills in Language Modelscited by 111
  29. 2023Fast Attention Requires Bounded Entriescited by 102
  30. 2021FedGNN: Federated Graph Neural Network for Privacy-Preserving Recommendationcited by 97
  31. 2023On Provable Copyright Protection for Generative Modelscited by 97
  32. 2017Recovery Guarantees for One-hidden-layer Neural Networkscited by 94
  33. 2023Identifying and Mitigating the Security Risks of Generative AIcited by 90
  34. 2017One Model To Learn Them Allcited by 86
  35. 2023How Do Transformers Learn Topic Structure: Towards a Mechanistic Understandingcited by 85
  36. 2023HyperAttention: Long-context Attention in Near-Linear Timecited by 85
  37. 2021Federated Learning: Opportunities and Challengescited by 83
  38. 2019Attention Interpretability Across NLP Taskscited by 71

50 most cited of 93 All 93 (JSON)

Built on by

Cited by year

Citing papers per year: 2023: 1; 2024: 2; 2025: 3.

6 papers.