Feature Bagging Provides Stability
Yuheng Ma, Qiang Sun
Abstract
We study feature bagging through the lens of algorithmic stability. Feature bagging is an ensemble strategy that aggregates base learners trained on randomly subsampled feature subsets, possibly in a data-dependent manner. We introduce feature instability (FI), the feature-axis analogue of instance instability (II), which measures sensitivity to removing a single feature. Smaller values of II or FI correspond to stronger stability, and our experiments show that FI captures generalization-relevant information complementary to II. Within this framework, we analyze feature bagging in both a parametric linear model and a model-free setting inspired by recursive feature subsampling in random forests. In both settings, we establish formal guarantees showing that feature bagging improves the relevant stability relative to its non-bagged counterpart, with larger improvements under more aggressive subsampling. We further show that a modest number of bagging rounds is sufficient to approach the infinite-bagging stability level.
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 b89d6031-cf20-4154-ab81-c5571d4a426dBuilds on9
- Fine-Grained Analysis of Stability and Generalization for Stochastic Gradient DescentYunwen Lei, Yiming YingICML 2020 · 165 citations
- Neural Ensemble Search for Uncertainty Estimation and Dataset ShiftSheheryar Zaidi, Arber Zela, Thomas Elsken, Chris C. Holmes et al.NeurIPS 2021 · 97 citations
- On the Algorithmic Stability of Adversarial TrainingYue Xing, Qifan Song, Guang ChengNeurIPS 2021 · 74 citations
- Stability of Random Forests and Coverage of Random-Forest Prediction IntervalsYan Wang, Huaiqing Wu, Dan NettletonNeurIPS 2023 · 17 citations
- Decision Tree for Locally Private Estimation with Public DataYuheng Ma, Han Zhang, Yuchao Cai, Hanfang YangNeurIPS 2023 · 13 citations
Related papers
- Algorithmic stability and generalization of an unsupervised feature selection algorithmXinxing Wu, Qiang ChengNeurIPS 2021 · 13 citations
- Toward Better Generalization Bounds with Locally Elastic StabilityZhun Deng, Hangfeng He, Weijie J. SuICML 2021 · 51 citations
- Learning Curves for Noisy Heterogeneous Feature-Subsampled Ridge EnsemblesBenjamin S. Ruben, Cengiz PehlevanNeurIPS 2023 · 1 citation
- Building a stable classifier with the inflated argmaxJake A. Soloff, Rina Barber, Rebecca WillettNeurIPS 2024 · 9 citations
- Provably Auditing Ordinary Least Squares in Low DimensionsAnkur Moitra, Dhruv RohatgiICLR 2023
