Lost in a Single Vector: Improving Long-Document Retrieval with Chunk Evidence Aggregation

Shanshan Lyu, Yiwei Wang, Yujun Cai, Jiafeng Guo, Shenghua Liu
6/17/2026
cs.CL

No confident code implementation found for this paper. 1 possible match can be shown below.

Abstract

Dense retrieval ranks one query vector against one document vector. On long documents, this interface can fail when a short but decisive span is weakened during document encoding before ranking. We study this failure mode as document-side early compression and introduce the Evidence Dilution Index (EDI) to measure how far a document-level representation falls below the strongest chunk-level evidence within the same gold document. Guided by this view, we propose DICE (Document Inference via Chunk Evidence), a training-free document-side strategy that splits documents into chunks, encodes them independently with a frozen model, and aggregates them back into a single vector while preserving the standard one-query-one-document interface. On LongEmbed, DICE improves retrieval across four backbones, with the largest gains on slices beyond 4k tokens: for Dream, Passkey >4k rises from 30.0 to 90.0 and Needle >4k from 23.3 to 74.0. Across 12,779 filtered samples, DICE yields lower EDI than the single-vector baseline in 92.8% of cases. These results establish document-level encoding as a practical and underexplored lever for long-document retrieval.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 1 weaker match is hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{lyu2026lost,
  title  = {Lost in a Single Vector: Improving Long-Document Retrieval with Chunk Evidence Aggregation},
  author = {Shanshan Lyu and Yiwei Wang and Yujun Cai and Jiafeng Guo and Shenghua Liu},
  year   = {2026},
  eprint = {2606.18781},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.18781v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2606.18781.svg)](https://codeofpaper.com/paper/2606.18781)

Discussion