← Search

Kai Hui

10 accepted papers

2025

Inference Scaling for Long-Context Retrieval Augmented Generation

ICLR 2025oral

The scaling of inference computation has unlocked the potential of long-context large language models (LLMs) across diverse settings. For knowledge-intensive tasks, the increased compute is often allocated to incorporate more external knowledge. However, without effectively utilizing such knowledge…

Cited by 25SourcePDFScholar
2024

Beyond Yes and No: Improving Zero-Shot LLM Rankers via Scoring Fine-Grained Relevance Labels

NAACL 2024short

Zero-shot text rankers powered by recent LLMs achieve remarkable ranking performance by simply prompting. Existing prompts for pointwise LLM rankers mostly ask the model to choose from binary relevance labels like “Yes” and “No”. However, the lack of intermediate relevance label options may cause th…

Cited by 78SourcePDFScholar
2024

Large Language Models are Effective Text Rankers with Pairwise Ranking Prompting

NAACL 2024findings

Ranking documents using Large Language Models (LLMs) by directly feeding the query and candidate documents into the prompt is an interesting and practical problem. However, researchers have found it difficult to outperform fine-tuned baseline rankers on benchmark datasets.We analyze pointwise and li…

Cited by 255SourcePDFScholar
2023

How Does Generative Retrieval Scale to Millions of Passages?

EMNLP 2023long main

The emerging paradigm of generative retrieval re-frames the classic information retrieval problem into a sequence-to-sequence modeling task, forgoing external indices and encoding an entire document corpus within a single Transformer. Although many different approaches have been proposed to improve…

Cited by 0SourceScholar
2023

Learning List-Level Domain-Invariant Representations for Ranking

NeurIPS 2023spotlight

Domain adaptation aims to transfer the knowledge learned on (data-rich) source domains to (low-resource) target domains, and a popular method is invariant representation learning, which matches and aligns the data distributions on the feature space. Although this method is studied extensively and ap…

Cited by 9SourcePDFScholar
2023

PaRaDe: Passage Ranking using Demonstrations with LLMs

EMNLP 2023short findings

Recent studies show that large language models (LLMs) can be instructed to effectively perform zero-shot passage re-ranking, in which the results of a first stage retrieval method, such as BM25, are rated and reordered to improve relevance. In this work, we improve LLM-based re-ranking by algorithmi…

Cited by 0SourceScholar
2023

RD-Suite: A Benchmark for Ranking Distillation

NeurIPS 2023poster

The distillation of ranking models has become an important topic in both academia and industry. In recent years, several advanced methods have been proposed to tackle this problem, often leveraging ranking information from teacher rankers that is absent in traditional classification settings. To dat…

Cited by 7SourcePDFScholar
2022

ED2LM: Encoder-Decoder to Language Model for Faster Document Re-ranking Inference

ACL 2022findings

State-of-the-art neural models typically encode document-query pairs using cross-attention for re-ranking. To this end, models generally utilize an encoder-only (like BERT) paradigm or an encoder-decoder (like T5) approach. These paradigms, however, are not without flaws, i.e., running the model on…

Cited by 15SourcePDFScholar
2022

ExT5: Towards Extreme Multi-Task Scaling for Transfer Learning

ICLR 2022poster

Despite the recent success of multi-task learning and transfer learning for natural language processing (NLP), few works have systematically studied the effect of scaling up the number of tasks during pre-training. Towards this goal, this paper introduces ExMix (Extreme Mixture): a massive collectio…

Cited by 222SourcePDFScholar
2022

Transformer Memory as a Differentiable Search Index

NeurIPS 2022accept

In this paper, we demonstrate that information retrieval can be accomplished with a single Transformer, in which all information about the corpus is encoded in the parameters of the model. To this end, we introduce the Differentiable Search Index (DSI), a new paradigm that learns a text-to-text mode…

Cited by 299SourcePDFScholar