| 2025 | ICLR | 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 |
| 2025 | ICML | From RAG to Memory: Non-Parametric Continual Learning for Large Language Models. | Bernal Jimnez Gutirrez, Yiheng Shu, Weijian Qi, Sizhe Zhou, Yu Su |
| 2024 | EACL | Distribution Shifts Are Bottlenecks: Extensive Evaluation for Grounding Language Models to Knowledge Bases. | Yiheng Shu, Zhiwei Yu |
| 2024 | EMNLP | 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 |
| 2023 | AAAI | Question Decomposition Tree for Answering Complex Questions over Knowledge Bases. | Xiang Huang, Sitao Cheng, Yiheng Shu, Yuheng Bao, Yuzhong Qu |
| 2022 | COLING | Logical Form Generation via Multi-task Learning for Complex Question Answering over Knowledge Bases. | Xixin Hu, Xuan Wu, Yiheng Shu, Yuzhong Qu |
| 2022 | EMNLP | TIARA: Multi-grained Retrieval for Robust Question Answering over Large Knowledge Base. | Yiheng Shu, Zhiwei Yu, Yuhan Li, Brje F. Karlsson, Tingting Ma, Yuzhong Qu, Chin-Yew Lin |
| 2019 | ICWE | Deep Learning-Based Sequential Recommender Systems: Concepts, Algorithms, and Evaluations. | Hui Fang, Guibing Guo, Danning Zhang, Yiheng Shu |