Skip to content

artefactory/abstention-reranker

Repository files navigation

Abstention Reranker

Reference article: "Towards Trustworthy Reranking: A Simple yet Effective Abstention Mechanism".

Abstract

Neural Information Retrieval (NIR) has significantly improved upon heuristic-based IR systems, yet failures remain frequent, as the models used sometimes do not manage to retrieve documents relevant to the user's query. We address these challenges by proposing a lightweight abstention mechanism tailored for real-world constraints, with particular emphasis placed on the reranking phase. We introduce a protocol for evaluating abstention strategies in a black-box scenario, demonstrating their efficacy, especially with access to reference data. We provide code for replicating experiments and implementing abstention mechanisms, fostering wider adoption and application in diverse contexts.

Installation

pip install -r requirements.txt

Computation of relevance scores

python scripts/run_on_datasets.py --config-path <path to config YML>

Experiment replication

See /notebooks/plots.ipynb.

Usage examples

See /notebooks/implem.ipynb.

Reference

If you found our work useful, please consider citing:

@misc{gisserotboukhlef2024trustworthy,
    title={Towards Trustworthy Reranking: A Simple yet Effective Abstention Mechanism}, 
    author={Hippolyte Gisserot-Boukhlef and Manuel Faysse and Emmanuel Malherbe and Céline Hudelot and Pierre Colombo},
    year={2024},
    eprint={2402.12997},
    archivePrefix={arXiv},
    primaryClass={cs.IR}

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published