| 2025 | AISTATS | Characterizing the Accuracy-Communication-Privacy Trade-off in Distributed Stochastic Convex Optimization. | Sudeep Salgia, Nikola Pavlovic, Yuejie Chi, Qing Zhao |
| 2025 | AISTATS | 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 |
| 2025 | ICASSP | Leveraging Multimodal Diffusion Models to Accelerate Imaging with Side Information. | Timofey Efimov, Harry Dong, Megna Shah, Jeff P. Simmons, Sean Donegan, Yuejie Chi |
| 2025 | ICLR | A Theoretical Analysis of Self-Supervised Learning for Vision Transformers. | Yu Huang, Zixin Wen, Yuejie Chi, Yingbin Liang |
| 2025 | ICLR | 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 |
| 2025 | ICLR | 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 J. Spanos |
| 2025 | ICLR | Vertical Federated Learning with Missing Features During Training and Inference. | Pedro Valdeira, Shiqiang Wang, Yuejie Chi |
| 2025 | ICML | Incentivize without Bonus: Provably Efficient Model-based Online Multi-agent RL for Markov Games. | Tong Yang, Bo Dai, Lin Xiao, Yuejie Chi |
| 2025 | ICML | Breaking the Curse of Multiagency in Robust Multi-Agent Reinforcement Learning. | Laixi Shi, Jingchu Gai, Eric Mazumdar, Yuejie Chi, Adam Wierman |
| 2025 | ICML | 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 |
| 2024 | AISTATS | Escaping Saddle Points in Heterogeneous Federated Learning via Distributed SGD with Communication Compression. | Sijin Chen, Zhize Li, Yuejie Chi |
| 2024 | GLOBECOM | Scalable Dynamic Resource Allocation via Domain Randomized Reinforcement Learning. | Yiqi Wang, Laixi Shi, Martin Hyungwoo Lee, Jaroslaw Sydir, Zhu Zhou, Yuejie Chi, Bin Li |
| 2024 | ICASSP | Communication-Efficient Federated Optimization over Semi-Decentralized Networks. | He Wang, Yuejie Chi |
| 2024 | ICLR | Towards Non-Asymptotic Convergence for Diffusion-Based Generative Models. | Gen Li, Yuting Wei, Yuxin Chen, Yuejie Chi |
| 2024 | ICML | Accelerating Convergence of Score-Based Diffusion Models, Provably. | Gen Li, Yu Huang, Timofey Efimov, Yuting Wei, Yuejie Chi, Yuxin Chen |
| 2024 | ICML | 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 |
| 2024 | ICML | Sample-Efficient Robust Multi-Agent Reinforcement Learning in the Face of Environmental Uncertainty. | Laixi Shi, Eric Mazumdar, Yuejie Chi, Adam Wierman |
| 2024 | ICML | Federated Offline Reinforcement Learning: Collaborative Single-Policy Coverage Suffices. | Jiin Woo, Laixi Shi, Gauri Joshi, Yuejie Chi |
| 2023 | CVPR | 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 |
| 2023 | ICASSP | Deep Unfolded Tensor Robust PCA With Self-Supervised Learning. | Harry Dong, Megna Shah, Sean Donegan, Yuejie Chi |
| 2023 | ICLR | Asynchronous Gradient Play in Zero-Sum Multi-agent Games. | Ruicheng Ao, Shicong Cen, Yuejie Chi |
| 2023 | ICLR | Faster Last-iterate Convergence of Policy Optimization in Zero-Sum Markov Games. | Shicong Cen, Yuejie Chi, Simon Shaolei Du, Lin Xiao |
| 2023 | ICML | The Power of Preconditioning in Overparameterized Low-Rank Matrix Sensing. | Xingyu Xu, Yandi Shen, Yuejie Chi, Cong Ma |
| 2023 | ICML | The Blessing of Heterogeneity in Federated Q-Learning: Linear Speedup and Beyond. | Jiin Woo, Gauri Joshi, Yuejie Chi |
| 2023 | UAI | A trajectory is worth three sentences: multimodal transformer for offline reinforcement learning. | Yiqi Wang, Mengdi Xu, Laixi Shi, Yuejie Chi |
| 2022 | AAAI | 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 |
| 2022 | ACSSC | Harvesting Curvatures for Communication-Efficient Distributed Optimization. | Diogo Cardoso, Boyue Li, Yuejie Chi, Joo Xavier |
| 2022 | AISTATS | Scaling and Scalability: Provable Nonconvex Low-Rank Tensor Completion. | Tian Tong, Cong Ma, Ashley Prater-Bennette, Erin E. Tripp, Yuejie Chi |
| 2022 | ICASSP | Privacy-Preserving Federated Multi-Task Linear Regression: A One-Shot Linear Mixing Approach Inspired By Graph Regularization. | Harlin Lee, Andrea L. Bertozzi, Jelena Kovacevic, Yuejie Chi |
| 2022 | ICASSP | Accelerating ILL-Conditioned Robust Low-Rank Tensor Regression. | Tian Tong, Cong Ma, Yuejie Chi |
| 2022 | ICDM | Active Heterogeneous Graph Neural Networks with Per-step Meta-Q-Learning. | Yuheng Zhang, Yinglong Xia, Yan Zhu, Yuejie Chi, Lei Ying, Hanghang Tong |
| 2022 | ICML | Pessimistic Q-Learning for Offline Reinforcement Learning: Towards Optimal Sample Complexity. | Laixi Shi, Gen Li, Yuting Wei, Yuxin Chen, Yuejie Chi |
| 2021 | COLT | Softmax Policy Gradient Methods Can Take Exponential Time to Converge. | Gen Li, Yuting Wei, Yuejie Chi, Yuantao Gu, Yuxin Chen |
| 2021 | ICIP | Plug-And-Play Image Reconstruction Meets Stochastic Variance-Reduced Gradient Methods. | Vincent Monardo, Abhiram Iyer, Sean Donegan, Marc De Graef, Yuejie Chi |
| 2021 | ICML | Tightening the Dependence on Horizon in the Sample Complexity of Q-Learning. | Gen Li, Changxiao Cai, Yuxin Chen, Yuantao Gu, Yuting Wei, Yuejie Chi |
| 2020 | AISTATS | Communication-Efficient Distributed Optimization in Networks with Gradient Tracking and Variance Reduction. | Boyue Li, Shicong Cen, Yuxin Chen, Yuejie Chi |
| 2020 | CISS | Support Stability of Spike Deconvolution via Total Variation Minimization. | Maxime Ferreira Da Costa, Yuejie Chi |
| 2020 | ICASSP | Manifold Gradient Descent Solves Multi-Channel Sparse Blind Deconvolution Provably and Efficiently. | Laixi Shi, Yuejie Chi |
| 2019 | ACSSC | Self-Calibrated Super Resolution. | Maxime Ferreira Da Costa, Yuejie Chi |
| 2019 | ACSSC | Beyond Procrustes: Balancing-free Gradient Descent for Asymmetric Low-Rank Matrix Sensing. | Cong Ma, Yuanxin Li, Yuejie Chi |
| 2019 | AISTATS | Nonconvex Matrix Factorization from Rank-One Measurements. | Yuanxin Li, Cong Ma, Yuxin Chen, Yuejie Chi |
| 2019 | ICASSP | Shift-invariant Subspace Tracking with Missing Data. | Myung Cho, Yuejie Chi |
| 2019 | ICASSP | On the Sensitivity of Spectral Initialization for Noisy Phase Retrieval. | Vincent Monardo, Yuejie Chi |
| 2019 | ICASSP | Solving Quadratic Equations via Amplitude-based Nonconvex Optimization. | Vincent Monardo, Yuanxin Li, Yuejie Chi |
| 2019 | ICASSP | Improving Graph Trend Filtering with Non-convex Penalties. | Rohan Varma, Harlin Lee, Yuejie Chi, Jelena Kovacevic |
| 2019 | ISIT | Local Geometry of Cross Entropy Loss in Learning One-Hidden-Layer Neural Networks. | Haoyu Fu, Yuejie Chi, Yingbin Liang |
| 2019 | SENSYS | Device-free Multiple People Localization through Floor Vibration. | Laixi Shi, Mostafa Mirshekari, Jonathon Fagert, Yuejie Chi, Hae Young Noh, Pei Zhang, Shijia Pan |
| 2018 | ICASSP | Terahertz Imaging of Binary Reflectance with Variational Bayesian Inference. | Pu Wang, Toshiaki Koike-Akino, Philip V. Orlik, Haoyu Fu, Yuejie Chi |
| 2018 | ICML | Implicit Regularization in Nonconvex Statistical Estimation: Gradient Descent Converges Linearly for Phase Retrieval and Matrix Completion. | Cong Ma, Kaizheng Wang, Yuejie Chi, Yuxin Chen |
| 2017 | ACSSC | Compressive spectrum estimation using quantized measurements. | Haoyu Fu, Yuejie Chi |
| 2016 | ACSSC | Principal subspace estimation for low-rank Toeplitz covariance matrices with binary sensing. | Haoyu Fu, Yuejie Chi |
| 2016 | ICASSP | Robust blind spikes deconvolution. | Yuejie Chi |
| 2016 | ICASSP | Outlier-robust recovery of low-rank positive semidefinite matrices from magnitude measurements. | Yue Sun, Yuanxin Li, Yuejie Chi |
| 2016 | ICML | Provable Non-convex Phase Retrieval with Outliers: Median TruncatedWirtinger Flow. | Huishuai Zhang, Yuejie Chi, Yingbin Liang |
| 2015 | ACSSC | Blind super-resolution of sparse spike signals. | Yuejie Chi |
| 2015 | ACSSC | Change-point estimation of high-dimensional streaming data via sketching. | Yuejie Chi, Yihong Wu |
| 2015 | ICASSP | Compressive graph clustering from random sketches. | Yuejie Chi |
| 2015 | ICASSP | Covariance tracking from sketches of rapid data streams. | Yiran Jiang, Yuejie Chi |
| 2015 | ISIT | Super-resolution of mutually interfering signals. | Yuanxin Li, Yuejie Chi |
| 2014 | ICASSP | Estimation of simultaneously structured covariance matrices from quadratic measurements. | Yuxin Chen, Yuejie Chi, Andrea J. Goldsmith |
| 2014 | ICASSP | Joint sparsity recovery for spectral compressed sensing. | Yuejie Chi |
| 2014 | ISIT | Robust and universal covariance estimation from quadratic measurements via convex programming. | Yuxin Chen, Yuejie Chi, Andrea J. Goldsmith |
| 2013 | ACSSC | Nearest subspace classification with missing data. | Yuejie Chi |
| 2013 | ACSSC | Compressive recovery of 2-D off-grid frequencies. | Yuejie Chi, Yuxin Chen |
| 2013 | ICASSP | Knowledge-enhanced Matching Pursuit. | Yuejie Chi, A. Robert Calderbank |
| 2013 | ICASSP | Analysis of fisher information and the Cramer-Rao bound for nonlinear parameter estimation after compressed sensing. | Pooria Pakrooh, Louis L. Scharf, Ali Pezeshki, Yuejie Chi |
| 2013 | ICML | Spectral Compressed Sensing via Structured Matrix Completion. | Yuxin Chen, Yuejie Chi |
| 2012 | CVPR | Connecting the dots in multi-class classification: From nearest subspace to collaborative representation. | Yuejie Chi, Fatih Porikli |
| 2012 | ICASSP | PETRELS: Subspace estimation and tracking from partial observations. | Yuejie Chi, Yonina C. Eldar, A. Robert Calderbank |
| 2011 | VTC | On Training Signal Design for Multi-User MIMO-OFDM: Performance Analysis and Tradeoffs. | Ahmad Gomaa, Yuejie Chi, Naofal Al-Dhahir, A. Robert Calderbank |
| 2010 | ICASSP | Sensitivity to basis mismatch in compressed sensing. | Yuejie Chi, Ali Pezeshki, Louis L. Scharf, A. Robert Calderbank |
| 2010 | ICIP | Compressive blind source separation. | Yiyue Wu, Yuejie Chi, A. Robert Calderbank |
| 2010 | ISIT | Regularized blind detection for MIMO communications. | Yuejie Chi, Yiyue Wu, A. Robert Calderbank |