{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/enabling-fast-differentially-private-sgd-via","title":"Enabling Fast Differentially Private SGD via Just-in-Time Compilation and Vectorization","arxiv_id":"2010.09063","date":"2020-10-18","proceeding":"NeurIPS 2021 12","authors":["Pranav Subramani","Nicholas Vadivelu","Gautam Kamath"],"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: JAX and TensorFlow. JAX provides rich support for these primitives as core features of the language through the XLA compiler. We also rebuild core parts of TensorFlow Privacy, integrating features from TensorFlow 2 as well as XLA compilation, granting significant memory and runtime improvements over the current release version. These approaches allow us to achieve up to 50x speedups in comparison to the best alternatives. Our code is available at https://github.com/TheSalon/fast-dpsgd.","url_abs":"https://arxiv.org/abs/2010.09063v2","url_pdf":"https://arxiv.org/pdf/2010.09063v2.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"enabling-fast-differentially-private-sgd-via","repo_url":"https://github.com/thesalon/fast-dpsgd","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"jax","reach":{"status":"ok","spdx":"MIT"}}],"tasks":[],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":"https://syntology.ai/paper/2010.09063","atlas_url":"https://app.syntology.ai/?focus=2010.09063","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2010.09063"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-25T09:33:49+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"provenance":"deterministic:regex_extraction","url":"https://github.com/TheSalon/fast-dpsgd","reach":{"status":"ok","spdx":"MIT"}},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/thesalon/fast-dpsgd","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"unverified":11},"by_repo_kind":{"official":{"samples":11,"ran":0,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":0,"samples":[{"code_sha256_prefix":"8e6323221fce65a0","entry":"ffnn_model","repo":"TheSalon/fast-dpsgd","repo_kind":"official","path":"jaxdp.py","file_url":"https://github.com/TheSalon/fast-dpsgd/blob/HEAD/jaxdp.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"8e6323221fce65a0"}},{"code_sha256_prefix":"1ec7af9e54653150","entry":"get_ffnn_model","repo":"TheSalon/fast-dpsgd","repo_kind":"official","path":"tf2dp.py","file_url":"https://github.com/TheSalon/fast-dpsgd/blob/HEAD/tf2dp.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"1ec7af9e54653150"}},{"code_sha256_prefix":"9f2833868c634062","entry":"get_logreg_model","repo":"TheSalon/fast-dpsgd","repo_kind":"official","path":"tf2dp.py","file_url":"https://github.com/TheSalon/fast-dpsgd/blob/HEAD/tf2dp.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"9f2833868c634062"}},{"code_sha256_prefix":"813d74ecc59ea4de","entry":"get_mnist_model","repo":"TheSalon/fast-dpsgd","repo_kind":"official","path":"tf2dp.py","file_url":"https://github.com/TheSalon/fast-dpsgd/blob/HEAD/tf2dp.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"813d74ecc59ea4de"}},{"code_sha256_prefix":"95ad6d77417195b7","entry":"get_parser","repo":"TheSalon/fast-dpsgd","repo_kind":"official","path":"utils.py","file_url":"https://github.com/TheSalon/fast-dpsgd/blob/HEAD/utils.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"95ad6d77417195b7"}},{"code_sha256_prefix":"b61d063e75caff81","entry":"load_cifar10","repo":"TheSalon/fast-dpsgd","repo_kind":"official","path":"data.py","file_url":"https://github.com/TheSalon/fast-dpsgd/blob/HEAD/data.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"b61d063e75caff81"}},{"code_sha256_prefix":"32038231373d2d7f","entry":"load_dummy_cifar10","repo":"TheSalon/fast-dpsgd","repo_kind":"official","path":"data.py","file_url":"https://github.com/TheSalon/fast-dpsgd/blob/HEAD/data.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"32038231373d2d7f"}},{"code_sha256_prefix":"923ce6d32026c59f","entry":"load_mnist","repo":"TheSalon/fast-dpsgd","repo_kind":"official","path":"data.py","file_url":"https://github.com/TheSalon/fast-dpsgd/blob/HEAD/data.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"923ce6d32026c59f"}},{"code_sha256_prefix":"f0ae747648db4abb","entry":"logistic_model","repo":"TheSalon/fast-dpsgd","repo_kind":"official","path":"jaxdp.py","file_url":"https://github.com/TheSalon/fast-dpsgd/blob/HEAD/jaxdp.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"f0ae747648db4abb"}},{"code_sha256_prefix":"4b2b5048bdb44d1e","entry":"make_broadcastable","repo":"TheSalon/fast-dpsgd","repo_kind":"official","path":"backpackdp.py","file_url":"https://github.com/TheSalon/fast-dpsgd/blob/HEAD/backpackdp.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"4b2b5048bdb44d1e"}},{"code_sha256_prefix":"10e1c3a30493b84e","entry":"mnist_model","repo":"TheSalon/fast-dpsgd","repo_kind":"official","path":"jaxdp.py","file_url":"https://github.com/TheSalon/fast-dpsgd/blob/HEAD/jaxdp.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"10e1c3a30493b84e"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}