HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Simple Applications of BERT for Ad Hoc Document Retrieval

Wei Yang; Haotian Zhang; Jimmy Lin

Simple Applications of BERT for Ad Hoc Document Retrieval

Abstract

Following recent successes in applying BERT to question answering, we explore simple applications to ad hoc document retrieval. This required confronting the challenge posed by documents that are typically longer than the length of input BERT was designed to handle. We address this issue by applying inference on sentences individually, and then aggregating sentence scores to produce document scores. Experiments on TREC microblog and newswire test collections show that our approach is simple yet effective, as we report the highest average precision on these datasets by neural approaches that we are aware of.

Code Repositories

castorini/birch
pytorch
Mentioned in GitHub
kasys-lab/anserini-kasys
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
ad-hoc-information-retrieval-on-trec-robust04BERT FT(Microblog)
MAP: 0.3278
P@20: 0.4287

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp