Chi, Yuejie

68 publications

TMLR 2026 Beyond Expectations: Learning with Stochastic Dominance Made Practical Shicong Cen, Jincheng Mei, Hanjun Dai, Dale Schuurmans, Yuejie Chi, Bo Dai

ICLR 2025 A Theoretical Analysis of Self-Supervised Learning for Vision Transformers Yu Huang, Zixin Wen, Yuejie Chi, Yingbin Liang

ICML 2025 Breaking the Curse of Multiagency in Robust Multi-Agent Reinforcement Learning Laixi Shi, Jingchu Gai, Eric Mazumdar, Yuejie Chi, Adam Wierman

AISTATS 2025 Characterizing the Accuracy-Communication-Privacy Trade-Off in Distributed Stochastic Convex Optimization Sudeep Salgia, Nikola Pavlovic, Yuejie Chi, Qing Zhao

NeurIPS 2025 Exploration from a Primal-Dual Lens: Value-Incentivized Actor-Critic Methods for Sample-Efficient Online RL Tong Yang, Bo Dai, Lin Xiao, Yuejie Chi

AISTATS 2025 Faster WIND: Accelerating Iterative Best-of-$n$ Distillation for LLM Alignment Tong Yang, Jincheng Mei, Hanjun Dai, Zixin Wen, Shicong Cen, Dale Schuurmans, Yuejie Chi, Bo Dai

ICML 2025 Incentivize Without Bonus: Provably Efficient Model-Based Online Multi-Agent RL for Markov Games Tong Yang, Bo Dai, Lin Xiao, Yuejie Chi

NeurIPS 2025 Multi-Head Transformers Provably Learn Symbolic Multi-Step Reasoning via Gradient Descent Tong Yang, Yu Huang, Yingbin Liang, Yuejie Chi

ICLR 2025 Robust Gymnasium: A Unified Modular Benchmark for Robust Reinforcement Learning Shangding Gu, Laixi Shi, Muning Wen, Ming Jin, Eric Mazumdar, Yuejie Chi, Adam Wierman, Costas Spanos

ICML 2025 ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inference Hanshi Sun, Li-Wen Chang, Wenlei Bao, Size Zheng, Ningxin Zheng, Xin Liu, Harry Dong, Yuejie Chi, Beidi Chen

JMLR 2025 The Blessing of Heterogeneity in Federated Q-Learning: Linear Speedup and Beyond Jiin Woo, Gauri Joshi, Yuejie Chi

NeurIPS 2025 Transformers Provably Learn Chain-of-Thought Reasoning with Length Generalization Yu Huang, Zixin Wen, Aarti Singh, Yuejie Chi, Yuxin Chen

ICLR 2025 Value-Incentivized Preference Optimization: A Unified Approach to Online and Offline RLHF Shicong Cen, Jincheng Mei, Katayoon Goshvadi, Hanjun Dai, Tong Yang, Sherry Yang, Dale Schuurmans, Yuejie Chi, Bo Dai

ICLR 2025 Vertical Federated Learning with Missing Features During Training and Inference Pedro Valdeira, Shiqiang Wang, Yuejie Chi

ICML 2024 Accelerating Convergence of Score-Based Diffusion Models, Provably Gen Li, Yu Huang, Timofey Efimov, Yuting Wei, Yuejie Chi, Yuxin Chen

JMLR 2024 Distributionally Robust Model-Based Offline Reinforcement Learning with Near-Optimal Sample Complexity Laixi Shi, Yuejie Chi

AISTATS 2024 Escaping Saddle Points in Heterogeneous Federated Learning via Distributed SGD with Communication Compression Sijin Chen, Zhize Li, Yuejie Chi

JMLR 2024 Fast Policy Extragradient Methods for Competitive Games with Entropy Regularization Shicong Cen, Yuting Wei, Yuejie Chi

NeurIPS 2024 Federated Natural Policy Gradient and Actor Critic Methods for Multi-Task Reinforcement Learning Tong Yang, Shicong Cen, Yuting Wei, Yuxin Chen, Yuejie Chi

ICML 2024 Federated Offline Reinforcement Learning: Collaborative Single-Policy Coverage Suffices Jiin Woo, Laixi Shi, Gauri Joshi, Yuejie Chi

ICML 2024 Get More with LESS: Synthesizing Recurrence with KV Cache Compression for Efficient LLM Inference Harry Dong, Xinyu Yang, Zhenyu Zhang, Zhangyang Wang, Yuejie Chi, Beidi Chen

