Publications
You can also find my articles on my Google Scholar profile.
Memory-Efficient LLM Training by Various-Grained Low-Rank Projection of Gradients
Published in ICLR (Under review), 2026
Recommended citation: Yezhen Wang*, Zhouhao Yang*, et al. "Memory-Efficient LLM Training by Various-Grained Low-Rank Projection of Gradients." arXiv preprint arXiv:2505.01744 (2025).
Fast Intent Classification for LLM Routing via Statistical Analysis of Representations
Published in ICLR (Under review), 2026
Recommended citation: Zhouhao Yang*, Nan Chen*, Soufiane Hayou. "Fast Intent Classification for LLM Routing via Statistical Analysis of Representations." to appear on arxiv
A Two-fold Randomization Framework for Impulse Control Problems
Published in SIAM Journal on Control and Optimization (Under review), 2025
Recommended citation: Haoyang Cao*, Yuchao Dong*, and Zhouhao Yang*, "A Two-fold Randomization Framework for Impulse Control Problems." arXiv preprint arXiv:2509.12018 (2025).
Statistical Mean Estimation with Coded Relayed Observations
Published in IEEE Transactions on Information Theory (under review), 2025
Recommended citation: Yan Hao Ling, Zhouhao Yang, and Jonathan Scarlett. "Statistical Mean Estimation with Coded Relayed Observations." arXiv preprint arXiv:2505.09098 (2025).
Memory-Efficient Gradient Unrolling for Large-Scale Bi-level Optimization
Published in Neurips, 2024
Recommended citation: Qianli Shen, Yezhen Wang, Zhouhao Yang, et al. "Memory-Efficient Gradient Unrolling for Large-Scale Bi-level Optimization." arXiv preprint arXiv:2406.14095 (2024).
Bias-Variance Trade-off in Physics-Informed Neural Networks with Randomized Smoothing for High-Dimensional PDEs
Published in SIAM Journal on Scientific Computing, 2023
Recommended citation: Zheyuan Hu*, Zhouhao Yang*, Yezhen Wang*, George Em Karniadakis, Kenji Kawaguchi, "Bias-variance trade-off in physics-informed neural networks with randomized smoothing for high-dimensional PDEs." SIAM Journal on Scientific Computing 47.4 (2025): C846-C872.
Distributionally Robust Policy Gradient for Offline Contextual Bandits
Published in AISTATS, 2023
Recommended citation: Zhouhao Yang, et al. "Distributionally robust policy gradient for offline contextual bandits." International Conference on Artificial Intelligence and Statistics. PMLR, 2023.
A General Framework for Accurate and Private Mean Estimation
Published in IEEE Signal Processing Letters, 2022
Recommended citation: Zhouhao Yang, Xingyu Xu, and Yuantao Gu. "A general framework for accurate and private mean estimation." IEEE Signal Processing Letters 29 (2022): 2293-2297.