Zexue He

I'm a Postdoc Fellow at Human-Centered AI Institude (HAI) in Stanford University, working with Alex 'Sandy' Pentland , Yejin Choi , and Erik Brynjolfsson. I recieved my Ph.D. in Computer Science and Engineering department at University of California, San Diego, advised by Prof. Julian McAuley, and was honored to be awarded the IBM PhD Fellowship in 2022.

I previously spent two years as a Research Scientist at the MIT-IBM Lab , responsible for the safety and alignment training of IBM's multimodal foundation models (Granite Vision Series). I also worked as an intern at the Microsoft Research and Google.

Contact: zexueh [AT] stanford.edu | zehe [AT] ucsd.edu
Full Publications: Google Scholar

Research Highlights

My research advances human-centered Large Language Models and Vision-Language Models that can remember, reason, and act reliably in complex and failure-prone decision-making settings:

  • When models must be safe and aligned with humans: I work on LLM safety and alignment, building loyal agents and personalized AI systems that reason under uncertainty while respecting diverse human preferences.
  • When the modeling horizon is long: I develop memory-augmented LLMs and agentic architectures that support persistent state tracking, structured long-context reasoning, and multi-session decision-making.
  • When applications are high-stakes: : I design and evaluate the reliability and societal impact LLMs for human-centered domains such as healthcare, low-resource settings, education, and employment.

News

  • [3/2026]: I will present at Gates Foundation Workshop on Future of Personal AI: Global Portable Memory Workshop in Seattle.
  • [2/2026]: I'm co-organizing the first workshop on Emerging Directions in Data for Multimodal Foundation Models at CVPR 2026.
  • [10/2025]: I'm organizing the first workshop on Memory and Vision at ICCV 2025.
  • [7/2025]: I'm organizing the second workshop on Long-Context Foundation Models at ICML 2025.
  • [4/2025]: Invited panlist on New Frontiers in Associative Memories Workshop at ICLR 2025.
  • [4/2025]: Invited talk on Safe AI and NLP for High-Stakes Human-Centric Tasks at Yale.
  • [3/2025]: Invited talk on Trustworthy AI for High-Stakes Human-Centric Tasks at UMiami.
  • [1/2025]: Invited talk on Responsible AI for High-Stakes Human-Centric Domains at Tufts.
  • [1/2025]: Invited talk on Responsible and Trustworthy AI at UCLA.
  • [11/2024] : Invited talk on Responsible AI for Human-centered Challenges at MIT.
  • [10/2024]: Talk on Trustworthy GenAI at UCSD CSE.

Publications
/ view by year / view by topic
Patent
Intelligent system and method of optimizing natural language processing models
Education

Postdoc    2025 - present
                       Human-Centered AI Institute (HAI), Stanford University, U.S.
                       Postdoc Fellow in HAI
                       Advisors: Prof. Alex 'Sandy' Pentland, Prof. Yejin Choi, Prof. Erik Brynjolfsson

Ph.D.          2020 - 2024
                       Computer Science and Engineering, University of California San Diego (UCSD), U.S.
                       Ph.D. student in Computer Science
                       Advisor: Prof. Julian McAuley
                       Committee: Prof. Taylor Berg-Kirkpatrick, Prof. Jingbo Shang, Prof. Zhiting Hu

B.S.              2015 - 2019
                       College of Information Science and Technology, Beijing Normal University (BNU), China
                       B.S. in Computer Science and Technology
Work Experience

MIT-IBM AI Lab, Cambridge, MA
Research Scientist (Prev. Research Intern) • summer 2023 - till now
• Memory-augmented Foundation Models for Efficient Long-Context Modeling
Collaborator: Dr. Dmitry Krotov at MIT-IBM
PI: Prof. Yoon Kim at MIT; Prof. Donghyun Kim at Korea University
        Dr. Leonid Karlinsky and Dr. Rogerio Feris at MIT-IBM
• Foundation Models -- LLM Alignment and Reasoning
Collaborator: Maohao Shen at MIT
PI: Prof.Gregory Wornell at MIT; Dr. Subhro Das at MIT-IBM

Microsoft, Redmond, Washington
Research Intern • summer 2022
Targeted Data Generation
Advisor: Dr. Fereshte Khani Prof. Marco Tulio Ribeiro
Office of Applied Research

NEC Labs, Princeton, NJ
Research Intern • June. 2021 to Sept. 2021
Multimodality Data Representation Learning
Advisor: Dr. Yuncong Chen
Data Science & System Security Group

Microsoft Research Asia, Beijing, China
Research Intern • Oct. 2019 to Dec. 2019
Algorithmic Trading: High-Frequency Time Series Machine Learning and Data Mining
Advisor: Dr. Kan Ren
Machine Learning Group

Google, Beijing, China
Engineering Practicum Intern • Jul. 2017 to Sept. 2017
Knowledge Graph Source Discovery: Wikipedia-like Sites Discovery and Analysis
Advisor: Jiang Bian, team manager
Dataz Group
Research Experiences

Machine Learning Department, Carnegie Mellon University, Pittsburgh, U.S.
Research Intern • Apr. 2018 to Oct. 2018
Robust Learning for Domain Generalization (DG) without Domain Information
Mentor: Haohan Wang, Ph.D. candidate at LTI, CMU
Advisors: Prof. Zachary C. Lipton.
Information Retrieval Group, Tsinghua University, Beijing, China
Research Assistant • Jun. 2017 to May 2018
Investigating Human Examination Behavior on Mobile Search
Advisor: Prof. Yiqun Liu
Key Laboratory of Computational Linguistic, Peking University, Beijing, China
Research Assistant • Dec. 2017 to May 2018
Leveraging Gloss Knowledge in Neural Word Sense Disambiguation (WSD)
Chinese Word Segmentation (CWS) with Character Glyph Embedding
Advisor: Prof. Baobao Chang
Engineering Research Center of Virtual Reality and Applications, Beijing Normal University, Beijing, China
Research Assistant • Jul. 2017 to Feb. 2017
Human Brain's CT-MRI Heterogeneous Data Fusion and Visualization
Advisor: Prof. Yanlin Luo
School of Life Science, Beijing Normal University, Beijing, China
Research Assistant • Nov. 2015 to Sept. 2017
Genetic Biological Parallel Computing System for NP-hard Problems
Advisor: Prof. Xudong Zhu
Scholarships

UCSD Chancellor's Dissertation Medal (Finalist) • 2025 • UCSD
IBM Ph.D. Fellowship • 2022-2024 • IBM
TwoSigma Ph.D. Fellowship Final Nomination • 2022-2024 • TwoSigma
Jacobs School of Engineering Fellowship • 2020-2021 • University of California San Diego
The First-class Scholarship for Academic Excellence • 2018 • Beijing Normal University
The First-class Scholarship for Competition Excellence • 2016, 2017, 2018 • Beijing Normal University
Google Intern Scholarship • 2017 • Google Inc.

Gold Medal in International Genetically Engineered Machine Competition (iGEM) at Boston, Massachusetts, U.S. • 2016
Silver Medal in International Collegiate Programming Contest at Beijing regional site (ACM/ICPC, Beijing) • 2016
Bronze Medal in International Collegiate Programming Contest at Dalian regional site (ACM/ICPC, Dalian) • 2016
Best Female Team in China Collegiate Programming Final Contest (CCPC Final) • 2016