ICMLW 2024 How Transformers Learn Diverse Attention Correlations in Masked Vision Pretraining Yu Huang, Zixin Wen, Yuejie Chi, Yingbin Liang

NeurIPS 2024 In-Context Learning with Representations: Contextual Generalization of Trained Transformers Tong Yang, Yu Huang, Yingbin Liang, Yuejie Chi

ICMLW 2024 In-Context Learning with Representations: Contextual Generalization of Trained Transformers Tong Yang, Yu Huang, Yingbin Liang, Yuejie Chi

NeurIPS 2024 Learning Discrete Concepts in Latent Hierarchical Models Lingjing Kong, Guangyi Chen, Biwei Huang, Eric P. Xing, Yuejie Chi, Kun Zhang

ICMLW 2024 Prompt-Prompted Adaptive Structured Pruning for Efficient LLM Generation Harry Dong, Beidi Chen, Yuejie Chi

NeurIPS 2024 Provably Robust Score-Based Diffusion Posterior Sampling for Plug-and-Play Image Reconstruction Xingyu Xu, Yuejie Chi

ICML 2024 Sample-Efficient Robust Multi-Agent Reinforcement Learning in the Face of Environmental Uncertainty Laixi Shi, Eric Mazumdar, Yuejie Chi, Adam Wierman

NeurIPS 2024 The Sample-Communication Complexity Trade-Off in Federated Q-Learning Sudeep Salgia, Yuejie Chi

ICLR 2024 Towards Non-Asymptotic Convergence for Diffusion-Based Generative Models Gen Li, Yuting Wei, Yuxin Chen, Yuejie Chi

UAI 2023 A Trajectory Is Worth Three Sentences: Multimodal Transformer for Offline Reinforcement Learning Yiqi Wang, Mengdi Xu, Laixi Shi, Yuejie Chi

ICLR 2023 Asynchronous Gradient Play in Zero-Sum Multi-Agent Games Ruicheng Ao, Shicong Cen, Yuejie Chi

NeurIPS 2023 Counterfactual Generation with Identifiability Guarantees Hanqi Yan, Lingjing Kong, Lin Gui, Yuejie Chi, Eric P. Xing, Yulan He, Kun Zhang

ICLR 2023 Faster Last-Iterate Convergence of Policy Optimization in Zero-Sum Markov Games Shicong Cen, Yuejie Chi, Simon Shaolei Du, Lin Xiao

NeurIPS 2023 Identification of Nonlinear Latent Hierarchical Models Lingjing Kong, Biwei Huang, Feng Xie, Eric P. Xing, Yuejie Chi, Kun Zhang

ECML-PKDD 2023 Offline Reinforcement Learning with On-Policy Q-Function Regularization Laixi Shi, Robert Dadashi, Yuejie Chi, Pablo Samuel Castro, Matthieu Geist

NeurIPS 2023 Reward-Agnostic Fine-Tuning: Provable Statistical Benefits of Hybrid Reinforcement Learning Gen Li, Wenhao Zhan, Jason Lee, Yuejie Chi, Yuxin Chen

NeurIPS 2023 Seeing Is Not Believing: Robust Reinforcement Learning Against Spurious Correlation Wenhao Ding, Laixi Shi, Yuejie Chi, Ding Zhao

ICML 2023 The Blessing of Heterogeneity in Federated Q-Learning: Linear Speedup and Beyond Jiin Woo, Gauri Joshi, Yuejie Chi

NeurIPS 2023 The Curious Price of Distributional Robustness in Reinforcement Learning with a Generative Model Laixi Shi, Gen Li, Yuting Wei, Yuxin Chen, Matthieu Geist, Yuejie Chi

ICML 2023 The Power of Preconditioning in Overparameterized Low-Rank Matrix Sensing Xingyu Xu, Yandi Shen, Yuejie Chi, Cong Ma

ICMLW 2023 Towards Structured Sparsity in Transformers for Efficient Inference Harry Dong, Beidi Chen, Yuejie Chi

CVPR 2023 Understanding Masked Autoencoders via Hierarchical Latent Variable Models Lingjing Kong, Martin Q. Ma, Guangyi Chen, Eric P. Xing, Yuejie Chi, Louis-Philippe Morency, Kun Zhang

AISTATS 2022 Scaling and Scalability: Provable Nonconvex Low-Rank Tensor Completion Tian Tong, Cong Ma, Ashley Prater-Bennette, Erin Tripp, Yuejie Chi

NeurIPSW 2022 A Multi-Token Coordinate Descent Method for Vertical Federated Learning Pedro Valdeira, Yuejie Chi, Claudia Soares, Joao Xavier

