Yunbei Xu

Research

• Pointwise Generalization in Deep Neural Networks

  with Shaojie Li.

  Journal submission under review.

• Pointwise Complexity for Gaussian Fields: Upper Envelopes, Algorithmic Lower Bounds, and Separation

  Preprint.

• Bellman-sufficient Information Complexity

  Preprint.

• Pointwise Worst-Case Resource Complexity: From Partial-Information Sorting to Public-Instance Optimality

  Working paper.

• Finite-Time Minimax Bounds and an Optimal Lyapunov Policy in Queueing Control

  with Yujie Liu and Vincent Tan.

  Major Revision at Operations Research.

  Stochastic Networks Conference Outstanding Poster Prize

• Triply Robust Estimation for Continuous Treatments: Curvature-Aware Debiasing Beyond the n^−4/5 Barrier

  with Hongseok Namkoong, Isaac Scheinfeld, and Shunri Zheng.

  Working paper.

• Statistical Properties of Robust Learning under Distribution Shifts

  with Zhiyi Li, Xiaojia Mao, and Ruohan Zhan.

  In preparation for journal submission.

• Autoregressive Learning in Joint KL: Sharp Oracle Bounds and Lower Bounds

  with Yuzhe Yuan and Ruohan Zhan.

  Conference submission under review.

  Spotlight (top 4.3%) in NeurIPS 2025 ML×OR Workshop.

• On the Blessing of Pre-training in Weak-to-Strong Generalization

  with Wei Yao, Wang Zhaoyang, Gengze Xu, Chen Qian, Dongrui Liu, Ziqiao Wang, and Yong Liu.

  Conference submission under review.

• On the Power of Adaptivity for ε-Best Arm Identification in Linear Bandits

  with Arnab Maiti and Kevin Jamieson.

  Conference on Learning Theory (COLT) 2026.

• In-Context Learning for Data-Driven Censored Inventory Control

  with Sohom Mukherjee, Anh-Duy Pham,and Richard Pibernik.

  Preprint.

• Thompson Sampling for Repeated Newsvendor

  with Weizhou Zhang, Chen Li, Hanzhang Qin, and Ruihao Zhu.

  Preprint.

• Assouad, Fano, and Le Cam with Interaction: A Unifying Lower Bound Framework and Characterization for Bandit Learnability

  with Fan Chen, Dylan Foster, Yanjun Han, Jian Qian, and Alexander Rakhlin.

  Conference on Neural Information Processing Systems (NeurIPS) 2024.

  Spotlight (top 2.5%)

  Journal version under preparation.

• Statistical Properties of Robust Satisficing

  with Zhiyi Li and Ruohan Zhan.

  International Conference on Machine Learning (ICML) 2024.

  INFORMS Undergraduate Operations Research Prize, Finalist

• Bayesian Design Principles for Frequentist Sequential Learning

  with Assaf Zeevi.

  Journal of the ACM, 2025. Code

  Short version in International Conference on Machine Learning (ICML) 2023.

  ICML Outstanding Paper Award

  INFORMS George Nicholson Student Paper Competition, First Place

  Applied Probability Society Best Student Paper Award, Finalist

• Towards Optimal Problem Dependent Generalization Error Bounds in Statistical Learning Theory

  with Assaf Zeevi.

  Mathematics of Operations Research, 2025.

  Applied Probability Society Best Student Paper Award, Finalist

  ---

  Towards Problem-dependent Optimal Learning Rates

  with Assaf Zeevi.

  Conference on Neural Information Processing Systems (NeurIPS) 2020.

  Spotlight (top 4.1%)

• Upper Counterfactual Confidence Bounds: a New Optimism Principle for Contextual Bandits

  with Assaf Zeevi.

  Under Revision at Journal of Machine Learning Research.

• Acceleration of Primal-Dual Methods by Preconditioning and Simple Subproblem Procedures

  with Yanli Liu and Wotao Yin.

  Journal of Scientific Computing, 2021. Code