Scalable Exploration via Ensemble++
Yingru Li, Jiawei Xu, Baoxiang Wang, Zhi-Quan Tom Luo
Abstract
Thompson Sampling is a principled method for balancing exploration and exploitation, but its real-world adoption faces computational challenges in large-scale or non-conjugate settings. While ensemble-based approaches offer partial remedies, they typically require prohibitively large ensemble sizes. We propose Ensemble++, a scalable exploration framework using a novel shared-factor ensemble architecture with random linear combinations. For linear bandits, we provide theoretical guarantees showing that Ensemble++ achieves regret comparable to exact Thompson Sampling with only ensemble sizes--significantly outperforming prior methods. Crucially, this efficiency holds across both compact and finite action sets with either time-invariant or time-varying contexts without configuration changes. We extend this theoretical foundation to nonlinear rewards by replacing fixed features with learnable neural representations while preserving the same incremental update principle, effectively bridging theory and practice for real-world tasks. Comprehensive experiments across linear, quadratic, neural, and GPT-based contextual bandits validate our theoretical findings and demonstrate Ensemble++'s superior regret-computation tradeoff versus state-of-the-art methods.
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 746cd38e-a118-479a-8a6d-c49fd0f75346Cited by top-tier papers2
- Contextual Thompson Sampling via Generation of Missing DataKelly W. Zhang, Tiffany Tianhui Cai, Hongseok Namkoong, Daniel RussoNeurIPS 2025 · 5 citations
- Bayesian Ensemble for Sequential Decision-MakingRui Liu, Enmin Zhao, Lu Wang, Yu Li et al.ICLR 2026 · 2 citations
Builds on10
- Neural Contextual Bandits with UCB-based ExplorationDongruo Zhou, Lihong Li, Quanquan GuICML 2020 · 329 citations
- Epistemic Neural NetworksIan Osband, Zheng Wen, Seyed Mohammad Asghari, Vikranth Dwaracherla et al.NeurIPS 2023 · 142 citations
- Can large language models explore in-context?Akshay Krishnamurthy, Keegan Harris, Dylan J. Foster, Cyril Zhang et al.NeurIPS 2024 · 95 citations
- Hypermodels for ExplorationVikranth Dwaracherla, Xiuyuan Lu, Morteza Ibrahimi, Ian Osband et al.ICLR 2020 · 49 citations
- Langevin Monte Carlo for Contextual BanditsPan Xu, Hongkai Zheng, Eric V. Mazumdar, Kamyar Azizzadenesheli et al.ICML 2022 · 34 citations
Related papers
- Ensemble sampling for linear bandits: small ensembles sufficeDavid Janz, Alexander E. Litvak, Csaba SzepesváriNeurIPS 2024 · 8 citations
- EE-Net: Exploitation-Exploration Neural Networks in Contextual BanditsYikun Ban, Yuchen Yan, Arindam Banerjee, Jingrui HeICLR 2022 · 62 citations
- An Analysis of Ensemble SamplingChao Qin, Zheng Wen, Xiuyuan Lu, Benjamin Van RoyNeurIPS 2022 · 30 citations
- Improved Regret of Linear Ensemble SamplingHarin Lee, Min-hwan OhNeurIPS 2024 · 8 citations
- Exploration via Feature Perturbation in Contextual BanditsSeouh-won Yi, Min-hwan OhNeurIPS 2025
