Dynamic Superblock Pruning for Fast Learned Sparse Retrieval

📅 2025-04-23
📈 Citations: 0
✨ Influential: 0
📄 PDF
🤖 AI Summary
To address the low online top-k document retrieval efficiency in learned sparse retrieval, this paper proposes Dynamic Superblock Pruning (DSP). DSP introduces a hierarchical superblock structure—departing from conventional flat blocks or clusters—and models the hierarchical relationships among document blocks to enable early, group-level pruning at the superblock granularity. It further integrates a dynamic threshold-driven selection and pruning strategy, ensuring rank-safety or controllable approximation accuracy under high-relevance competition constraints. The method is fully optimized for single-threaded CPU execution, requiring no specialized hardware. Evaluated on the MS MARCO passage collection, DSP significantly outperforms state-of-the-art sparse retrieval baselines: it achieves substantial speedup in online retrieval while maintaining high recall.

Technology Category

Search and Optimization: Distributed SearchMachine Learning: Learning Preferences or RankingsData Mining & Knowledge Management: Conversational Systems for Recommendation & Retrieval

Application Category

Search and Retrieval-Augmented AI: Web learning to rank, online learning, and counterfactual learning for rankingUser Modeling, Personalization and Recommendation: Fairness-aware retrieval and rankingGraph Algorithms and Modeling for the Web: Querying, indexing, and retrieval in Web-related graphs
📝 Abstract
This paper proposes superblock pruning (SP) during top-k online document retrieval for learned sparse representations. SP structures the sparse index as a set of superblocks on a sequence of document blocks and conducts a superblock-level selection to decide if some superblocks can be pruned before visiting their child blocks. SP generalizes the previous flat block or cluster-based pruning, allowing the early detection of groups of documents that cannot or are less likely to appear in the final top-k list. SP can accelerate sparse retrieval in a rank-safe or approximate manner under a high-relevance competitiveness constraint. Our experiments show that the proposed scheme significantly outperforms state-of-the-art baselines on MS MARCO passages on a single-threaded CPU.
Problem

Research questions and friction points this paper is trying to address.

Accelerates sparse retrieval via superblock pruning
Enables early pruning of irrelevant document groups
Improves efficiency while maintaining relevance competitiveness
Innovation

Methods, ideas, or system contributions that make the work stand out.

Superblock pruning for fast sparse retrieval
Early detection of irrelevant document groups
Rank-safe acceleration under relevance constraints
🔎 Similar Papers
No similar papers found.
P
Parker Carlson
University of California, Santa Barbara
W
Wentai Xie
University of California, Santa Barbara
S
Shanxiu He
University of California, Santa Barbara
T
Tao Yang
University of California, Santa Barbara