publications

Click on a title to see details.

InfoFlow KV: Information-Flow-Aware KV Recomputation for Long Context
[Preprint] Xin Teng*, Canyu Zhang*, Shaoyi Zheng*, Danyang Zhuo, Tianyi Zhou, Shengjie Wang  ·  Under review at ICML 2026
Submodular Context Partitioning and Compression for In-Context Learning
[Preprint] Shaoyi Zheng, Canyu Zhang, Tianyi Zhou, Shengjie Wang
SAGE: Benchmarking and Improving Retrieval for Deep Research Agents
[Preprint] Tiansheng Hu, Yilun Zhao, Canyu Zhang, Arman Cohan, Chen Zhao  ·  Under review at ACL ARR 2026

* Equal contribution