Vitaly Feldman, Audra McMillan, Kunal Talwar
This paper introduces a simple mechanism using dummy data points to privatize batch sizes in DP-SGD, enabling privacy amplification under truncation and observable batch counts. It provides tight privacy accounting for both Poisson sampling and random allocation, reducing the expected number of dummy points needed compared to direct noise-based approaches. Empirical results show improved privacy-efficiency tradeoffs over prior methods.
In real-world DP-SGD deployments, batch size variability and truncation due to system constraints can undermine privacy guarantees. This work enables practical, privacy-preserving training in distributed settings where batch sizes may be exposed, without sacrificing efficiency or accuracy.
Thank you to arXiv for use of its open access interoperability. arxiv.org. aibrief.ai is not affiliated with or endorsed by arXiv.