Ziwei Chai (柴子炜)

yang.png
Email zwchai[at]zju[dot]edu[dot]cn

I am a final-year Ph.D. student at Zhejiang University advised by Prof. Yang Yang. I received my B.E. degree in Software Engineering (2021) from Zhejiang University. My research interests revolve around LLM reasoning, data-centric pre-training, and connecting language models with graph-structured knowledge.

I’m currently an intern at Zhipu AI, where I work on the GLM pre-training data pipeline, focusing on the curation, synthesis, and evaluation of pre-training data, especially in STEM domains. I genuinely enjoy the humble work of helping language models build the boundaries of their world.

Before that, I worked at ByteDance Seed on multi-agent LLM systems, and spent some time at Ant Group working on graph intelligence for financial application.

selected publications

  1. Technical Report
    glm.jpg
    Glm-4.5: Agentic, reasoning, and coding (arc) foundation models
    Contributor to the GLM Team
    arXiv preprint arXiv:2508.06471, 2025
  2. ACL
    acl.jpg
    An Expert is Worth One Token: Synergizing Multiple Expert LLMs as Generalist via Expert Token Routing
    Ziwei Chai, Guoyin Wang, Jing Su, Tianjie Zhang, Xuanwen Huang, Xuwu Wang, Jingjing Xu, and 4 more authors
    In Annual Meeting of the Association for Computational Linguistics, 2024
  3. NeurIPS
    cypher.jpg
    Cypher-RI: Reinforcement Learning for Integrating Schema Selection into Cypher Generation
    Hanchen Su, Xuyuan Li, Yan Zhou, Ziwei Chai, Haozheng Wang, Chen Zhang, Yang Yang, and 1 more author
    In Thirty-ninth Annual Conference on Neural Information Processing Systems, 2025
  4. ICML
    da.jpg
    Infiagent-dabench: Evaluating agents on data analysis tasks
    Xueyu Hu, Ziyu Zhao, Shuang Wei, Ziwei Chai, Guoyin Wang, Xuwu Wang, Jing Su, and 4 more authors
    In International Conference on Machine Learning, 2024
  5. TBD
    graphllm.jpg
    Graphllm: Boosting graph reasoning ability of large language model
    Ziwei Chai, Tianjie Zhang, Liang Wu, Kaiqiao Han, Xiaohai Hu, Xuanwen Huang, and Yang Yang
    IEEE Transactions on Big Data, 2023
  6. AAAI
    aml.jpg
    Towards Learning to Discover Money Laundering Sub-network in Massive Transaction Network
    Ziwei Chai, Yang Yang, Jiawang Dan, Sheng Tian, Changhua Meng, Weiqiang Wang, and Yifei Sun
    In Thirty-Seventh AAAI Conference on Artificial Intelligence, 2023
  7. IJCAI
    amnet.jpg
    Can abnormality be detected by graph neural networks?
    Ziwei Chai, Siqi You, Yang Yang, Shiliang Pu, Jiarong Xu, Haoyang Cai, and Weihao Jiang
    In International Joint Conferences on Artificial Intelligence, 2022