用多视角语义ID提升文本到视频检索的召回质量与效率
Generative Recall, Dense Reranking: Learning Multi-View Semantic IDs for Efficient Text-to-Video Retrieval
- 为每段视频生成多个语义ID,通过查询引导的多视图分词器捕捉多样语义
- 在基准测试中达到与密集检索相当的准确率,索引存储减少10倍
- 适合需要快速大规模视频检索的场景,如视频平台推荐系统
文本到视频检索(TVR)在视频平台中至关重要。基于双模态编码器的密集检索虽精度高,但计算与存储开销随语料库增大而急剧上升。因此,实时大规模应用普遍采用两阶段检索:先由快速召回模型筛选候选集,再由高性能密集重排器精细匹配。由于候选集大幅缩减,重排器可使用任意现成的密集检索模型而不影响效率,这意味着召回模型决定了两阶段TVR的性能上限。近年来,生成式检索(GR)将视频嵌入替换为离散的语义ID,通过解码文本查询生成ID令牌实现检索,具有近乎恒定的推理与存储复杂度。其语义ID通过量化捕获视频高层特征,特别适合快速过滤无关候选。然而,作为两阶段检索中的召回模型,GR存在(i)语义模糊性:每个视频被强制映射至单一语义ID,无法满足多样查询;(ii)跨模态错位:语义ID仅基于视觉特征生成,缺乏文本监督。本文提出生成式召回与密集重排(GRDR),设计新型生成式检索方法以提升召回候选质量。GRDR为每段视频分配多个语义ID,采用查询引导的多视图分词器揭示多样语义路径,并通过共享码本联合训练分词器与生成式检索器,使语义ID成为文本与视频间的语义桥梁。推理时,采用前缀树约束解码生成紧凑候选集,交由密集模型进行细粒度匹配。在多个TVR基准上实验表明,GRDR在精度上媲美强效密集检索器,同时索引存储降低一个数量级,全库检索速度提升最高达300倍。
原文摘要 · Abstract (English)
Text-to-Video Retrieval (TVR) is essential in video platforms. Dense retrieval with dual-modality encoders leads in accuracy, but its computation and storage scale poorly with corpus size. Thus, real-time large-scale applications adopt two-stage retrieval, where a fast recall model gathers a small candidate pool, which is reranked by an advanced dense retriever. Due to hugely reduced candidates, the reranking model can use any off-the-shelf dense retriever without hurting efficiency, meaning the recall model bounds two-stage TVR performance. Recently, generative retrieval (GR) replaces dense video embeddings with discrete semantic IDs and retrieves by decoding text queries into ID tokens. GR offers near-constant inference and storage complexity, and its semantic IDs capture high-level video features via quantization, making it ideal for quickly eliminating irrelevant candidates during recall. However, as a recall model in two-stage TVR, GR suffers from (i) semantic ambiguity, where each video satisfies diverse queries but is forced into one semantic ID; and (ii) cross-modal misalignment, as semantic IDs are solely derived from visual features without text supervision. We propose Generative Recall and Dense Reranking (GRDR), designing a novel GR method to uplift recalled candidate quality. GRDR assigns multiple semantic IDs to each video using a query-guided multi-view tokenizer exposing diverse semantic access paths, and jointly trains the tokenizer and generative retriever via a shared codebook to cast semantic IDs as the semantic bridge between texts and videos. At inference, trie-constrained decoding generates a compact candidate set reranked by a dense model for fine-grained matching. Experiments on TVR benchmarks show GRDR matches strong dense retrievers in accuracy while reducing index storage by an order of magnitude and accelerating up to 300$\times$ in full-corpus retrieval.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。