Pranav Subramani

dblp:234/7713 · DBLP profile ↗
← Back
1ranked-venue papers
1as first author
1since 2021 · last 2021
—ORCID · none

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 1 · 1 first-author · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
1 paper
Trustworthy machine learning · 25% Optimization for machine learning · 25% 3D vision · 25%
Software engineering, system software, and programming languages
1 paper
Compilers and program optimization · 100%

Topics — the 3 heaviest of 5, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Optimization for machine learning › stochastic gradient descent
differentially private SGD
0.512021
Enabling Fast Differentially Private SGD via Just-in-Time Compilation and Vectorization · NeurIPS 2021
Machine learning › Trustworthy machine learning › privacy
privacy-preserving machine learning
0.512021
Enabling Fast Differentially Private SGD via Just-in-Time Compilation and Vectorization · NeurIPS 2021
Computer vision › 3D vision
runtime optimization
0.512021
Enabling Fast Differentially Private SGD via Just-in-Time Compilation and Vectorization · NeurIPS 2021

Methods — techniques the papers use, named apart from their topics

vectorization · 1.0static graph optimization · 1.0just-in-time compilation · 1.0XLA compiler · 1.0
YearPublicationVenuePosition
2021 Enabling Fast Differentially Private SGD via Just-in-Time Compilation and Vectorization
abstract
A common pain point in differentially private machine learning is the significant runtime overhead incurred when executing Differentially Private Stochastic Gradient Descent (DPSGD), which may be as large as two orders of magnitude. We thoroughly demonstrate that by exploiting powerful language primitives, including vectorization, just-in-time compilation, and static graph optimization, one can dramatically reduce these overheads, in many cases nearly matching the best non-private running times. These gains are realized in two frameworks: one is JAX, which provides rich support for these primitives through the XLA compiler. We also rebuild core parts of TensorFlow Privacy, integrating more effective vectorization as well as XLA compilation, granting significant memory and runtime improvements over previous release versions. Our proposed approaches allow us to achieve up to 50x speedups compared to the best alternatives. Our code is available at https://github.com/TheSalon/fast-dpsgd.
Pranav Subramani, Nicholas Vadivelu, Gautam Kamath 0001
NeurIPS1