Alert button
Picture for Yuejie Chi

Yuejie Chi

Alert button

Sample Complexity of Offline Distributionally Robust Linear Markov Decision Processes

Mar 19, 2024
He Wang, Laixi Shi, Yuejie Chi

Viaarxiv icon

Accelerating Convergence of Score-Based Diffusion Models, Provably

Mar 06, 2024
Gen Li, Yu Huang, Timofey Efimov, Yuting Wei, Yuejie Chi, Yuxin Chen

Figure 1 for Accelerating Convergence of Score-Based Diffusion Models, Provably
Figure 2 for Accelerating Convergence of Score-Based Diffusion Models, Provably
Viaarxiv icon

Transformers Provably Learn Feature-Position Correlations in Masked Image Modeling

Mar 04, 2024
Yu Huang, Zixin Wen, Yuejie Chi, Yingbin Liang

Viaarxiv icon

Counterfactual Generation with Identifiability Guarantees

Feb 23, 2024
Hanqi Yan, Lingjing Kong, Lin Gui, Yuejie Chi, Eric Xing, Yulan He, Kun Zhang

Viaarxiv icon

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

Feb 14, 2024
Harry Dong, Xinyu Yang, Zhenyu Zhang, Zhangyang Wang, Yuejie Chi, Beidi Chen

Viaarxiv icon

Federated Offline Reinforcement Learning: Collaborative Single-Policy Coverage Suffices

Feb 08, 2024
Jiin Woo, Laixi Shi, Gauri Joshi, Yuejie Chi

Viaarxiv icon

Beyond Expectations: Learning with Stochastic Dominance Made Practical

Feb 05, 2024
Shicong Cen, Jincheng Mei, Hanjun Dai, Dale Schuurmans, Yuejie Chi, Bo Dai

Viaarxiv icon

Communication-Efficient Federated Optimization over Semi-Decentralized Networks

Nov 30, 2023
He Wang, Yuejie Chi

Viaarxiv icon

Federated Natural Policy Gradient Methods for Multi-task Reinforcement Learning

Nov 01, 2023
Tong Yang, Shicong Cen, Yuting Wei, Yuxin Chen, Yuejie Chi

Viaarxiv icon

Escaping Saddle Points in Heterogeneous Federated Learning via Distributed SGD with Communication Compression

Oct 29, 2023
Sijin Chen, Zhize Li, Yuejie Chi

Figure 1 for Escaping Saddle Points in Heterogeneous Federated Learning via Distributed SGD with Communication Compression
Figure 2 for Escaping Saddle Points in Heterogeneous Federated Learning via Distributed SGD with Communication Compression
Figure 3 for Escaping Saddle Points in Heterogeneous Federated Learning via Distributed SGD with Communication Compression
Viaarxiv icon