Picture for Yuejie Chi

Yuejie Chi

Robust Gymnasium: A Unified Modular Benchmark for Robust Reinforcement Learning

Add code
Feb 27, 2025
Viaarxiv icon

Characterizing the Accuracy-Communication-Privacy Trade-off in Distributed Stochastic Convex Optimization

Add code
Jan 06, 2025
Viaarxiv icon

Vertical Federated Learning with Missing Features During Training and Inference

Add code
Oct 29, 2024
Figure 1 for Vertical Federated Learning with Missing Features During Training and Inference
Figure 2 for Vertical Federated Learning with Missing Features During Training and Inference
Figure 3 for Vertical Federated Learning with Missing Features During Training and Inference
Viaarxiv icon

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

Add code
Oct 28, 2024
Viaarxiv icon

Faster WIND: Accelerating Iterative Best-of-$N$ Distillation for LLM Alignment

Add code
Oct 28, 2024
Viaarxiv icon

Leveraging Multimodal Diffusion Models to Accelerate Imaging with Side Information

Add code
Oct 07, 2024
Figure 1 for Leveraging Multimodal Diffusion Models to Accelerate Imaging with Side Information
Figure 2 for Leveraging Multimodal Diffusion Models to Accelerate Imaging with Side Information
Figure 3 for Leveraging Multimodal Diffusion Models to Accelerate Imaging with Side Information
Figure 4 for Leveraging Multimodal Diffusion Models to Accelerate Imaging with Side Information
Viaarxiv icon

Can We Break the Curse of Multiagency in Robust Multi-Agent Reinforcement Learning?

Add code
Sep 30, 2024
Viaarxiv icon

The Sample-Communication Complexity Trade-off in Federated Q-Learning

Add code
Aug 30, 2024
Figure 1 for The Sample-Communication Complexity Trade-off in Federated Q-Learning
Figure 2 for The Sample-Communication Complexity Trade-off in Federated Q-Learning
Figure 3 for The Sample-Communication Complexity Trade-off in Federated Q-Learning
Figure 4 for The Sample-Communication Complexity Trade-off in Federated Q-Learning
Viaarxiv icon

In-Context Learning with Representations: Contextual Generalization of Trained Transformers

Add code
Aug 19, 2024
Viaarxiv icon

A Sharp Convergence Theory for The Probability Flow ODEs of Diffusion Models

Add code
Aug 05, 2024
Viaarxiv icon