← Search

Yiqun Shen

3 accepted papers

2025

LAVa: Layer-wise KV Cache Eviction with Dynamic Budget Allocation

EMNLP 2025

KV Cache is commonly used to accelerate LLM inference with long contexts, yet its high memory demand drives the need for cache compression. Existing compression methods, however, are largely heuristic and lack dynamic budget allocation. To address this limitation, we introduce a unified framework fo

2025

daDPO: Distribution-Aware DPO for Distilling Conversational Abilities

ACL 2025finding

Large language models (LLMs) have demonstrated exceptional performance across various applications, but their conversational abilities decline sharply as model size decreases, presenting a barrier to their deployment in resource-constrained environments. Knowledge distillation (KD) with Direct Prefe…

2024

Retrospex: Language Agent Meets Offline Reinforcement Learning Critic

EMNLP 2024main

Large language models (LLMs) possess extensive knowledge and commonsense reasoning capabilities, making them valuable for creating powerful agents. However, existing LLM agent frameworks have not fully utilized past experiences for improvement. This work introduces a new LLM-based agent framework ca…