Generalization and Robustness of the Tilted Empirical Risk
Gholamali Aminian, Amir R. Asadi, Tian Li, Ahmad Beirami, Gesine Reinert, Samuel N. Cohen
Abstract
The generalization error (risk) of a supervised statistical learning algorithm quantifies its prediction ability on previously unseen data. Inspired by exponential tilting, Li et al. (2021) proposed the tilted empirical risk (TER) as a non-linear risk metric for machine learning applications such as classification and regression problems. In this work, we examine the generalization error of the tilted empirical risk in the robustness regime under negative tilt. Our first contribution is to provide uniform and information-theoretic bounds on the tilted generalization error, defined as the difference between the population risk and the tilted empirical risk, under negative tilt for unbounded loss function under bounded (1 + ϵ)-th moment of loss function for some ϵ ∈ (0, 1] with a convergence rate of O(n -ϵ/(1+ϵ) ) where n is the number of training samples, revealing a novel application for TER under no distribution shift. Secondly, we study the robustness of the tilted empirical risk with respect to noisy outliers at training time and provide theoretical guarantees under distribution shift for the tilted empirical risk. We empirically corroborate our findings in simple experimental setups where we evaluate our bounds to select the value of tilt in a data-driven manner.
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 8fdfd590-6ec4-4ba0-904b-7c022f8cf5efCited by top-tier papers1
Ask how each one uses itBuilds on8
- Towards a Theoretical Framework of Out-of-Distribution GeneralizationHaotian Ye, Chuanlong Xie, Tianle Cai, Ruichen Li et al.NeurIPS 2021 · 159 citations
- PAC-Bayes Analysis Beyond the Usual BoundsOmar Rivasplata, Ilja Kuzborskij, Csaba Szepesvári, John Shawe-TaylorNeurIPS 2020 · 101 citations
- An Exact Characterization of the Generalization Error for the Gibbs AlgorithmGholamali Aminian, Yuheng Bu, Laura Toni, Miguel R. D. Rodrigues et al.NeurIPS 2021 · 75 citations
- Conditioning and Processing: Techniques to Improve Information-Theoretic Generalization BoundsHassan Hafez-Kolahi, Zeinab Golgooni, Shohreh Kasaei, Mahdieh SoleymaniNeurIPS 2020 · 63 citations
- Learning Bounds for Risk-sensitive LearningJaeho Lee, Sejun Park, Jinwoo ShinNeurIPS 2020 · 52 citations
Related papers
- Tilted Empirical Risk MinimizationTian Li, Ahmad Beirami, Maziar Sanjabi, Virginia SmithICLR 2021 · 42 citations
- Tilted Sparse Additive ModelsYingjie Wang, Hong Chen, Weifeng Liu, Fengxiang He et al.ICML 2023 · 6 citations
- Adapting to Online Label Shift with Provable GuaranteesYong Bai, Yu-Jie Zhang, Peng Zhao, Masashi Sugiyama et al.NeurIPS 2022 · 43 citations
- Loss Function Learning for Domain Generalization by Implicit GradientBoyan Gao, Henry Gouk, Yongxin Yang, Timothy M. HospedalesICML 2022 · 29 citations
- Model-Agnostic Random Weighting for Out-of-Distribution GeneralizationYue He, Pengfei Tian, Renzhe Xu, Xinwei Shen et al.KDD 2024 · 1 citation
