You Wu
Author directoryOther people with similar names: You Wu, You Wu, You Wu
Unverified author pages with similar names: You Wu
2026
GRKV: Global Regression for Training-Free KV Cache Compression in Long-Context LLMs
Junjie Peng | You Wu | Haoyi Wu | Jialong Han | Xiaohua Xie | Kewei Tu | Jianhuang Lai
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Junjie Peng | You Wu | Haoyi Wu | Jialong Han | Xiaohua Xie | Kewei Tu | Jianhuang Lai
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Large language models (LLMs) with extended context lengths rely on the key-value (KV) cache to support attention over prior tokens. However, maintaining the KV cache incurs substantial memory overhead, motivating KV-cache compression methods that enforce a fixed budget through eviction and merging. Modern eviction methods increasingly adopt span-based retention because preserving contiguous spans is empirically effective and better preserves semantic coherence. Yet, when combined with post-eviction merging, span-based retention concentrates merges onto a small set of span-boundary carrier tokens, producing a highly imbalanced merge pattern that exacerbates over-merging and increases information loss. To address this imbalance, we propose GRKV (Global Regression for KV Cache), a training-free KV-cache merging method that directly minimizes the discrepancy between compressed-cache and full-cache attention outputs. GRKV uses ridge-regression-based merge steps to distribute information from evicted tokens across retained tokens, while regularizing the updates to prevent over-smoothing. Across the LongBench and RULER long-context benchmarks, GRKV is the only merging method that improves overall performance with minimal overhead. Our code is available at https://github.com/pjunjie/GRKV.
SpiralFormer: Looped Transformers Can Learn Hierarchical Dependencies via Multi-Resolution Recursion
Chengting Yu | Xiaobo Shu | Yadao Wang | Yizhen Zhang | Haoyi Wu | You Wu | Rujiao Long | Ziheng Chen | Yuchi Xu | Wenbo Su | Bo Zheng
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Chengting Yu | Xiaobo Shu | Yadao Wang | Yizhen Zhang | Haoyi Wu | You Wu | Rujiao Long | Ziheng Chen | Yuchi Xu | Wenbo Su | Bo Zheng
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Recursive (looped) Transformers decouple computational depth from parameter depth by repeatedly applying shared layers, providing an explicit architectural primitive for iterative refinement and latent reasoning. However, early looped Transformers often underperform non-recursive baselines of equal compute. While recent literature has introduced more effective recursion mechanisms to mitigate this gap, existing architectures still operate at a fixed, full-token resolution, neglecting the potential efficiency of computing over compressed latent representations. In this paper, we propose SpiralFormer, a looped Transformer that executes recurrence under a multi-resolution recursion schedule. We provide probing evidence that multi-resolution recursion enables the model to learn hierarchical dependencies by inducing iteration-wise functional specialization across different scales. Empirically, SpiralFormer achieves better parameter and compute efficiency than both looped and non-looped baselines across model scales from 160M to 1.4B, establishing sequence resolution as a potential axis for scaling recursive architectures.
2025
A Systematic Study of Cross-Layer KV Sharing for Efficient LLM Inference
You Wu | Haoyi Wu | Kewei Tu
Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 2: Short Papers)
You Wu | Haoyi Wu | Kewei Tu
Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 2: Short Papers)
Recently, sharing key-value (KV) cache across layers has been found effective in efficient inference of large language models (LLMs). To systematically investigate different techniques of cross-layer KV sharing, we propose a unified framework that covers several recent methods and their novel variants. We conduct comprehensive experiments on all the configurations of the framework, evaluating their generation throughput and performance in language modeling and downstream tasks. We find that when reducing the size of the KV cache by 2×, most configurations can achieve higher throughput than standard transformers while maintaining competitive performance.When further reducing the size of the KV cache, however, pairing queries of all layers with KVs of upper layers performs better, at the expense of additional training cost and prefilling latency. We hope that this work will help users make more informed choices of cross-layer KV sharing approaches and facilitate future research on efficient LLM inference.