English

Information Complexity of Stochastic Convex Optimization: Applications to Generalization and Memorization

Machine Learning 2024-07-19 v2

Abstract

In this work, we investigate the interplay between memorization and learning in the context of \emph{stochastic convex optimization} (SCO). We define memorization via the information a learning algorithm reveals about its training data points. We then quantify this information using the framework of conditional mutual information (CMI) proposed by Steinke and Zakynthinou (2020). Our main result is a precise characterization of the tradeoff between the accuracy of a learning algorithm and its CMI, answering an open question posed by Livni (2023). We show that, in the L2L^2 Lipschitz--bounded setting and under strong convexity, every learner with an excess error ε\varepsilon has CMI bounded below by Ω(1/ε2)\Omega(1/\varepsilon^2) and Ω(1/ε)\Omega(1/\varepsilon), respectively. We further demonstrate the essential role of memorization in learning problems in SCO by designing an adversary capable of accurately identifying a significant fraction of the training samples in specific SCO problems. Finally, we enumerate several implications of our results, such as a limitation of generalization bounds based on CMI and the incompressibility of samples in SCO problems.

Keywords

Cite

@article{arxiv.2402.09327,
  title  = {Information Complexity of Stochastic Convex Optimization: Applications to Generalization and Memorization},
  author = {Idan Attias and Gintare Karolina Dziugaite and Mahdi Haghifam and Roi Livni and Daniel M. Roy},
  journal= {arXiv preprint arXiv:2402.09327},
  year   = {2024}
}

Comments

41 Pages, To appear in ICML 2024

R2 v1 2026-06-28T14:48:38.624Z