Research Scientist Intern
ByteDance
Research on AI agents for advertising optimization.
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.
2023.08 โ Present
2021.08 โ 2023.01
Find my recent work below, or get in touch to discuss research.
ByteDance
Research on AI agents for advertising optimization.
Learning and decision-making, from language models to sequential decisions.
Reasoning in language models and reliable alignment through preference optimization.
Preference optimization โValue learning and policy optimization under safety constraints.
Constrained optimization, Markov decision processes, and bandits.
Online constrained MDPs โ
View figure โ
IEEE Conference on Decision and Control, 2026.
@inproceedings{fan2026boundary,
title = {Boundary-Seeking Policy Gradient for Safe Reinforcement Learning},
author = {Chenhua Fan and Jiahui Zhu and Yuhang Zhang and Honghao Wei},
booktitle = {IEEE Conference on Decision and Control},
year = {2026},
url = {https://arxiv.org/pdf/2608.10204}
}
View figure โ
Transactions on Machine Learning Research, 2026.
@article{zhu2026tmlr,
title = {Divide and Conquer: Selective Value Learning and Policy Optimization for Offline Safe Reinforcement Learning},
author = {Jiahui Zhu and Honghao Wei and Lei Ying},
journal = {TMLR},
year = {2026},
url = {https://openreview.net/pdf?id=4KYrv6qYMl}
}
View figure โ
International Conference on Learning Representations, 2026.
@inproceedings{zhu2026iclr,
title = {Keep the Best, Forget the Rest: Reliable Alignment with Order-Aware Preference Optimization},
author = {Jiahui Zhu and Yuanjie Shi and Xiyue Peng and Xin Liu and Yan Yan and Honghao Wei},
booktitle = {ICLR},
year = {2026},
url = {https://openreview.net/forum?id=LrHfYPFTtg}
}
View figure โ
International Conference on Machine Learning, 2025.
@inproceedings{zhu2025icml,
title = {Online Constrained Markov Decision Processes},
author = {Jiahui Zhu and Kihyun Yu and Dabeen Lee and Xin Liu and Honghao Wei},
booktitle = {ICML},
year = {2025},
url = {https://openreview.net/forum?id=fFgiXamW8E}
}Washington State University
Washington State University
Invited talk