Yuejie Chi

ORCID

Publication Activity (10 Years)

Years Active: 2010-2024
Publications (10 Years): 184

Top Venues

CoRR

ICASSP

NeurIPS

IEEE Trans. Signal Process.

Publications

Lingjing Kong, Guangyi Chen, Biwei Huang, Eric P. Xing, Yuejie Chi, Kun Zhang
Learning Discrete Concepts in Latent Hierarchical Models. CoRR (2024)
Harry Dong, Xinyu Yang, Zhenyu Zhang, Zhangyang Wang, Yuejie Chi, Beidi Chen
Get More with LESS: Synthesizing Recurrence with KV Cache Compression for Efficient LLM Inference. CoRR (2024)
Gen Li, Changxiao Cai, Yuxin Chen, Yuting Wei, Yuejie Chi
Is Q-Learning Minimax Optimal? A Tight Sample Complexity Analysis. Oper. Res. 72 (1) (2024)
Xingyu Xu, Yuejie Chi
Provably Robust Score-Based Diffusion Posterior Sampling for Plug-and-Play Image Reconstruction. CoRR (2024)
Jiin Woo, Laixi Shi, Gauri Joshi, Yuejie Chi
Federated Offline Reinforcement Learning: Collaborative Single-Policy Coverage Suffices. CoRR (2024)
Yu Huang, Zixin Wen, Yuejie Chi, Yingbin Liang
Transformers Provably Learn Feature-Position Correlations in Masked Image Modeling. CoRR (2024)
Gen Li, Weichen Wu, Yuejie Chi, Cong Ma, Alessandro Rinaldo, Yuting Wei
High-Probability Sample Complexities for Policy Evaluation With Linear Function Approximation. IEEE Trans. Inf. Theory 70 (8) (2024)
Hanqi Yan, Lingjing Kong, Lin Gui, Yuejie Chi, Eric P. Xing, Yulan He, Kun Zhang
Counterfactual Generation with Identifiability Guarantees. CoRR (2024)
He Wang, Laixi Shi, Yuejie Chi
Sample Complexity of Offline Distributionally Robust Linear Markov Decision Processes. CoRR (2024)
He Wang, Yuejie Chi
Communication-Efficient Federated Optimization over Semi-Decentralized Networks. ICASSP (2024)
Pedro Valdeira, João Xavier, Cláudia Soares, Yuejie Chi
Communication-efficient Vertical Federated Learning via Compressed Error Feedback. CoRR (2024)
Gen Li, Yuting Wei, Yuxin Chen, Yuejie Chi
Towards Non-Asymptotic Convergence for Diffusion-Based Generative Models. ICLR (2024)
Gen Li, Yuting Wei, Yuejie Chi, Yuxin Chen
Breaking the Sample Size Barrier in Model-Based Reinforcement Learning with a Generative Model. Oper. Res. 72 (1) (2024)
Harry Dong, Beidi Chen, Yuejie Chi
Prompt-prompted Mixture of Experts for Efficient LLM Generation. CoRR (2024)
Laixi Shi, Eric Mazumdar, Yuejie Chi, Adam Wierman
Sample-Efficient Robust Multi-Agent Reinforcement Learning in the Face of Environmental Uncertainty. CoRR (2024)
Sijin Chen, Zhize Li, Yuejie Chi
Escaping Saddle Points in Heterogeneous Federated Learning via Distributed SGD with Communication Compression. AISTATS (2024)
Shicong Cen, Jincheng Mei, Hanjun Dai, Dale Schuurmans, Yuejie Chi, Bo Dai
Beyond Expectations: Learning with Stochastic Dominance Made Practical. CoRR (2024)
Shicong Cen, Jincheng Mei, Katayoon Goshvadi, Hanjun Dai, Tong Yang, Sherry Yang, Dale Schuurmans, Yuejie Chi, Bo Dai
Value-Incentivized Preference Optimization: A Unified Approach to Online and Offline RLHF. CoRR (2024)
Gen Li, Yu Huang, Timofey Efimov, Yuting Wei, Yuejie Chi, Yuxin Chen
Accelerating Convergence of Score-Based Diffusion Models, Provably. CoRR (2024)
Harry Dong, Megna Shah, Sean Donegan, Yuejie Chi
Deep Unfolded Tensor Robust PCA With Self-Supervised Learning. ICASSP (2023)
Jiin Woo, Gauri Joshi, Yuejie Chi
The Blessing of Heterogeneity in Federated Q-Learning: Linear Speedup and Beyond. ICML (2023)
Harry Dong, Sean Donegan, Megna Shah, Yuejie Chi
A Lightweight Transformer for Faster and Robust EBSD Data Collection. CoRR (2023)
Wenhao Ding, Laixi Shi, Yuejie Chi, Ding Zhao
Seeing is not Believing: Robust Reinforcement Learning against Spurious Correlation. CoRR (2023)
Ruicheng Ao, Shicong Cen, Yuejie Chi
Asynchronous Gradient Play in Zero-Sum Multi-agent Games. ICLR (2023)
Lingjing Kong, Martin Q. Ma, Guangyi Chen, Eric P. Xing, Yuejie Chi, Louis-Philippe Morency, Kun Zhang
Understanding Masked Autoencoders via Hierarchical Latent Variable Models. CoRR (2023)
Laixi Shi, Robert Dadashi, Yuejie Chi, Pablo Samuel Castro, Matthieu Geist
Offline Reinforcement Learning with On-Policy Q-Function Regularization. CoRR (2023)
Sijin Chen, Zhize Li, Yuejie Chi
Escaping Saddle Points in Heterogeneous Federated Learning via Distributed SGD with Communication Compression. CoRR (2023)
Xingyu Xu, Yandi Shen, Yuejie Chi, Cong Ma
The Power of Preconditioning in Overparameterized Low-Rank Matrix Sensing. ICML (2023)
Xingyu Xu, Yandi Shen, Yuejie Chi, Cong Ma
The Power of Preconditioning in Overparameterized Low-Rank Matrix Sensing. CoRR (2023)
Shicong Cen, Yuejie Chi, Simon Shaolei Du, Lin Xiao
Faster Last-iterate Convergence of Policy Optimization in Zero-Sum Markov Games. ICLR (2023)
Gen Li, Weichen Wu, Yuejie Chi, Cong Ma, Alessandro Rinaldo, Yuting Wei
Sharp high-probability sample complexities for policy evaluation with linear function approximation. CoRR (2023)
Cong Ma, Xingyu Xu, Tian Tong, Yuejie Chi
Provably Accelerating Ill-Conditioned Low-rank Estimation via Scaled Gradient Descent, Even with Overparameterization. CoRR (2023)
Laixi Shi, Robert Dadashi, Yuejie Chi, Pablo Samuel Castro, Matthieu Geist
Offline Reinforcement Learning with On-Policy Q-Function Regularization. ECML/PKDD (4) (2023)
Maxime Ferreira Da Costa, Yuejie Chi
Local Geometry of Nonconvex Spike Deconvolution From Low-Pass Measurements. IEEE J. Sel. Areas Inf. Theory 4 (2023)
Gen Li, Wenhao Zhan, Jason D. Lee, Yuejie Chi, Yuxin Chen
Reward-agnostic Fine-tuning: Provable Statistical Benefits of Hybrid Reinforcement Learning. CoRR (2023)
Tong Yang, Shicong Cen, Yuting Wei, Yuxin Chen, Yuejie Chi
Federated Natural Policy Gradient Methods for Multi-task Reinforcement Learning. CoRR (2023)
Shicong Cen, Yuejie Chi
Global Convergence of Policy Gradient Methods in Reinforcement Learning, Games and Control. CoRR (2023)
He Wang, Yuejie Chi
Communication-Efficient Federated Optimization over Semi-Decentralized Networks. CoRR (2023)
Jiin Woo, Gauri Joshi, Yuejie Chi
The Blessing of Heterogeneity in Federated Q-learning: Linear Speedup and Beyond. CoRR (2023)
Gen Li, Yanxi Chen, Yuejie Chi, H. Vincent Poor, Yuxin Chen
Fast Computation of Optimal Transport via Entropy-Regularized Extragradient Methods. CoRR (2023)
Lingjing Kong, Martin Q. Ma, Guangyi Chen, Eric P. Xing, Yuejie Chi, Louis-Philippe Morency, Kun Zhang
Understanding Masked Autoencoders via Hierarchical Latent Variable Models. CVPR (2023)
Laixi Shi, Gen Li, Yuting Wei, Yuxin Chen, Matthieu Geist, Yuejie Chi
The Curious Price of Distributional Robustness in Reinforcement Learning with a Generative Model. CoRR (2023)
Lingjing Kong, Biwei Huang, Feng Xie, Eric P. Xing, Yuejie Chi, Kun Zhang
Identification of Nonlinear Latent Hierarchical Models. CoRR (2023)
Pedro Valdeira, Yuejie Chi, Cláudia Soares, João Xavier
A Multi-Token Coordinate Descent Method for Semi-Decentralized Vertical Federated Learning. CoRR (2023)
Gen Li, Yuting Wei, Yuejie Chi, Yuxin Chen
Softmax policy gradient methods can take exponential time to converge. Math. Program. 201 (1) (2023)
Hanqi Yan, Lingjing Kong, Lin Gui, Yuejie Chi, Eric P. Xing, Yulan He, Kun Zhang
Counterfactual Generation with Identifiability Guarantees. NeurIPS (2023)
Boyue Li, Yuejie Chi
Convergence and Privacy of Decentralized Nonconvex Optimization with Gradient Clipping and Communication Compression. CoRR (2023)
Lingjing Kong, Biwei Huang, Feng Xie, Eric P. Xing, Yuejie Chi, Kun Zhang
Identification of Nonlinear Latent Hierarchical Models. NeurIPS (2023)
Wenhao Ding, Laixi Shi, Yuejie Chi, Ding Zhao
Seeing is not Believing: Robust Reinforcement Learning against Spurious Correlation. NeurIPS (2023)
Laixi Shi, Gen Li, Yuting Wei, Yuxin Chen, Matthieu Geist, Yuejie Chi
The Curious Price of Distributional Robustness in Reinforcement Learning with a Generative Model. NeurIPS (2023)
Yiqi Wang, Mengdi Xu, Laixi Shi, Yuejie Chi
A trajectory is worth three sentences: multimodal transformer for offline reinforcement learning. UAI (2023)
Wenhao Zhan, Shicong Cen, Baihe Huang, Yuxin Chen, Jason D. Lee, Yuejie Chi
Policy Mirror Descent for Regularized Reinforcement Learning: A Generalized Framework with Linear Convergence. SIAM J. Optim. 33 (2) (2023)
Gen Li, Wenhao Zhan, Jason D. Lee, Yuejie Chi, Yuxin Chen
Reward-agnostic Fine-tuning: Provable Statistical Benefits of Hybrid Reinforcement Learning. NeurIPS (2023)
Gen Li, Yuting Wei, Yuxin Chen, Yuejie Chi
Towards Faster Non-Asymptotic Convergence for Diffusion-Based Generative Models. CoRR (2023)
Haoyu Zhao, Boyue Li, Zhize Li, Peter Richtárik, Yuejie Chi
BEER: Fast $O(1/T)$ Rate for Decentralized Nonconvex Optimization with Communication Compression. NeurIPS (2022)
Gen Li, Yuejie Chi, Yuting Wei, Yuxin Chen
Minimax-Optimal Multi-Agent RL in Zero-Sum Markov Games With a Generative Model. CoRR (2022)
Zhize Li, Haoyu Zhao, Boyue Li, Yuejie Chi
SoteriaFL: A Unified Framework for Private Federated Learning with Communication Compression. CoRR (2022)
Harry Dong, Megna Shah, Sean Donegan, Yuejie Chi
Deep Unfolded Tensor Robust PCA with Self-supervised Learning. CoRR (2022)
Yuheng Zhang, Hanghang Tong, Yinglong Xia, Yan Zhu, Yuejie Chi, Lei Ying
Batch Active Learning with Graph Neural Networks via Multi-Agent Deep Reinforcement Learning. AAAI (2022)
Harry Dong, Tian Tong, Cong Ma, Yuejie Chi
Fast and Provable Tensor Robust Principal Component Analysis via Scaled Gradient Descent. CoRR (2022)
Maxime Ferreira Da Costa, Yuejie Chi
Local Geometry of Nonconvex Spike Deconvolution from Low-Pass Measurements. CoRR (2022)
Diogo Cardoso, Boyue Li, Yuejie Chi, João Xavier
Harvesting Curvatures for Communication-Efficient Distributed Optimization. IEEECONF (2022)
Ruicheng Ao, Shicong Cen, Yuejie Chi
Asynchronous Gradient Play in Zero-Sum Multi-agent Games. CoRR (2022)
Boyue Li, Zhize Li, Yuejie Chi
DESTRESS: Computation-Optimal and Communication-Efficient Decentralized Nonconvex Finite-Sum Optimization. SIAM J. Math. Data Sci. 4 (3) (2022)
Gen Li, Laixi Shi, Yuxin Chen, Yuejie Chi, Yuting Wei
Settling the Sample Complexity of Model-Based Offline Reinforcement Learning. CoRR (2022)
Shicong Cen, Fan Chen, Yuejie Chi
Independent Natural Policy Gradient Methods for Potential Games: Finite-time Global Convergence with Entropy Regularization. CDC (2022)
Zhize Li, Haoyu Zhao, Boyue Li, Yuejie Chi
SoteriaFL: A Unified Framework for Private Federated Learning with Communication Compression. NeurIPS (2022)
Haoyu Zhao, Boyue Li, Zhize Li, Peter Richtárik, Yuejie Chi
BEER: Fast O(1/T) Rate for Decentralized Nonconvex Optimization with Communication Compression. CoRR (2022)
Laixi Shi, Yuejie Chi
Distributionally Robust Model-Based Offline Reinforcement Learning with Near-Optimal Sample Complexity. CoRR (2022)
Shicong Cen, Fan Chen, Yuejie Chi
Independent Natural Policy Gradient Methods for Potential Games: Finite-time Global Convergence with Entropy Regularization. CoRR (2022)
Shicong Cen, Yuejie Chi, Simon S. Du, Lin Xiao
Faster Last-iterate Convergence of Policy Optimization in Zero-Sum Markov Games. CoRR (2022)
Shicong Cen, Chen Cheng, Yuxin Chen, Yuting Wei, Yuejie Chi
Fast Global Convergence of Natural Policy Gradient Methods with Entropy Regularization. Oper. Res. 70 (4) (2022)
Gen Li, Yuting Wei, Yuejie Chi, Yuantao Gu, Yuxin Chen
Sample Complexity of Asynchronous Q-Learning: Sharper Analysis and Variance Reduction. IEEE Trans. Inf. Theory 68 (1) (2022)
Tian Tong, Cong Ma, Ashley Prater-Bennette, Erin Tripp, Yuejie Chi
Scaling and Scalability: Provable Nonconvex Low-Rank Tensor Completion. AISTATS (2022)
Laixi Shi, Gen Li, Yuting Wei, Yuxin Chen, Yuejie Chi
Pessimistic Q-Learning for Offline Reinforcement Learning: Towards Optimal Sample Complexity. ICML (2022)
Tian Tong, Cong Ma, Ashley Prater-Bennette, Erin E. Tripp, Yuejie Chi
Scaling and Scalability: Provable Nonconvex Low-Rank Tensor Estimation from Incomplete Measurements. J. Mach. Learn. Res. 23 (2022)
Harlin Lee, Andrea L. Bertozzi, Jelena Kovacevic, Yuejie Chi
Privacy-Preserving Federated Multi-Task Linear Regression: A One-Shot Linear Mixing Approach Inspired By Graph Regularization. ICASSP (2022)
Tian Tong, Cong Ma, Yuejie Chi
Accelerating ILL-Conditioned Robust Low-Rank Tensor Regression. ICASSP (2022)
Gen Li, Yuejie Chi, Yuting Wei, Yuxin Chen
Minimax-Optimal Multi-Agent RL in Markov Games With a Generative Model. NeurIPS (2022)
Laixi Shi, Gen Li, Yuting Wei, Yuxin Chen, Yuejie Chi
Pessimistic Q-Learning for Offline Reinforcement Learning: Towards Optimal Sample Complexity. CoRR (2022)
Yuheng Zhang, Yinglong Xia, Yan Zhu, Yuejie Chi, Lei Ying, Hanghang Tong
Active Heterogeneous Graph Neural Networks with Per-step Meta-Q-Learning. ICDM (2022)
Gen Li, Yuting Wei, Yuejie Chi, Yuantao Gu, Yuxin Chen
Softmax Policy Gradient Methods Can Take Exponential Time to Converge. COLT (2021)
Vincent Monardo, Abhiram Iyer, Sean Donegan, Marc De Graef, Yuejie Chi
Plug-And-Play Image Reconstruction Meets Stochastic Variance-Reduced Gradient Methods. ICIP (2021)
Shicong Cen, Yuting Wei, Yuejie Chi
Fast Policy Extragradient Methods for Competitive Games with Entropy Regularization. NeurIPS (2021)
Cong Ma, Yuanxin Li, Yuejie Chi
Beyond Procrustes: Balancing-Free Gradient Descent for Asymmetric Low-Rank Matrix Sensing. CoRR (2021)
Tian Tong, Cong Ma, Yuejie Chi
Accelerating Ill-Conditioned Low-Rank Matrix Estimation via Scaled Gradient Descent. J. Mach. Learn. Res. 22 (2021)
Gen Li, Changxiao Cai, Yuxin Chen, Yuantao Gu, Yuting Wei, Yuejie Chi
Tightening the Dependence on Horizon in the Sample Complexity of Q-Learning. ICML (2021)
Cong Ma, Yuanxin Li, Yuejie Chi
Beyond Procrustes: Balancing-Free Gradient Descent for Asymmetric Low-Rank Matrix Sensing. IEEE Trans. Signal Process. 69 (2021)
Maxime Ferreira Da Costa, Yuejie Chi
Compressed Super-Resolution of Positive Sources. IEEE Signal Process. Lett. 28 (2021)
Tian Tong, Cong Ma, Ashley Prater-Bennette, Erin Tripp, Yuejie Chi
Scaling and Scalability: Provable Nonconvex Low-Rank Tensor Estimation from Incomplete Measurements. CoRR (2021)
Laixi Shi, Yuejie Chi
Manifold Gradient Descent Solves Multi-Channel Sparse Blind Deconvolution Provably and Efficiently. IEEE Trans. Inf. Theory 67 (7) (2021)
Gen Li, Changxiao Cai, Yuxin Chen, Yuantao Gu, Yuting Wei, Yuejie Chi
Tightening the Dependence on Horizon in the Sample Complexity of Q-Learning. CoRR (2021)
Boyue Li, Zhize Li, Yuejie Chi
DESTRESS: Computation-Optimal and Communication-Efficient Decentralized Nonconvex Finite-Sum Optimization. CoRR (2021)
Gen Li, Laixi Shi, Yuxin Chen, Yuantao Gu, Yuejie Chi
Breaking the Sample Complexity Barrier to Regret-Optimal Model-Free Reinforcement Learning. CoRR (2021)
Yuxin Chen, Yuejie Chi, Jianqing Fan, Cong Ma
Spectral Methods for Data Science: A Statistical Perspective. Found. Trends Mach. Learn. 14 (5) (2021)
Gen Li, Yuxin Chen, Yuejie Chi, Yuantao Gu, Yuting Wei
Sample-Efficient Reinforcement Learning Is Feasible for Linearly Realizable MDPs with Limited Revisiting. CoRR (2021)
Tian Tong, Cong Ma, Yuejie Chi
Low-Rank Matrix Recovery With Scaled Subgradient Methods: Fast and Robust Convergence Without the Condition Number. IEEE Trans. Signal Process. 69 (2021)
Yuanxin Li, Cong Ma, Yuxin Chen, Yuejie Chi
Nonconvex Matrix Factorization From Rank-One Measurements. IEEE Trans. Inf. Theory 67 (3) (2021)
Gen Li, Yuxin Chen, Yuejie Chi, Yuantao Gu, Yuting Wei
Sample-Efficient Reinforcement Learning Is Feasible for Linearly Realizable MDPs with Limited Revisiting. NeurIPS (2021)
Wenhao Zhan, Shicong Cen, Baihe Huang, Yuxin Chen, Jason D. Lee, Yuejie Chi
Policy Mirror Descent for Regularized Reinforcement Learning: A Generalized Framework with Linear Convergence. CoRR (2021)

Yuejie Chi.css-eqlu8i{font-size:small;padding-left:4px;}

Publication Activity (10 Years)

Top Topics

Top Venues

Yuejie Chi