Graded chunk relevance distilled from a query-aware context compressor, no re-labeling needed

antoine_chaffin · x · 2026-11-03

The approach distills graded chunk-level relevance from a query-aware context compressor. The teacher scores every token, and those scores are aggregated over arbitrary chunk boundaries — giving richer supervision that works with any chunking strategy without re-labeling documents.

The attached figure contrasts labeling schemes: the usual label marks one chunk relevant and all others irrelevant, while pplx-emvbed-v2-context scores every chunk (highest for the answer chunk, medium for context chunks, near zero for irrelevant ones). Scores come from the context compression model in Perplexity's answer pipeline; chunk scores form a distribution over the document's chunks that the model learns to match, plus the usual document-level contrastive loss.

Related event: Distilling Graded Chunk Relevance Fixes Contextual Embedding Training(2 posts)→

Original post →

More from Models

Models channel →