中文

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}
}