PyTerrier-GenRank:用于基于大语言模型的重排序的 PyTerrier 插件
信息检索
2024-12-10 v1 人工智能
计算与语言
摘要
使用大语言模型(LLM)作为重排序器需要探索各种超参数,如提示格式、模型选择和改写策略。我们介绍 PyTerrier-GenRank,一个便于与 LLM 进行无缝重排序实验的 PyTerrier 插件,支持诸如点排序(pointwise)和列表排序(listwise)等流行排序策略。我们通过 HuggingFace 和 OpenAI 托管的端点验证了该插件。
引用
@article{arxiv.2412.05339,
title = {PyTerrier-GenRank: The PyTerrier Plugin for Reranking with Large Language Models},
author = {Kaustubh D. Dhole},
journal= {arXiv preprint arXiv:2412.05339},
year = {2024}
}