Shunlong Wu
2026
GMSA: Enhancing Context Compression via Group Merging and Layer Semantic Alignment
Jiwei Tang | Zhicheng Zhang | Shunlong Wu | Jingheng Ye | Lichen Bai | Zitai Wang | Tingwei Lu | Lin Hai | Yiming Zhao | Hai-Tao Zheng | Hong-Gee Kim
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Jiwei Tang | Zhicheng Zhang | Shunlong Wu | Jingheng Ye | Lichen Bai | Zitai Wang | Tingwei Lu | Lin Hai | Yiming Zhao | Hai-Tao Zheng | Hong-Gee Kim
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Large Language Models (LLMs) have achieved remarkable performance across a wide range of Natural Language Processing (NLP) tasks. However, in long-context scenarios, they face two challenges: high computational cost and information redundancy. To address these challenges, we propose GMSA, an encoder-decoder context compression framework that generates a compact sequence of soft tokens for downstream tasks. GMSA introduces Group Merging to achieve more uniform aggregation, mitigating semantic dominance during autoencoder pretraining, and Layer Semantic Alignment (LSA) to bridge the semantic gap between high-level abstract semantics and low-level input semantics. We first pretrain GMSA as an autoencoder and then fine-tune it for downstream tasks. Experiments demonstrate that GMSA improves context reconstruction compared to existing soft prompt compression paradigm and outperforms baselines on multiple long-context question answering and summarization benchmarks across two backbone models, while maintaining low end-to-end latency.
2025
DAST: Context-Aware Compression in LLMs via Dynamic Allocation of Soft Tokens
Shaoshen Chen | Yangning Li | Zishan Xu | Yongqin Zeng | Shunlong Wu | Xinshuo Hu | Zifei Shan | Xin Su | Jiwei Tang | Yinghui Li | Hai-Tao Zheng
Findings of the Association for Computational Linguistics: ACL 2025
Shaoshen Chen | Yangning Li | Zishan Xu | Yongqin Zeng | Shunlong Wu | Xinshuo Hu | Zifei Shan | Xin Su | Jiwei Tang | Yinghui Li | Hai-Tao Zheng
Findings of the Association for Computational Linguistics: ACL 2025
Large Language Models (LLMs) face computational inefficiencies and redundant processing when handling long context inputs, prompting a focus on compression techniques. While existing semantic vector-based compression methods achieve promising performance, these methods fail to account for the intrinsic information density variations between context chunks, instead allocating soft tokens uniformly across context chunks. This uniform distribution inevitably diminishes allocation to information-critical regions. To address this, we propose Dynamic Allocation of Soft Tokens (DAST), a simple yet effective method that leverages the LLM’s intrinsic understanding of contextual relevance to guide compression. DAST combines perplexity-based local information with attention-driven global information to dynamically allocate soft tokens to the informative-rich chunks, enabling effective, context-aware compression. Experimental results across multiple benchmarks demonstrate that DAST surpasses state-of-the-art methods.