About Me

I’m a first-year doctoral student at The Ohio State University, advised by Prof. Yu Su. My research interests include language agents and retrieval-augmented generation (RAG): 1) how to ground large language models to real-world environments, and 2) how to retrieve from environments and augment large language models.

I received my Master degree from Nanjing University in 2023 and Bachelor degree from Northeastern University (CN) in 2020. I was research intern at MSRA in 2022.

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

Preprints

• [arXiv] 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]

Publications

• [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] [project] [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

• Reviewer: ARR 2024