SeDeM: Selective Decompression of Hidden-State Memories for Long-Context Question Answering

Maryam Haghifam, Jason Cong, Yizhou Sun


Abstract
Long-context inference with large language models (LLMs) is costly: self-attention during prefill scales quadratically with sequence length, and the key-value (KV) cache grows with the number of processed tokens. Larger context windows also do not ensure reliable evidence use. Context compression reduces this cost, but many soft-compression methods use LLMs as compressors and rely on compact memory tokens both to preserve information and to condition the decoder. We propose SeDeM, a selective decompression framework that decouples compact memory storage from decoder conditioning. SeDeM stores context as compact hidden-state memory blocks, selects query-relevant blocks, and decompresses only the selected blocks for decoder conditioning. Thus, the decoder avoids both full-context processing and direct generation from highly compressed memory slots. On four long-context QA benchmarks, SeDeM achieves higher QA scores than the compression baselines in our main comparison in both 1B and 3B same-backbone settings, and with the 3B backbone exceeds full-context fine-tuning on three datasets. The learned selector uses block-level evidence supervision during training. SeDeM also provides favorable quality–efficiency trade-offs, achieving 1.74–2.46× lower online time-to-first-token and 1.08–1.10× higher autoregressive decoding throughput relative to ICAE while maintaining strong answer quality.
Anthology ID:
2026.emnlp-main.2643
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
60119–60142
Language:
URL:
https://aclanthology.org/2026.emnlp-main.2643/
DOI:
Bibkey:
Cite (ACL):
Maryam Haghifam, Jason Cong, and Yizhou Sun. 2026. SeDeM: Selective Decompression of Hidden-State Memories for Long-Context Question Answering. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 60119–60142, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
SeDeM: Selective Decompression of Hidden-State Memories for Long-Context Question Answering (Haghifam et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.2643.pdf