Kehan Guo

Hi, I'm Kehan Guo.

I'm a fourth-year Ph.D. student in the MINE Lab at the University of Notre Dame, advised by Prof. Xiangliang Zhang.

Recently, I've been particularly interested in RL, world models, and reasoning for LLMs.

Research

I build large language models and generative AI systems that reason reliably in complex scientific domains.

My current focus is world modeling for large language models, where models learn and are constrained by rules so they mirror how the world works, and self-evolving LMs that critique, verify, and iteratively improve via tools and interaction loops.

Methodologically I work on RL post-training and alignment, uncertainty calibration, and agentic tool use. Application-wise I emphasize AI for Science with a focus on chemistry, including literature-grounded extraction, retrosynthesis planning, and data curation. I also design complex, cross-domain benchmarks and datasets, turning messy real-world data into structured, auditable tasks with clear provenance and rigorous scoring.

What's New

  • 2025.09 Two papers accepted to NeurIPS 2025: ChemOrch (LLMs + chemical tools) and AdaReasoner — Adaptive Reasoning Enables More Flexible Thinking (Spotlight).
  • 2025.09 New preprint: Causally‑Enhanced Reinforcement Policy Optimization (RL + causal coherence for LLMs). [arXiv](https://arxiv.org/abs/2509.23095)
  • 2025.08 Uncertainty‑aware Yield Prediction accepted to CIKM 2025 — see you in Seoul!
  • 2025.04 Survey “Artificial Intelligence in Spectroscopy” accepted to IJCAI 2025.
  • 2025.02 Check out latest preprint: AI in Spectroscopy, HHH principle
  • 2025.02 I will join Amazon AWS as an Applied Scientist Intern in 2025 Summer. See you in NYC!

Selected Publications

See full list in Publications.

Can LLMs Solve Molecule Puzzles? A Multimodal Benchmark for Molecular Structure Elucidation
Kehan Guo, Bozhao Nan, Yujun Zhou, Taicheng Guo, Zhichun Guo, Mihir Surve, Zhenwen Liang, Nitesh V. Chawla, Olaf Wiest, Xiangliang Zhang
NeurIPS'24 (Spotlight)
What can Large Language Models do in chemistry? A comprehensive benchmark on eight tasks
Taicheng Guo*, Kehan Guo*, Bozhao Nan, Zhenwen Liang, Zhichun Guo, Nitesh V. Chawla, Olaf Wiest, Xiangliang Zhang
NeurIPS'23
ChemOrch: Empowering LLMs with Chemical Intelligence via Synthetic Instructions
Yue Huang, Zhengzhe Jiang, Xiaonan Luo, Kehan Guo, Haomin Zhuang, Yujun Zhou, Zhengqing Yuan, Xiaoqi Sun, Jules Schleinitz, Yanbo Wang, Shuhao Zhang, Mihir Surve, Nitesh V. Chawla, Olaf Wiest, Xiangliang Zhang
NeurIPS'25
AdaReasoner: Adaptive Reasoning Enables More Flexible Thinking
Xiangqi Wang, Yue Huang, Yanbo Wang, Xiaonan Luo, Kehan Guo, Yujun Zhou, Xiangliang Zhang
NeurIPS'25 (Spotlight)
Artificial Intelligence in Spectroscopy: Advancing Chemistry from Prediction to Generation and Beyond
Kehan Guo, Yifan Shen, Gil A. Gonzalez-Montiel, Yue Huang, Yujun Zhou, Mihir Surve, Zhichun Guo, ...
IJCAI'25
Beyond Single-Value Metrics: Evaluating and Enhancing LLM Unlearning with Cognitive Diagnosis
Y. Lang*, Kehan Guo*, Yue Huang, Yujun Zhou, Haomin Zhuang, Tianyu Yang, Yu Su, Xiangliang Zhang
ACL'24 Findings arXiv:2502.13996

Contact

Email: kguo2@nd.edu

If my research interests you, feel free to reach out via email.