Back to Search Start Over

Lightning IR: Straightforward Fine-tuning and Inference of Transformer-based Language Models for Information Retrieval

Authors :
Schlatt, Ferdinand
Fröbe, Maik
Hagen, Matthias
Publication Year :
2024

Abstract

A wide range of transformer-based language models have been proposed for information retrieval tasks. However, including transformer-based models in retrieval pipelines is often complex and requires substantial engineering effort. In this paper, we introduce Lightning IR, an easy-to-use PyTorch Lightning-based framework for applying transformer-based language models in retrieval scenarios. Lightning IR provides a modular and extensible architecture that supports all stages of a retrieval pipeline: from fine-tuning and indexing to searching and re-ranking. Designed to be scalable and reproducible, Lightning IR is available as open-source: https://github.com/webis-de/lightning-ir.<br />Comment: Accepted as a demo at WSDM'25

Details

Database :
arXiv
Publication Type :
Report
Accession number :
edsarx.2411.04677
Document Type :
Working Paper
Full Text :
https://doi.org/10.1145/3701551.3704118