Suozhi Huang ☕️
Suozhi Huang

Grad Student Researcher

About Me

I am a Ph.D. student in Electrical and Computer Engineering at Princeton University. My research focuses on building and evaluating LLM agents for expert-level reasoning and decision-making in complex, dynamic environments. I am particularly interested in agentic reasoning, model adaptation, and reliable evaluation in real-world and scientific domains. I am also interested in Prof. Mengdi Wang’s LabOS initiative and its vision of AI-XR co-scientists that collaborate with human researchers in real-world scientific workflows.

My recent work spans agent evaluation and future prediction, multimodal agent systems, LLM architectures, and formal theorem proving. Before joining Princeton, I received my B.E. in Computer Science from Tsinghua University’s Yao Class. I began my research in collaborative robotic perception and later worked on AI for science and formal mathematical reasoning.

I have been fortunate to work with Prof. Anima Anandkumar at Caltech; Dr. Zheng Yuan and Prof. Dahua Lin through InternLM-Math; and Prof. Chen Feng and Dr. Yiming Li at NYU.

Download CV
Interests
  • Artificial Intelligence
  • Automated Theorem Proving
  • Reasoning in LLMs
Education
  • B.E. Computer Science (Major in AI)

    Yao Class, Tsinghua University

Publications

View all publications on Google Scholar ↗

2026

LabWorld: Learning Interactive 4D Worlds for Scientific Experimentation
Suozhi Huang*, Kelsey Fu*, Binxu Li, Luna Lyu, Kaize Wu, Wenbo Lu, David Smerkous, Jiahan Li, Siqi Zhu, Zhizhou Sha, Jiacheng Guo, Zhixuan Liang, Shilong Liu, Le Cong, and Mengdi Wang
arXiv preprint · * Equal contribution · Paper · Project · Code
GraphCast-Mamba: Learning Non-Markovian Corrections for Weather Forecasting
Ilya Vilkoviskiy, Liang-Hong Mo, Suozhi Huang, and Dmitry A. Abanin
arXiv preprint · Paper
From Gradients to Capabilities: Understanding Multi-Teacher On-Policy Distillation
Siqi Zhu, Suozhi Huang, Kaixuan Zhang, Yuheng Yang, Zhanyang Jin, Yihang Sun, and Jiaxuan You
arXiv preprint · Paper
Can Language Models Learn to Forecast Stock Prices
Jiacheng Guo, Suozhi Huang, Shuzhen Li, Yunlong Gao, Zerui Cheng, Jason Ge, Shushu Liang, Zihao Li, Hao Lu, Ming Yin, Shilong Liu, Jiashuo Liu, Xu Kuang, and Mengdi Wang
arXiv preprint · Paper
AQuA: Recursively Self-Improving Quantitative Trading Research Agents
Jiacheng Guo, Suozhi Huang, Yunlong Gao, Zihao Li, Jason Ge, Xu Kuang, and Mengdi Wang
arXiv preprint · Paper
Agentic Laboratories of the Future: Towards World Models for Scientific Discovery
Kelsey Fu, Luna Lyu, Shuzhen Li, Suozhi Huang, et al.
Preprints.org · Preprint (not peer-reviewed) · Paper
Any2Poster: Any-Source Poster Generation Across Modalities and Domains
Amogh Vinaykumar, Aiden Li, Suozhi Huang, and Shilong Liu
arXiv preprint · Paper · Code
FutureX-Pro: Extending Future Prediction to High-Value Vertical Domains
Jiashuo Liu, Siyuan Chen, Zaiyuan Wang, Zhiyuan Zeng, Jiacheng Guo, Liang Hu, Lingyue Yin, Suozhi Huang, et al.
arXiv preprint · Paper

2025

CryptoBench: A Dynamic Benchmark for Expert-Level Evaluation of LLM Agents in Cryptocurrency
Jiacheng Guo, Suozhi Huang, Zixin Yao, Yifan Zhang, Yifu Lu, Jiashuo Liu, et al.
arXiv preprint · Paper
Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Free
Zihan Qiu, Zekun Wang, Bo Zheng, Zeyu Huang, Kaiyue Wen, Songlin Yang, Rui Men, Le Yu, Fei Huang, Suozhi Huang, Dayiheng Liu, Jingren Zhou, and Junyang Lin
NeurIPS 2025 · Paper
LeanProgress: Guiding Search for Neural Theorem Proving via Proof Progress Prediction
Suozhi Huang, Peiyang Song, Robert Joseph George, and Anima Anandkumar
Transactions on Machine Learning Research · Paper

2024

InternLM2.5-StepProver: Advancing Automated Theorem Proving via Expert Iteration on Large-Scale LEAN Problems
Zijian Wu*, Suozhi Huang*, Zhejian Zhou, Huaiyuan Ying, Jiayu Wang, Dahua Lin, and Kai Chen
Technical report · Paper · Code · Dataset
A Knowledge-Data Dual-Driven Framework for Predicting the Molecular Properties of Rechargeable Battery Electrolytes
Yuchen Gao, Yu-Hang Yuan, Suozhi Huang, Nan Yao, Legeng Yu, Yao-Peng Chen, Qiang Zhang, and Xiang Chen
Angewandte Chemie International Edition · Paper
ActFormer: Scalable Collaborative Perception via Active Queries
Suozhi Huang*, Juexiao Zhang*, Yiming Li, and Chen Feng
ICRA 2024 · Paper · Code · Project