NeurIPS 2022 BEER: Fast $O(1/T)$ Rate for Decentralized Nonconvex Optimization with Communication Compression Haoyu Zhao, Boyue Li, Zhize Li, Peter Richtarik, Yuejie Chi

AAAI 2022 Batch Active Learning with Graph Neural Networks via Multi-Agent Deep Reinforcement Learning Yuheng Zhang, Hanghang Tong, Yinglong Xia, Yan Zhu, Yuejie Chi, Lei Ying

NeurIPS 2022 Minimax-Optimal Multi-Agent RL in Markov Games with a Generative Model Gen Li, Yuejie Chi, Yuting Wei, Yuxin Chen

ICML 2022 Pessimistic Q-Learning for Offline Reinforcement Learning: Towards Optimal Sample Complexity Laixi Shi, Gen Li, Yuting Wei, Yuxin Chen, Yuejie Chi

JMLR 2022 Scaling and Scalability: Provable Nonconvex Low-Rank Tensor Estimation from Incomplete Measurements Tian Tong, Cong Ma, Ashley Prater-Bennette, Erin Tripp, Yuejie Chi

NeurIPS 2022 SoteriaFL: A Unified Framework for Private Federated Learning with Communication Compression Zhize Li, Haoyu Zhao, Boyue Li, Yuejie Chi

JMLR 2021 Accelerating Ill-Conditioned Low-Rank Matrix Estimation via Scaled Gradient Descent Tian Tong, Cong Ma, Yuejie Chi

NeurIPS 2021 Breaking the Sample Complexity Barrier to Regret-Optimal Model-Free Reinforcement Learning Gen Li, Laixi Shi, Yuxin Chen, Yuantao Gu, Yuejie Chi

NeurIPS 2021 Fast Policy Extragradient Methods for Competitive Games with Entropy Regularization Shicong Cen, Yuting Wei, Yuejie Chi

NeurIPS 2021 Sample-Efficient Reinforcement Learning Is Feasible for Linearly Realizable MDPs with Limited Revisiting Gen Li, Yuxin Chen, Yuejie Chi, Yuantao Gu, Yuting Wei

COLT 2021 SoftMax Policy Gradient Methods Can Take Exponential Time to Converge Gen Li, Yuting Wei, Yuejie Chi, Yuantao Gu, Yuxin Chen

FnTML 2021 Spectral Methods for Data Science: A Statistical Perspective Yuxin Chen, Yuejie Chi, Jianqing Fan, Cong Ma

ICML 2021 Tightening the Dependence on Horizon in the Sample Complexity of Q-Learning Gen Li, Changxiao Cai, Yuxin Chen, Yuantao Gu, Yuting Wei, Yuejie Chi

NeurIPS 2020 Breaking the Sample Size Barrier in Model-Based Reinforcement Learning with a Generative Model Gen Li, Yuting Wei, Yuejie Chi, Yuantao Gu, Yuxin Chen

AISTATS 2020 Communication-Efficient Distributed Optimization in Networks with Gradient Tracking and Variance Reduction Boyue Li, Shicong Cen, Yuxin Chen, Yuejie Chi

JMLR 2020 Communication-Efficient Distributed Optimization in Networks with Gradient Tracking and Variance Reduction Boyue Li, Shicong Cen, Yuxin Chen, Yuejie Chi

NeurIPS 2020 Sample Complexity of Asynchronous Q-Learning: Sharper Analysis and Variance Reduction Gen Li, Yuting Wei, Yuejie Chi, Yuantao Gu, Yuxin Chen

AISTATS 2019 Nonconvex Matrix Factorization from Rank-One Measurements Yuanxin Li, Cong Ma, Yuxin Chen, Yuejie Chi

ICML 2018 Implicit Regularization in Nonconvex Statistical Estimation: Gradient Descent Converges Linearly for Phase Retrieval and Matrix Completion Cong Ma, Kaizheng Wang, Yuejie Chi, Yuxin Chen

JMLR 2017 A Nonconvex Approach for Phase Retrieval: Reshaped Wirtinger Flow and Incremental Algorithms Huishuai Zhang, Yingbin Liang, Yuejie Chi

ICML 2016 Provable Non-Convex Phase Retrieval with Outliers: Median TruncatedWirtinger Flow Huishuai Zhang, Yuejie Chi, Yingbin Liang

ICML 2013 Spectral Compressed Sensing via Structured Matrix Completion Yuxin Chen, Yuejie Chi

CVPR 2012 Connecting the Dots in Multi-Class Classification: From Nearest Subspace to Collaborative Representation Yuejie Chi, Fatih Porikli