Ziyang Wu
Author directory2026
EvalMem: An Operation-Level Diagnostic Framework for Long-Term Memory Systems
Zeyu Liu | Jian Zhong | Rongduo Han | Ziyang Wu | Shunye Tang | Chenghao He | Yaxuan Yang | Yihang Qiu | Ailing Wang | Xiao Liang | Guohuan Xie | Xiaokang Xue | Gongchen Li | Haining Zhang | Wei Wang
Findings of the Association for Computational Linguistics: EMNLP 2026
Zeyu Liu | Jian Zhong | Rongduo Han | Ziyang Wu | Shunye Tang | Chenghao He | Yaxuan Yang | Yihang Qiu | Ailing Wang | Xiao Liang | Guohuan Xie | Xiaokang Xue | Gongchen Li | Haining Zhang | Wei Wang
Findings of the Association for Computational Linguistics: EMNLP 2026
Long-horizon interactions with LLM-based assistants require memory systems that preserve and update user states, preferences, and interaction histories. Existing evaluations report end-to-end QA accuracy and cannot determine whether errors arise from encoding, retrieval, or generation. We introduce **EvalMem**, an operation-level diagnostic framework with three parallel Examiners. For each query, the Encoding Examiner checks whether the target fact is stored, the Retrieval Examiner assesses whether the native retriever returns usable evidence, and the Generation Examiner tests whether the model can answer from oracle evidence. Their outputs form fine-grained multi-label defect codes. To improve store-level diagnosis, we adapt agentic RAG with a recall-first strategy that searches using both the query and source evidence, increasing recall of present evidence on LoCoMo from 70.2% to 95.6%. Evaluations of seven memory systems on LoCoMo, LongMemEval-S, and dynamic DynaMem-Bench identify retrieval as the most frequently attributed failure layer; in default LoCoMo, retrieval defects reach 22.1%, compared with 7.7% for encoding and 6.5% for generation. Guided by this diagnosis, **MemWiki**, a search-friendly auxiliary structure built from each system’s memory export, improves mean accuracy by 2.5 and 2.3 percentage points on LoCoMo and LongMemEval-S, respectively.
2024
LLoCO: Learning Long Contexts Offline
Sijun Tan | Xiuyu Li | Shishir G Patil | Ziyang Wu | Tianjun Zhang | Kurt Keutzer | Joseph E. Gonzalez | Raluca Ada Popa
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing
Sijun Tan | Xiuyu Li | Shishir G Patil | Ziyang Wu | Tianjun Zhang | Kurt Keutzer | Joseph E. Gonzalez | Raluca Ada Popa
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing
Processing long contexts remains a challenge for large language models (LLMs) due to the quadratic computational and memory overhead of the self-attention mechanism and the substantial KV cache sizes during generation. We propose LLoCO, a novel approach to address this problem by learning contexts offline through context compression and in-domain parameter-efficient finetuning with LoRA. Our method enables an LLM to create a concise representation of the original context and efficiently retrieve relevant information to answer questions accurately. Our approach extends the effective context window of a 4k token LLaMA2-7B model to handle up to 128k tokens. We evaluate our approach on several long-context question-answering datasets, demonstrating that LLoCO significantly outperforms in-context learning while using 30 × fewer tokens during inference. LLoCO achieves up to 7.62 × speed-up during inference and 11.52 × higher throughput during finetuning, substantially reduces the cost of long document question answering. This makes it a promising solution for efficient long context processing.