Selected Publications
For a complete list, please refer to my Google Scholar profile. The publications are listed reverse chronologically.
* indicates equal contribution.
Preprints
One Proposal for Every Margin: Zero-Shot Amortized Sequential Importance Sampling for Binary Matrices
arXiv preprint, 2026.
Optimal Skill Selection for LLM Agents with Provable Bicriteria Guarantees
arXiv preprint, 2026.
When Context Returns: Toward Robust Internalization in On-Policy Distillation
arXiv preprint, 2026.
Publications
-
On the Sublinear Regret of Continuous K-Max Bandits
Forty-second Conference on Uncertainty in Artificial Intelligence, 2026.
Best-of-Both-Worlds for Heavy-Tailed Markov Decision Processes
Forty-third International Conference on Machine Learning, 2026.
PowerFlow: Unlocking the Dual Nature of LLMs via Principled Distribution Matching
Forty-third International Conference on Machine Learning, 2026.
Finite-time Convergence Analysis of Actor-Critic with Evolving Reward
Forty-third International Conference on Machine Learning, 2026.
Finite-Time Convergence Analysis of ODE-based Generative Models for Stochastic Interpolants
The Fourteenth International Conference on Learning Representations, 2026.
-
Finite-Time Analysis of Discrete-Time Stochastic Interpolants
Forty-second International Conference on Machine Learning, 2025.
uniINF: Best-of-Both-Worlds Algorithm for Parameter-Free Heavy-Tailed MABs
The Thirteenth International Conference on Learning Representations, 2025. Spotlight, Top 5%
-
Provable Risk-Sensitive Distributional Reinforcement Learning with General Function Approximation
Forty-first International Conference on Machine Learning, 2024.
Provably Efficient Partially Observable Risk-Sensitive Reinforcement Learning with Hindsight Observation
Forty-first International Conference on Machine Learning, 2024.
Multi-User Delay-Constrained Scheduling With Deep Recurrent Reinforcement Learning
IEEE/ACM Transactions on Networking, 2024.
Provably Efficient Iterated CVaR Reinforcement Learning with Function Approximation and Human Feedback
The Twelfth International Conference on Learning Representations, 2024.
-
Towards Minimax Optimal Reward-free Reinforcement Learning in Linear MDPs
The Eleventh International Conference on Learning Representations, 2023.
-
Effective multi-user delay-constrained scheduling with deep recurrent reinforcement learning
Twenty-third International Symposium on Theory, Algorithmic Foundations, and Protocol Design for Mobile Networks and Mobile Computing, 2022.
Nearly Minimax Optimal Reinforcement Learning with Linear Function Approximation
Thirty-ninth International Conference on Machine Learning, 2022.