DiVa: An Accelerator for Differentially Private Machine Learning
Not on the map. Why?
Abstract (arXiv)
The widespread deployment of machine learning (ML) is raising serious concerns on protecting the privacy of users who contributed to the collection of training data. Differential privacy (DP) is rapidly gaining momentum in the industry as a practical standard for privacy protection. Despite DP's importance, however, little has been explored within the computer systems community regarding the implication of this emerging ML algorithm on system designs. In this work, we conduct a detailed workload characterization on a state-of-the-art differentially private ML training algorithm named DP-SGD. We uncover several unique properties of DP-SGD (e.g., its high memory capacity and computation requirements vs. non-private ML), root-causing its key bottlenecks. Based on our analysis, we propose an accelerator for differentially private ML named DiVa, which provides a significant improvement in compute utilization, leading to 2.6x higher energy-efficiency vs. conventional systolic arrays.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 35 of its 89 references that are in the map. First posted on arXiv 26 Aug 2022.
Builds on
35 of 89 references.
References · 35
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,845
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2016Deep Learning with Differential Privacycited by 2,728
- 2016Membership Inference Attacks against Machine Learning Modelscited by 2,281
- 2020Extracting Training Data from Large Language Modelscited by 2,189
- 2016SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model sizecited by 1,060
- 2017In-Datacenter Performance Analysis of a Tensor Processing Unitcited by 618
- 2016Learning to Communicate with Deep Multi-Agent Reinforcement Learningcited by 575
- 2022Using DeepSpeed and Megatron to Train Megatron-Turing NLG 530B, A Large-Scale Generative Language Modelcited by 559
23 more
- 2021Large Language Models Can Be Strong Differentially Private Learnerscited by 412
- 2021Differentially Private Fine-tuning of Language Modelscited by 400
- 2016Semi-supervised Knowledge Transfer for Deep Learning from Private Training Datacited by 348
- 2016EIE: Efficient Inference Engine on Compressed Deep Neural Networkcited by 295
- 2014cuDNN: Efficient Primitives for Deep Learningcited by 245
- 2019A Study of BFLOAT16 for Deep Learning Trainingcited by 232
- 2017SCNN: An Accelerator for Compressed-sparse Convolutional Neural Networkscited by 135
- 2021Large-Scale Differentially Private BERTcited by 118
- 2014Differentially Private Empirical Risk Minimization: Efficient Algorithms and Tight Error Boundscited by 107
- 2022Toward Training at ImageNet Scale with Differential Privacycited by 87
- 2020Enabling Fast Differentially Private SGD via Just-in-Time Compilation and Vectorizationcited by 58
- 2020SpArch: Efficient Architecture for Sparse Matrix Multiplicationcited by 49
- 2020Scaling up Differentially Private Deep Learning with Fast Per-Example Gradient Clippingcited by 47
- 2019TensorDIMM: A Practical Near-Memory Processing Architecture for Embeddings and Tensor Operations in Deep Learningcited by 43
- 2017Compressing DMA Engine: Leveraging Activation Sparsity for Training Deep Neural Networkscited by 37
- 2020ConfuciuX: Autonomous Hardware Resource Assignment for DNN Accelerators using Reinforcement Learningcited by 35
- 2019PREMA: A Predictive Multi-task Scheduling Algorithm For Preemptible Neural Processing Unitscited by 30
- 2020Centaur: A Chiplet-based, Hybrid Sparse-Dense Accelerator for Personalized Recommendationscited by 21
- 2022DarKnight: An Accelerated Framework for Privacy and Integrity Preserving Deep Learning Using Trusted Hardwarecited by 14
- 2020Tensor Casting: Co-Designing Algorithm-Architecture for Personalized Recommendation Trainingcited by 13
- 2019Beyond the Memory Wall: A Case for Memory-centric HPC System for Deep Learningcited by 9
- 2017Dropping Activation Outputs with Localized First-layer Deep Network for Enhancing User Privacy and Data Securitycited by 8
- 2019NeuMMU: Architectural Support for Efficient Address Translations in Neural Processing Unitscited by 4
Built on by
Citing papers per year: 2024: 2; 2025: 2.
4 papers.
- 2024Debunking the CUDA Myth Towards GPU-based AI Systemscited by 2
- 2024LazyDP: Co-Designing Algorithm-Software for Scalable Training of Differentially Private Recommendation Modelscited by 1
- 2025Cocoon: A System Architecture for Differentially Private Training with Correlated Noisescited by 1
- 2025Empowering Vector Architectures for ML: The CAMP Architecture for Matrix Multiplicationcited by 0
- 2025Cocoon: A System Architecture for Differentially Private Training with Correlated Noises8 Oct 2025
- 2025Empowering Vector Architectures for ML: The CAMP Architecture for Matrix Multiplication10 Apr 2025
- 2024Debunking the CUDA Myth Towards GPU-based AI Systems31 Dec 2024
- 2024LazyDP: Co-Designing Algorithm-Software for Scalable Training of Differentially Private Recommendation Models12 Apr 2024