PAC-Bayes Bounds for Multivariate Linear Regression and Linear Autoencoders
Ruixin Guo, Ruoming Jin, Xinyu Li, Yang Zhou
Abstract
Linear Autoencoders (LAEs) have shown strong performance in state-of-the-art recommender systems. However, this success remains largely empirical, with limited theoretical understanding. In this paper, we investigate the generalizability -- a theoretical measure of model performance in statistical learning -- of multivariate linear regression and LAEs. We first propose a PAC-Bayes bound for multivariate linear regression, extending the earlier bound for single-output linear regression by Shalaeva et al., and establish sufficient conditions for its convergence. We then show that LAEs, when evaluated under a relaxed mean squared error, can be interpreted as constrained multivariate linear regression models on bounded data, to which our bound adapts. Furthermore, we develop theoretical methods to improve the computational efficiency of optimizing the LAE bound, enabling its practical evaluation on large models and real-world datasets. Experimental results demonstrate that our bound is tight and correlates well with practical ranking metrics such as Recall@K and NDCG@K.
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.
Cited by top-tier papers1
Ask how each one uses itBuilds on7
- Autoencoders that don't overfit towards the IdentityHarald SteckNeurIPS 2020 · 72 citations
- Towards a Better Understanding of Linear Models for RecommendationRuoming Jin, Dong Li, Jing Gao, Zhi Liu et al.KDD 2021 · 21 citations
- Improved PAC-Bayesian Bounds for Linear RegressionVera Shalaeva, Alireza Fakhrizadeh Esfahani, Pascal Germain, Mihály PetreczkyAAAI 2020 · 20 citations
- Fine-grained Generalization Analysis of Inductive Matrix CompletionAntoine Ledent, Rodrigo Alves, Yunwen Lei, Marius KloftNeurIPS 2021 · 14 citations
- PAC-Bayes-Chernoff bounds for unbounded lossesIoar Casado, Luis A. Ortega Andrés, Aritz Pérez, Andrés R. MasegosaNeurIPS 2024 · 14 citations
Related papers
- It's Enough: Relaxing Diagonal Constraints in Linear Autoencoders for RecommendationJaewan Moon, Hye-young Kim, Jongwuk LeeSIGIR 2023 · 3 citations
- Statistical Guarantees for Variational Autoencoders using PAC-Bayesian TheorySokhna Diarra Mbacke, Florence Clerc, Pascal GermainNeurIPS 2023 · 22 citations
- Information-Theoretic Generalization Bounds for VAEs: A Role of Encoder and Latent VariableFutoshi Futami, Masahiro FujisawaICML 2026
- A Limitation of the PAC-Bayes FrameworkRoi Livni, Shay MoranNeurIPS 2020 · 26 citations
- Generalization Bounds for Meta-Learning via PAC-Bayes and Uniform StabilityAlec Farid, Anirudha MajumdarNeurIPS 2021 · 46 citations
