Skip to content

Latest commit

 

History

History
47 lines (35 loc) · 2.56 KB

README.md

File metadata and controls

47 lines (35 loc) · 2.56 KB

LexBoost

This repository contains the code to reproduce results in the paper "LexBoost: Improving Lexical Document Retrieval with Nearest Neighbors" accepted at The 24th ACM Symposium on Document Engineering (DocEng 2024).

In this work we extensively evaluate the proposed method LexBoost and show robustness and statistically significant improvements across baselines, datasets and corpus graph construction methods with virtually no additional latency overheads. We also show that LexBoost improves reranking results as well.

The paper can be found here: https://dl.acm.org/doi/10.1145/3685650.3685658

Citation

If you use this code please cite the following paper:

@inproceedings{10.1145/3685650.3685658,
author = {Kulkarni, Hrishikesh and Goharian, Nazli and Frieder, Ophir and MacAvaney, Sean},
title = {LexBoost: Improving Lexical Document Retrieval with Nearest Neighbors},
year = {2024},
isbn = {9798400711695},
publisher = {Association for Computing Machinery},
address = {New York, NY, USA},
url = {https://doi.org/10.1145/3685650.3685658},
doi = {10.1145/3685650.3685658},
booktitle = {Proceedings of the ACM Symposium on Document Engineering 2024},
articleno = {16},
numpages = {10},
keywords = {Corpus Graph, Dense Retrieval, Lexical Retrieval},
location = {San Jose, CA, USA},
series = {DocEng '24}
}

Code

lexboost-main.py

This code evaluates the proposed method on top of BM25 baseline at different fusion parameters, varying number of nearest neighbors from the TCT-ColBERT-HNP based corpus graph on the MSMARCO TREC DL 19 and 20 datasets.

lexboost-baselines.py

This code evaluates the proposed method on top of BM25, PL2, DPH and QLD baselines at different fusion parameters, considering 16 nearest neighbors from the TCT-ColBERT-HNP based corpus graph on the MSMARCO TREC DL 19 and 20 datasets.

lexboost-covid.py

This code evaluates the proposed method on top of BM25 baseline at different fusion parameters, varying number of nearest neighbors from the TCT-ColBERT-HNP based corpus graph on the CORD 19 - TREC COVID dataset.

lexboost-alternategraph.py

This code evaluates the proposed method on top of BM25 baseline at different fusion parameters, varying number of nearest neighbors from the TCT-ColBERT-HNP based and TAS-B based corpus graphs on the MSMARCO TREC DL 19 and 20 datasets.

lexboost-reranking.py

This code evaluates reranking using TAS-B on the proposed method on top of BM25 baseline at different fusion parameters, varying number of nearest neighbors from the TCT-ColBERT-HNP based corpus graph on the MSMARCO TREC DL 19 and 20 datasets.