Research at the intersection of data management, knowledge-intensive AI, and reliable agent behavior.

Yushi Sun

孙宇石

Senior Researcher at Tencent Games · LightSpeed Studios

Ph.D. from HKUST (2025), advised by Prof. Lei Chen

My research vision is to make AI systems reason reliably over data. Rooted in data management, I develop retrieval-augmented generation and agent memory that turn raw data and knowledge into trustworthy LLM and agent behavior — from question answering to scientific discovery.

Email Google Scholar GitHub LinkedIn CV Chinese CV

Research Research Interests

I build data-centric methods and benchmarks for language models that must retrieve, remember, and reason over structured and unstructured knowledge.

Reliable Retrieval-Augmented Generation

Systems and benchmarks for grounding LLM behavior in external data, knowledge graphs, and verifiable retrieval pipelines.

Agent Memory and Long-Horizon Reasoning

Memory structures, freshness checks, and anchoring mechanisms for agents operating across extended conversations and tasks.

Data Management for AI Systems

Methods for semantic typing, text-to-SQL, taxonomy reasoning, and scientific data interfaces that connect models with real data.

Tip: click a topic to filter papers · Shift-click to add more topics.

Retrieval-Augmented Generation (RAG) Agent Memory Knowledge Graphs Text-to-SQL Multimodal LLMs AI4S Data Management

Papers Selected Publications

* Equal contribution ⁺ Corresponding author

Preprints

Are LLMs Ready for Scientific Discovery? A Capability-Oriented Benchmark for AI Scientists

Preprint 2026 Chuhan Shi, Xiaoquan Ren, Sicheng Song, Haobo Li, Rui Sheng⁺, Yushi Sun⁺

Paper

When Classic Cache Policies Fail: Learning-Augmented Replacement for Semantic Retrieval Buffers

Preprint 2026 Yushi Sun*, Bowen Cao*, Wai Lam

Paper

STALE: Can LLM Agents Know When Their Memories Are No Longer Valid?

Preprint 2026 Hanxiang Chao*, Yihan Bai*, Rui Sheng, Tianle Li, Yushi Sun⁺

Paper

GRAVITY: Architecture-Agnostic Structured Anchoring for Long-Horizon Conversational Memory

Preprint 2026 Yushi Sun*, Bowen Cao*, Dong Fang⁺, Lingfeng Su, Wai Lam

Paper

LakeHopper: Cross Data Lakes Column Type Annotation through Model Adaptation

Preprint 2026 Yushi Sun, Xijia Li, Nan Tang, Quanqing Xu, Chuanhui Yang, Lei Chen

Paper

MedKGI: Iterative Differential Diagnosis with Medical Knowledge Graphs and Information-Guided Inquiring

Preprint 2026 Qipeng Wang, Rui Sheng, Yafei Li, Huamin Qu, Yushi Sun⁺, Min Zhu⁺

Paper

CacheRAG: A Semantic Caching System for Retrieval-Augmented Generation in Knowledge Graph Question Answering

Preprint 2026 Yushi Sun, Lei Chen

Paper Data & Code

Published

VeriLLMed: Interactive Visual Debugging of Medical Large Language Models with Knowledge Graphs

VIS 2026 Yurui Xiang, Xingyi Mao, Rui Sheng, Zixin Chen, Zelin Zang, Yuyang Wu, Haipeng Zeng, Huamin Qu, Yushi Sun⁺, Yanna Lin⁺

Paper

Navigating the Mirage: A Dual-Path Agentic Framework for Robust Misleading Chart Question Answering

MM 2026 Yanjie Zhang, Yafei Li, Rui Sheng, Zixin Chen, Yanna Lin, Huamin Qu, Lei Chen, Yushi Sun⁺

Paper

APEX-SQL: Talking to the Data via Agentic Exploration for Text-to-SQL

KDD 2026 Bowen Cao, Weibin Liao, Yushi Sun⁺, Dong Fang⁺, Haitao Li, Wai Lam

Paper Code Slides (CN)

KERAG: Knowledge-Enhanced Retrieval-Augmented Generation for Advanced Question Answering

EMNLP 2025 Yushi Sun, Kai Sun, Ethan Yifan Xu, Xiao Yang, Xin Luna Dong, Nan Tang, Lei Chen

Paper Data & Code

CRAG — Comprehensive RAG Benchmark

NeurIPS 2024 Xiao Yang*, Kai Sun*, Hao Xin*, Yushi Sun*, Nikita Bhalla, Xiangsen Chen, Sajal Choudhary, Rongze Daniel Gui, Ziran Will Jiang, Ziyu Jiang, Lingkun Kong, Brian Moran, Jiaqi Wang, Ethan Yifan Xu, An Yan, Chenyu Yang, Eting Yuan, Hanwen Zha, Nan Tang, Lei Chen, Nicolas Scheffer, Yue Liu, Nirav Shah, Rakesh Wanga, Anuj Kumar, Wen-tau Yih, Xin Luna Dong

Paper KDD Cup

Are Large Language Models a Good Replacement of Taxonomies?

VLDB 2024 Yushi Sun, Hao Xin, Kai Sun, Ethan Yifan Xu, Xiao Yang, Xin Luna Dong, Nan Tang, Lei Chen

Paper Data & Code Slides

Cross-domain-aware Worker Selection with Training for Crowdsourced Annotation

ICDE 2024 Yushi Sun, Jiachuan Wang, Peng Cheng, Libin Zheng, Lei Chen, Jian Yin

Paper Data & Code Slides

VLDB 2023 Yushi Sun, Hao Xin, Lei Chen

Paper Data & Code Slides