arXiv:2605.27810cs.IR2026-05

用聚类与图结构提升大候选集排序,显著提高准确性与可扩展性

LRanker: LLM Ranker for Massive Candidates

论文配图:LRanker: LLM Ranker for Massive Candidates
图 1 · 摘自论文原文
  • 通过K-means聚类聚合候选信息,构建全局表征
  • 采用图式测试时扩展机制,多嵌入集成提升排序性能
  • 在超百万候选场景下仍保持30%以上性能提升,适合大规模应用

大型语言模型(LLMs)在捕捉语义相关性并跨领域适应方面展现出强大潜力,但现有方法受限于上下文长度和高计算成本,难以应用于候选集达数百万的真实场景。为此,我们提出LRanker框架,专为大规模候选排序设计。该框架包含候选聚合编码器,利用K-means聚类显式建模全局候选信息;以及基于图的测试时扩展机制,将候选分组、生成多个查询嵌入,并通过集成策略融合。通过聚合多样化嵌入而非依赖单一表示,显著提升鲁棒性和表达能力,实现更精准的大规模排序。我们在RBench三个不同规模场景下的七个任务上评估,结果表明:在RBench-Small中提升超过30%,在RBench-Large中MRR提升3-9%,在包含超680万候选的RBench-Ultra场景中仍保持20-30%改进。消融实验验证了关键组件的有效性。综合结果证明LRanker在鲁棒性、可扩展性和有效性上的优势。

原文摘要 · Abstract (English)

Large language models (LLMs) have recently shown strong potential for ranking by capturing semantic relevance and adapting across diverse domains, yet existing methods remain constrained by limited context length and high computational costs, restricting their applicability to real-world scenarios where candidate pools often scale to millions. To address this challenge, we propose LRanker, a framework tailored for large-candidate ranking. LRanker incorporates a candidate aggregation encoder that leverages K-means clustering to explicitly model global candidate information, and a graph-based test-time scaling mechanism that partitions candidates into subsets, generates multiple query embeddings, and integrates them through an ensemble procedure. By aggregating diverse embeddings instead of relying on a single representation, this mechanism enhances robustness and expressiveness, leading to more accurate ranking over massive candidate pools. We evaluate LRanker on seven tasks across three scenarios in RBench with different candidate scales. Experimental results show that LRanker achieves over 30% gains in the RBench-Small scenario, improves by 3-9% in MRR in the RBench-Large scenario, and sustains scalability with 20-30% improvements in the RBench-Ultra scenario with more than 6.8M candidates. Ablation studies further verify the effectiveness of its key components. Together, these findings demonstrate the robustness, scalability, and effectiveness of LRanker for massive-candidate ranking.

大候选排序LLM应用可扩展性聚类

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。