Hi, I'm Kai Zhang.

I'm a founding MTS at NeoCognition. Previously, I got my PhD at OSU NLP and worked at Meta MSL, MSR, and Google DeepMind.

I am interested in multimodal models and agents, and their data solutions.

Find me on , and .


Selected Publications

  • Agent Learning via Early Experience

    Kai Zhang, Xiangchao Chen, Bo Liu, Tianci Xue, Zeyi Liao, Zhihan Liu, Xiyao Wang, Yuting Ning, Zhaorun Chen, Xiaohan Fu, Jian Xie, Yuxuan Sun, Boyu Gou, et al.

    ICML'26 Paper

  • Is Your LLM Secretly a World Model of the Internet? Model-Based Planning for Web Agents

    Yu Gu*, Kai Zhang*, Yuting Ning*, Boyuan Zheng*, Boyu Gou, Tianci Xue, Cheng Chang, Sanjari Srivastava, Yanan Xie, Peng Qi, Huan Sun, Yu Su

    TMLR'25 Paper Code Data

  • MagicLens: Self-Supervised Image Retrieval with Open-Ended Instructions

    Kai Zhang, Yi Luan, Hexiang Hu, Kenton Lee, Siyuan Qiao, Wenhu Chen, Yu Su, Ming-Wei Chang

    ICML'24 Oral (1.5%) Paper Website Code

  • TravelPlanner: A Benchmark for Real-World Planning with Language Agents

    Jian Xie*, Kai Zhang*, Jiangjie Chen, Tinghui Zhu, Renze Lou, Yuandong Tian, Yanghua Xiao, Yu Su

    ICML'24 Spotlight (3.5%) Paper Website Code

  • MMMU: A Massive Multi-discipline Multimodal Understanding and Reasoning Benchmark for Expert AGI

    Xiang Yue*, Yuansheng Ni*, Kai Zhang*, Tianyu Zheng*, Ruoqi Liu, Ge Zhang, Samuel Stevens, Dongfu Jiang, Weiming Ren, Yuxuan Sun, Cong Wei, Botao Yu, Ruibin Yuan, Renliang Sun, Ming Yin, Boyuan Zheng, Zhenzhu Yang, Yibo Liu, Wenhao Huang, Huan Sun, Yu Su, Wenhu Chen

    CVPR'24 Best Paper Finalist (0.2%) Paper Website Code Data