arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2601.03211cs.IRcs.AIcs.CL
收藏

微调小型语言模型作为高效的企业搜索相关性标注器

Fine-tuning Small Language Models as Efficient Enterprise Search Relevance Labelers

Microsoft(微软)
Yue Kang, Zhuoyi Huang, Benji Schussheim, Diana Licon, Dina Atia, Shixing Cao, Jacob Danovitch, Kunho Kim, Billy Norcilien, Jonah Karpman, Mahmound Sayed, Mike … 展开作者

Yue Kang, Zhuoyi Huang, Benji Schussheim, Diana Licon, Dina Atia, Shixing Cao, Jacob Danovitch, Kunho Kim, Billy Norcilien, Jonah Karpman, Mahmound Sayed, Mike Taylor, Tao Sun, Pavel Metrikov, Vipul Agarwal, Chris Quirk, Ye-Yi Wang, Nick Craswell, Irene Shaffer, Tianwei Chen, Sulaiman Vesal, Soundar Srinivasan

AI总结:

本文提出通过微调小型语言模型实现高效企业搜索相关性标注,利用合成数据生成提升标注质量和吞吐量,比大型模型更高效且成本更低。

摘要:

在企业搜索中,构建高质量的大规模数据集仍然是一个核心挑战,因为获取标记数据困难。为了解决这一挑战,我们提出了一种高效的微调小型语言模型(SLMs)的方法,用于准确的相关性标注,使能够实现高吞吐量、领域特定的标注,其质量可与甚至优于最先进的大型语言模型(LLMs)。为了克服企业领域中高质量和可获取的数据集的缺乏,我们的方法利用合成数据生成。具体来说,我们使用一个LLM从种子文档中合成现实的企业查询,应用BM25检索困难的负样本,并使用教师LLM分配相关性分数。生成的数据集随后被蒸馏成一个SLM,产生一个紧凑的相关性标注器。我们在一个高质量的基准上评估我们的方法,该基准由923个企业查询-文档对组成,由训练过的人类标注员注释。我们显示,蒸馏后的SLM在与人类判断的一致性上与或优于教师LLM。此外,我们的微调标注器显著提高了吞吐量,实现了17倍的提升,同时成本效率提高了19倍。这种方法使企业级检索应用的可扩展和低成本的相关性标注成为可能,支持现实世界中的快速离线评估和迭代。

英文摘要:

In enterprise search, building high-quality datasets at scale remains a central challenge due to the difficulty of acquiring labeled data. To resolve this challenge, we propose an efficient approach to fine-tune small language models (SLMs) for accurate relevance labeling, enabling high-throughput, domain-specific labeling comparable or even better in quality to that of state-of-the-art large language models (LLMs). To overcome the lack of high-quality and accessible datasets in the enterprise domain, our method leverages on synthetic data generation. Specifically, we employ an LLM to synthesize realistic enterprise queries from a seed document, apply BM25 to retrieve hard negatives, and use a teacher LLM to assign relevance scores. The resulting dataset is then distilled into an SLM, producing a compact relevance labeler. We evaluate our approach on a high-quality benchmark consisting of 923 enterprise query-document pairs annotated by trained human annotators, and show that the distilled SLM achieves agreement with human judgments on par with or better than the teacher LLM. Furthermore, our fine-tuned labeler substantially improves throughput, achieving 17 times increase while also being 19 times more cost-effective. This approach enables scalable and cost-effective relevance labeling for enterprise-scale retrieval applications, supporting rapid offline evaluation and iteration in real-world settings.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