Yuejie Chi

papers

1,042

total citations

papers (22)

Breaking the Sample Size Barrier in Model-Based Reinforcement Learning with a Generative Model

NEURIPS 2020arXiv

143

citations

Sample Complexity of Asynchronous Q-Learning: Sharper Analysis and Variance Reduction

NEURIPS 2020arXiv

131

citations

Fast Policy Extragradient Methods for Competitive Games with Entropy Regularization

NEURIPS 2021arXiv

citations

Get More with LESS: Synthesizing Recurrence with KV Cache Compression for Efficient LLM Inference

ICML 2024arXiv

citations

Accelerating Convergence of Score-Based Diffusion Models, Provably

ICML 2024arXiv

citations

ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inference

ICML 2025arXiv

citations

BEER: Fast $O(1/T)$ Rate for Decentralized Nonconvex Optimization with Communication Compression

NEURIPS 2022arXiv

citations

Breaking the Sample Complexity Barrier to Regret-Optimal Model-Free Reinforcement Learning

NEURIPS 2021arXiv

citations

SoteriaFL: A Unified Framework for Private Federated Learning with Communication Compression

NEURIPS 2022arXiv

citations

The Curious Price of Distributional Robustness in Reinforcement Learning with a Generative Model

NEURIPS 2023arXiv

citations

Understanding Masked Autoencoders via Hierarchical Latent Variable Models

CVPR 2023arXiv

citations

Seeing is not Believing: Robust Reinforcement Learning against Spurious Correlation

NEURIPS 2023arXiv

citations

Sample-Efficient Reinforcement Learning Is Feasible for Linearly Realizable MDPs with Limited Revisiting

NEURIPS 2021arXiv

citations

Identification of Nonlinear Latent Hierarchical Models

NEURIPS 2023arXiv

citations

Minimax-Optimal Multi-Agent RL in Markov Games With a Generative Model

NEURIPS 2022arXiv

citations

Sample-Efficient Robust Multi-Agent Reinforcement Learning in the Face of Environmental Uncertainty

ICML 2024arXiv

citations

Reward-agnostic Fine-tuning: Provable Statistical Benefits of Hybrid Reinforcement Learning

NEURIPS 2023arXiv

citations

Counterfactual Generation with Identifiability Guarantees

NEURIPS 2023arXiv

citations

Federated Offline Reinforcement Learning: Collaborative Single-Policy Coverage Suffices

ICML 2024arXiv

citations

Transformers Provably Learn Chain-of-Thought Reasoning with Length Generalization

NEURIPS 2025arXiv

citations

Breaking the Curse of Multiagency in Robust Multi-Agent Reinforcement Learning

ICML 2025arXiv

citations

A Theoretical Analysis of Self-Supervised Learning for Vision Transformers

ICLR 2025arXiv

citations