Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

About

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.

Shanshan Lyu, Yiwei Wang, Yujun Cai, Jiafeng Guo, Shenghua Liu• 2026

Related benchmarks

TaskDatasetResultRank
Information RetrievalFollowIR--
55
Long document retrievalLongEmbed
Average Score81.92
8
Information RetrievalFollowIR Core
MAP@100031.14
5
Information RetrievalFollowIR Robust
MAP@100030.27
5
Showing 4 of 4 rows

Other info

Follow for update