KEEP-BenchEvaluating Evidence Preservation in Context Compression

Ziqing ManDi WuJia-Chen GuJunyi ZhangKai-Wei ChangNanyun Peng
University of California, Los Angeles
Scroll sideways to see the whole figure.

Figure 1. Downstream QA checks only the final answer, so an unfaithful compression that still yields the right answer passes. KEEP-Bench checks each evidence proposition against the compressed context and catches the loss.

Abstract

Context compression can introduce faithfulness issues through an information bottleneck, and QA metrics miss them. KEEP-Bench checks the compressed context itself, on 4,102 cases that require keeping plausible reference resolutions, explicit qualifiers and minority perspectives.

Read the full abstract

Context compression improves long-context efficiency but may introduce faithfulness issues through an information bottleneck. Existing evaluations rely on QA metrics, but a reader can still answer correctly from an unfaithful compression, so these metrics underestimate such loss. To bridge this gap, we introduce KEEP-Bench, a diagnostic benchmark that directly assesses compression faithfulness with 4,102 cases across three challenging settings that require keeping evidence for plausible reference resolutions, explicit qualifiers, and minority perspectives.

Experiments show that even with tailored prompting, the strongest models remain unreliable: despite context windows exceeding 100K tokens and inputs averaging only 3.1K tokens, the best open-weight model (31B) preserves all required evidence in only 66.8% of cases, and the best proprietary model in only 61.6%. Performance also varies sharply across evidence structures, so conclusions from limited structures may be biased.

We further introduce a metric, Evidence Preservation Return (EPR), to relate evidence recall to inference FLOPs, and find that it peaks at moderate scales (4B–8B), where recall remains low. Additional output budget mainly benefits already strong models and widens capability gaps, yet the stronger preservation of these models comes at a steep computational cost.

These findings highlight the importance for future work to investigate compressors with strong information preservation under a small computation budget, rather than relying on a large output budget or model scaling.

Pilot study

Correct answers can hide lost evidence

Lost evidence hurts the reader

On 200 parallel cases, a reader answers correctly in 96.0% of cases from faithful compressions but in only 11.5% from unfaithful ones.

Yet QA accuracy misses part of it

Those 23 cases pass QA even though required evidence is gone, so KEEP-Bench checks the compressed context itself.

Reader answers both questions correctly

A case is correct only if the reader matches both gold answers.

Benchmark

Single-path QA leaves three evidence structures untested

4,102cases
3.1Ktokens per context on average
15,953proposition checks per run
93.0%judge agreement with humans

Keep both interpretations

Underspecified mentions admit two valid interpretations, each with its own reasoning chain, so a faithful compression must keep both.

900 cases, two to four hops, built from MuSiQue with fictional entities so lost evidence cannot be recalled from memory.

Para-Rec: the evidence that separates the two interpretations survives. All-Rec: every proposition in both chains survives.

Parallel Evidence
Query
Who is the current mayor of the city where Eldrin was born?
Evidence Propositions
Eldrin Voss was born in Aetherm.
Aetherm's mayor is Celeste Duvall.
Eldrin Thorne was born in Luminara.
Luminara's mayor is Kaelen Thorne.

Main results

No compressor reliably keeps all the evidence

The best model keeps every proposition in only 66.8% of cases, and the leader changes with the structure: Claude-Haiku-4.5 is first on parallel evidence but trails GPT-5.4-mini on science.

Show
ParallelConditionalImbalanced

Default budget for all runs; RECOMP (abstractive) has no length control. Values are All-Rec unless marked, Overall spans all 4,102 cases, and a green underline marks the best in each column. EPR (Evidence Preservation Return) is 6·√(A/F), with A the overall All-Rec in percent and F the average TFLOPs per case; API models have no FLOP estimate.

Compute

No compressor is both cheap and faithful

Larger models keep more evidence, but compute grows faster than recall, so EPR peaks at moderate sizes. The cheapest method, RECOMP (extractive), keeps all evidence in under 5% of cases.

Each point is one method. Points on a dashed curve share an EPR; up and to the left is better.

Scatter plot of overall All-Rec against average TFLOPs per case on a log scale, for fifteen compressors.

Output budget

A bigger budget mostly helps models that are already strong

The weakest models stay low at every budget, so extra budget widens the gap instead of closing it.

All-Rec at 0.5×, 1× and 2× the default budget; imbalanced pools science and policy. Click a name to toggle its line. Dashed lines are dedicated compressors.

Evidence position

Compressors favor nearby and earlier evidence

We place the two evidence blocks of the 400 two-hop parallel cases side by side, about half the context apart, or at opposite ends, and compress each version with three models.

Adjacent
Middle
Distant
Chain 1 blockChain 2 blockOther paragraphs

Separated evidence is kept together less often

From Adjacent to Middle, all three models lose 2.1 to 8.5 points of Para‑Rec, and Gemma-4-E4B-it recovers at Distant. Both charts average over absolute placement and block order.

Evidence that comes first is kept more often

Recall of each chain's first proposition when its block comes earlier (filled) or later (open). Earlier wins by 2.5 to 10.6 points; the one exception is Gemma-4-E4B-it at Distant.

Case study

Most compressors keep the answers but break the reasoning chains

Only Claude-Haiku-4.5 keeps all four propositions. Ten of the other sixteen keep both counties, so the answers survive, but lose the statements that tie each series to its state.

parallel_2hop_0241
In what county is the community of Lirion found, in the state where the series Midnight Spotlight is set?
p1The drama series Midnight Spotlight is set in Virelia.
p2Lirion (community) is located in Luminara County, Virelia.
p3The documentary series Midnight Spotlight is set in Nocturnia.
p4Lirion (community) is located in Ecliptica County, Nocturnia.
Compressorp1p2p3p4All-Rec
keptlostClick a row to read that output.

Virelia interpretationNocturnia interpretationdistractor

Resources

Use KEEP-Bench

@article{man2026keepbench,
  title   = {{KEEP-Bench}: Evaluating Evidence Preservation in Context Compression},
  author  = {Man, Ziqing and Wu, Di and Gu, Jia-Chen and Zhang, Junyi and Chang, Kai-Wei and Peng, Nanyun},
  year    = {2026}
}