Multi-Task Differential Privacy Under Distribution Skew
Walid Krichene, Prateek Jain, Shuang Song, Mukund Sundararajan, Abhradeep Guha Thakurta, Li Zhang
Abstract
We study the problem of multi-task learning under user-level differential privacy, in which users contribute data to tasks, each involving a subset of users. One important aspect of the problem, that can significantly impact quality, is the distribution skew among tasks. Certain tasks may have much fewer data samples than others, making them more susceptible to the noise added for privacy. It is natural to ask whether algorithms can adapt to this skew to improve the overall utility. We give a systematic analysis of the problem, by studying how to optimally allocate a user's privacy budget among tasks. We propose a generic algorithm, based on an adaptive reweighting of the empirical loss, and show that when there is task distribution skew, this gives a quantifiable improvement of excess empirical risk. Experimental studies on recommendation problems that exhibit a long tail of small tasks, demonstrate that our methods significantly improve utility, achieving the state of the art on two standard benchmarks.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 6090f8c3-9b04-4bd5-8de0-bbadf0061731Builds on10
- Deep Learning with Differential PrivacyMartín Abadi, Andy Chu, Ian J. Goodfellow, H. Brendan McMahan et al.CCS 2016 · 7,620 citations
- Membership Inference Attacks Against Machine Learning ModelsReza Shokri, Marco Stronati, Congzheng Song, Vitaly ShmatikovS&P 2017 · 5,137 citations
- Differentially Private Meta-LearningJeffrey Li, Mikhail Khodak, Sebastian Caldas, Ameet TalwalkarICLR 2020 · 125 citations
- Individual Privacy Accounting via a Rényi FilterVitaly Feldman, Tijana ZrnicNeurIPS 2021 · 124 citations
- Learning with User-Level PrivacyDaniel Levy, Ziteng Sun, Kareem Amin, Satyen Kale et al.NeurIPS 2021 · 113 citations
Related papers
- User-Level Differential Privacy With Few Examples Per UserBadih Ghazi, Pritish Kamath, Ravi Kumar, Pasin Manurangsi et al.NeurIPS 2023 · 19 citations
- Smoothly Bounding User Contributions in Differential PrivacyAlessandro Epasto, Mohammad Mahdian, Jieming Mao, Vahab S. Mirrokni et al.NeurIPS 2020 · 17 citations
- Task-aware Privacy Preservation for Multi-dimensional DataJiangnan Cheng, Ao Tang, Sandeep ChinchaliICML 2022 · 8 citations
- Algorithms for bounding contribution for histogram estimation under user-level privacyYuhan Liu, Ananda Theertha Suresh, Wennan Zhu, Peter Kairouz et al.ICML 2023 · 14 citations
- Differentially Private Cross-Silo Recommendation from Implicit FeedbackXun Ran, Qingqing Ye, Xin Huang, Jianliang Xu et al.ICML 2026
