Yiheng Shu

PhD student, The Ohio State University

Columbus, OH shu.251 [at] osu.edu CV

About Me

I’m a doctoral student at The Ohio State University since 2023, advised by Prof. Yu Su.

My research explores retrieval-augmented generation (RAG) and memory for language agents. Specifically, I focus on: 1) developing efficient information retrieval approaches to enhance reasoning and long-term memory capabilities [HippoRAG] [HippoRAG 2] [REMem], and 2) integrating these models with complex external environments such as the Web, knowledge graphs [TIARA], and databases [Middleware].

I received my Master degree from Nanjing University and Bachelor degree from Northeastern University (China). I was a research intern at Google Research, Intuit and Microsoft Research Asia.

Feel free to reach out to me if you’re interested in my research.

Preprint

• AgentCL: Toward Rigorous Evaluation of Continual Learning in Language Agents
  Yiheng Shu, Bernal Jiménez Gutiérrez, Saisri Padmaja Jonnalagedda, Yuguang Yao, Huan Sun, Yu Su
  [paper] [data]

Publications

*: Equal Contribution

• [EMNLP'26 Findings] Building Long-Term LLM Memory with Neo-Davidsonian Discourse Units
  Sizhe Zhou, Yanzhen Shen, Siru Ouyang, Yuyang Bai, Yiheng Shu, Yu Zhang, Jiawei Han

• [ICLR'26] REMem: Reasoning with Episodic Memory in Language Agent
  Yiheng Shu, Saisri Padmaja Jonnalagedda, Xiang Gao, Bernal Jiménez Gutiérrez, Weijian Qi, Kamalika Das, Huan Sun, Yu Su
  [paper] [code] [poster] [video]

• [NeurIPS'25] Mind2Web 2: Evaluating Agentic Search with Agent-as-a-Judge
  Boyu Gou*, Zanming Huang*, Yuting Ning*, Yu Gu, Michael Lin, Weijian Qi, Andrei Kopanev, Botao Yu, Bernal Jiménez Gutiérrez, Yiheng Shu, Chan Hee Song, Jiaman Wu, Shijie Chen, Hanane Nour Moussa, Tianshu Zhang, Jian Xie, Yifei Li, Tianci Xue, Zeyi Liao, Kai Zhang, Boyuan Zheng, Zhaowei Cai, Viktor Rozgic, Morteza Ziyadi, Huan Sun, Yu Su
  [paper] [project] [BibTex] [Workshop]

• [ICML'25] From RAG to Memory: Non-Parametric Continual Learning for Large Language Model
  Bernal Jiménez Gutiérrez*, Yiheng Shu*, Weijian Qi, Sizhe Zhou, Yu Su
  [paper] [code (3k stars)] [data] [poster] [talk] [BibTex]

• [ICLR'25 Oral] Navigating the Digital World as Humans Do: Universal Visual Grounding for GUI Agents
  Boyu Gou, Ruohan Wang, Boyuan Zheng, Yanan Xie, Cheng Chang, Yiheng Shu, Huan Sun, Yu Su
  [paper] [code] [home] [poster] [BibTex]

• [NeurIPS'24] HippoRAG: Neurobiologically Inspired Long-Term Memory for Large Language Models
  Bernal Jiménez Gutiérrez, Yiheng Shu, Yu Gu, Michihiro Yasunaga, Yu Su
  [paper] [code] [proceedings] [poster] [BibTex]

• [EMNLP'24] Middleware for LLMs: Tools Are Instrumental for Language Agents in Complex Environments
  Yu Gu, Yiheng Shu, Hao Yu, Xiao Liu, Yuxiao Dong, Jie Tang, Jayanth Srinivasa, Hugo Latapie, Yu Su
  [paper] [code] [benchmark] [proceedings] [BibTeX]

• [EACL'24 SRW] Distribution Shifts Are Bottlenecks: Extensive Evaluation for Grounding Language Models to Knowledge Bases
  Yiheng Shu, Zhiwei Yu
  [paper] [code] [data] [poster] [video] [proceedings] [BibTeX]

• [AAAI'23] Question Decomposition Tree for Answering Complex Questions over Knowledge Bases
  Xiang Huang, Sitao Cheng, Yiheng Shu, Yuheng Bao, Yuzhong Qu
  [paper] [code] [proceedings] [BibTeX]

• [EMNLP'22] TIARA: Multi-grained Retrieval for Robust Question Answering over Large Knowledge Base
  Yiheng Shu, Zhiwei Yu, Yuhan Li, Börje F. Karlsson, Tingting Ma, Yuzhong Qu, Chin-Yew Lin
  [paper] [code] [data] [poster] [slides] [video] [proceedings] [BibTeX]

• [COLING'22] Logical Form Generation via Multi-task Learning for Complex Question Answering over Knowledge Bases
  Xixin Hu, Xuan Wu, Yiheng Shu, Yuzhong Qu
  [paper] [code] [slides] [video] [proceedings] [BibTeX]

• [ISWC'21] EDG-based Question Decomposition for Complex Question Answering over Knowledge Bases
  Xixin Hu, Yiheng Shu, Xiang Huang, Yuzhong Qu
  [paper] [code] [home] [video] [proceedings] [BibTeX]

• [TOIS'20] Deep Learning for Sequential Recommendation: Algorithms, Influential Factors, and Evaluations
  Hui Fang, Danning Zhang, Yiheng Shu, Guibing Guo
  [paper] [code] [ACM Library] [BibTeX]

• [ICWE'19 tutorial] Deep Learning-based Sequential Recommender Systems: Concepts, Algorithms, and Evaluations
  Hui Fang, Danning Zhang, Guibing Guo, Yiheng Shu
  [slides] [intro] [Springer Link] [BibTeX]

Services

• Conference reviewer: NeurIPS'26, ARR'26, AAAI'26, ICLR'26, NeurIPS'25, ARR'25, ARR'24
• Workshop reviewer: WiNLP’24, ICLR’25 Reasoning and Planning for LLMs
• Journal reviewer: IEEE Trans. Big Data

Awards and Honors

• Excellent Graduates, Nanjing University, 2023
• Excellent Postgraduate Student, Nanjing University, 2022
• First-class Yingcai Scholarship, Nanjing University, 2022
• First-class Academic Scholarship for Master Students, Nanjing University, 2020-2022
• Provincial Excellent Graduates, Education Department of Liaoning Province, 2020
• Meritorious Winner of The Mathematical Contest in Modeling, COMAP, 2019
• Suzhou Industrial Park Scholarship, Northeastern University, 2019
• National Scholarship, Ministry of Education of the P.R. China, 2018
• Pacemaker to Excellent Students, Northeastern University, 2018
• Excellent Student, Northeastern University, 2017

Last update: Aug. 23, 2026