Listen: BlockRank: A Faster, Smarter Way to Rank Documents with LLMs

BlockRank is a novel method for in-context ranking that uses structured sparse attention and contrastive training to improve LLM efficiency and accuracy.

Listen

Transcript

Large language models are transforming how we search for information, especially through a process called in-context ranking. This is where a model looks at a query and a list of documents to find the best match. But as the document list grows, the computational cost skyrockets. Standard models scale quadratically, meaning doubling the documents can quadruple the processing time. To solve this, researchers have introduced BlockRank. It relies on a key insight: when ranking documents, the model's attention is naturally sparse. It doesn't need to connect every single word across different documents. BlockRank changes how the model pays attention. Document words only look at other words in the same document. Meanwhile, the query words look at everything to make the final decision. This simple shift drops the computational complexity from quadratic to linear. To make it even faster, BlockRank uses a special training method that teaches query words to point directly to the right answer. Instead of slowly generating text word-by-word, the system can simply read the model's internal attention scores to rank the documents instantly. The results are impressive. BlockRank is nearly five times faster than standard models, and it can rank hundreds of documents in less than a second. Best of all, it actually improves accuracy, outperforming existing state-of-the-art rankers.