Chunsheng Zuo
Author directory2026
More Than Efficiency: Embedding Compression Improves Domain Adaptation in Dense Retrieval
Chunsheng Zuo | Daniel Khashabi
Proceedings of the First Workshop on Structured Understanding, Retrieval, and Generation in the LLM Era (SURGeLLM 2026)
Chunsheng Zuo | Daniel Khashabi
Proceedings of the First Workshop on Structured Understanding, Retrieval, and Generation in the LLM Era (SURGeLLM 2026)
Dense retrievers powered by pretrained embeddings are widely used for document retrieval but struggle in specialized domains due to the mismatches between the training and target domain distributions. Domain adaptation typically requires costly annotation and retraining of query-document pairs. In this work, we revisit an overlooked alternative: applying PCA to domain embeddings to derive lower-dimensional representations that preserve domain-relevant features while discarding non-discriminative components. Though traditionally used for efficiency, we demonstrate that this simple embedding compression can effectively improve retrieval performance. Evaluated across 9 retrievers and 14 MTEB datasets, PCA applied solely to query embeddings improves NDCG@10 in 75.4% of model-dataset pairs, offering a simple and lightweight method for domain adaptation.
Rethinking LoRA Memory Through the Lens of KV Cache Compression
Chunsheng Zuo | Liaoyaqi Wang | William Jurayj | William Fleshman | Benjamin Van Durme
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Chunsheng Zuo | Liaoyaqi Wang | William Jurayj | William Fleshman | Benjamin Van Durme
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Parametric retrieval augmentation encodes document information into lightweight, document-specific modules such as LoRA adapters, reducing the need to include all evidence as input context. However, it remains unclear how this parameter-side memory interacts with context-side memory stored in the KV cache. We study this interaction in document-level question answering by progressively evicting document key-value states and measuring when a document LoRA contributes beyond the retained context. We find that document LoRA adds little when the KV cache is largely intact, but becomes increasingly useful under aggressive compression, recovering 13–21 ROUGE-L points when no document context remains. The gain is largest when the base model encodes the document, and the adapter is applied only during answer generation, suggesting that LoRA serves as decoding-time parametric memory rather than as a document encoder. Finally, QA-style supervision produces substantially stronger adapters than raw-context, continuation, or context-plus-QA objectives. These results position document LoRA as a complementary memory channel whose value emerges precisely when context-side evidence is scarce.
2025
GRPO-LEAD: A Difficulty-Aware Reinforcement Learning Approach for Concise Mathematical Reasoning in Language Models
Jixiao Zhang | Chunsheng Zuo
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Jixiao Zhang | Chunsheng Zuo
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Group Relative Policy Optimization (GRPO), which is widely adopted by R1-like reasoning models, has advanced mathematical reasoning. Nevertheless, GRPO faces challenges in reward sparsity, verbosity, and inadequate focus on problem difficulty. We propose GRPO-LEAD, enhancing GRPO with: (1) length-regularized rewards to encourage conciseness while maintaining accuracy; (2) explicit penalties for incorrect solutions to improve model precision; and (3) difficulty-aware advantage reweighting for robust generalization on challenging problems. Comprehensive evaluations demonstrate that GRPO-LEAD significantly improves reasoning accuracy, conciseness, and efficiency. Our approach achieves state-of-the-art performance for 14B-scale models, underscoring the synergy of our methods with appropriate model scale and high-quality data. Our source code, generated dataset, and models are available at https://github.com/aeroplanepaper/GRPO-LEAD.
Position Information Emerges in Causal Transformers Without Positional Encodings via Similarity of Nearby Embeddings
Chunsheng Zuo | Pavel Guerzhoy | Michael Guerzhoy
Proceedings of the 31st International Conference on Computational Linguistics
Chunsheng Zuo | Pavel Guerzhoy | Michael Guerzhoy
Proceedings of the 31st International Conference on Computational Linguistics
Transformers with causal attention can solve tasks that require positional information without using positional encodings. In this work, we propose and investigate a new hypothesis about how positional information can be stored without using explicit positional encoding. We observe that nearby embeddings are more similar to each other than faraway embeddings, allowing the transformer to potentially reconstruct the positions of tokens. We show that this pattern can occur in both the trained and the randomly initialized Transformer models with causal attention and no positional encodings over a common range of hyperparameters.