Alert button
Picture for Yuejie Chi

Yuejie Chi

Alert button

Prompt-prompted Mixture of Experts for Efficient LLM Generation

Add code
Bookmark button
Alert button
Apr 05, 2024
Harry Dong, Beidi Chen, Yuejie Chi

Viaarxiv icon

Provably Robust Score-Based Diffusion Posterior Sampling for Plug-and-Play Image Reconstruction

Add code
Bookmark button
Alert button
Mar 25, 2024
Xingyu Xu, Yuejie Chi

Viaarxiv icon

Sample Complexity of Offline Distributionally Robust Linear Markov Decision Processes

Add code
Bookmark button
Alert button
Mar 19, 2024
He Wang, Laixi Shi, Yuejie Chi

Viaarxiv icon

Accelerating Convergence of Score-Based Diffusion Models, Provably

Add code
Bookmark button
Alert button
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

Add code
Bookmark button
Alert button
Mar 04, 2024
Yu Huang, Zixin Wen, Yuejie Chi, Yingbin Liang

Viaarxiv icon

Counterfactual Generation with Identifiability Guarantees

Add code
Bookmark button
Alert button
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

Add code
Bookmark button
Alert button
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

Add code
Bookmark button
Alert button
Feb 08, 2024
Jiin Woo, Laixi Shi, Gauri Joshi, Yuejie Chi

Viaarxiv icon

Beyond Expectations: Learning with Stochastic Dominance Made Practical

Add code
Bookmark button
Alert button
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

Add code
Bookmark button
Alert button
Nov 30, 2023
He Wang, Yuejie Chi

Viaarxiv icon