About Me

Hi there! Iโ€™m Jiahui Zhu

My current interests lie at the intersection of LLM reasoning, reinforcement learning, and constrained optimization. Broadly, I work on LLM alignment, constrained reinforcement learning, and bandits.

Washington State University

2023.08 โ€” Present

Boston University

2021.08 โ€” 2023.01

Find my recent work below, or get in touch to discuss research.

Experience

Research Scientist Intern

ByteDance

San Jose, CA, USA โ€” Present

Research on AI agents for advertising optimization.

Research Interests

Learning and decision-making, from language models to sequential decisions.

News

Show more newsShow less news

Selected Publications

Google Scholar โ†—

2026

IEEE CDC 2026Figure 1: Local update geometry of ESPO and BSPG on the feasible and infeasible sides of the active constraint boundary.View figure โ†—

Boundary-Seeking Policy Gradient for Safe Reinforcement Learning

Chenhua Fan, Jiahui Zhu, Yuhang Zhang, Honghao Wei

IEEE Conference on Decision and Control, 2026.

TMLR 2026Research figure for Divide and Conquer: Selective Value Learning and Policy Optimization for Offline Safe Reinforcement LearningView figure โ†—

Divide and Conquer: Selective Value Learning and Policy Optimization for Offline Safe Reinforcement Learning

Jiahui Zhu, Honghao Wei, Lei Ying

Transactions on Machine Learning Research, 2026.

ICLR 2026Research figure for Keep the Best, Forget the Rest: Reliable Alignment with Order-Aware Preference OptimizationView figure โ†—

Keep the Best, Forget the Rest: Reliable Alignment with Order-Aware Preference Optimization

Jiahui Zhu, Yuanjie Shi, Xiyue Peng, Xin Liu, Yan Yan, Honghao Wei

International Conference on Learning Representations, 2026.

2025

ICML 2025Research figure for Online Constrained Markov Decision ProcessesView figure โ†—

Online Constrained Markov Decision Processes

Jiahui Zhu, Kihyun Yu, Dabeen Lee, Xin Liu, Honghao Wei

International Conference on Machine Learning, 2025.

Invited Talks

Generative Pre-Training โ†—

Washington State University

GenAI Seminar

Washington State University

SIGMETRICS 2025

Invited talk