arXiv:2412.12504cs.IRcs.LG2024-12被引 7

提升大模型搜索相关性判断能力,解决过自信与分布外失效问题

Boosting LLM-based Relevance Modeling with Distribution-Aware Robust Learning

  • 设计感知分布的损失函数,增强细粒度相关性区分能力
  • 引入分布外样本增广模块,主动筛选难样本提升泛化性
  • 多阶段微调策略兼顾分布内与分布外性能,适合真实搜索场景

随着预训练大语言模型(LLMs)的快速发展,其在相关性建模中的应用显著提升了性能。通常通过在特定标注数据集上微调LLM来判断查询与项目之间的相关性。然而,直接使用LLM进行相关性建模存在两大局限:一是难以胜任复杂任务,如细粒度相关性评估(强相关、弱相关、无关),易产生过度自信;二是面对现实场景中的数据分布偏移时性能显著下降。为此,本文提出一种新的分布感知鲁棒学习框架(DaRL),用于支付宝搜索的相关性建模。我们设计了有效损失函数以增强不同细粒度相关性下的判别能力;提出分布感知样本增广(DASA)模块,利用分布外(OOD)检测技术主动选取原训练集覆盖不足的样本用于微调;同时采用多阶段微调策略,同步提升分布内(ID)与分布外(OOD)性能,缩小两者差距。DaRL已在线部署,服务于支付宝保险产品搜索。

原文摘要 · Abstract (English)

With the rapid advancement of pre-trained large language models (LLMs), recent endeavors have leveraged the capabilities of LLMs in relevance modeling, resulting in enhanced performance. This is usually done through the process of fine-tuning LLMs on specifically annotated datasets to determine the relevance between queries and items. However, there are two limitations when LLMs are naively employed for relevance modeling through fine-tuning and inference. First, it is not inherently efficient for performing nuanced tasks beyond simple yes or no answers, such as assessing search relevance. It may therefore tend to be overconfident and struggle to distinguish fine-grained degrees of relevance (e.g., strong relevance, weak relevance, irrelevance) used in search engines. Second, it exhibits significant performance degradation when confronted with data distribution shift in real-world scenarios. In this paper, we propose a novel Distribution-Aware Robust Learning framework (DaRL) for relevance modeling in Alipay Search. Specifically, we design an effective loss function to enhance the discriminability of LLM-based relevance modeling across various fine-grained degrees of query-item relevance. To improve the generalizability of LLM-based relevance modeling, we first propose the Distribution-Aware Sample Augmentation (DASA) module. This module utilizes out-of-distribution (OOD) detection techniques to actively select appropriate samples that are not well covered by the original training set for model fine-tuning. Furthermore, we adopt a multi-stage fine-tuning strategy to simultaneously improve in-distribution (ID) and OOD performance, bridging the performance gap between them. DaRL has been deployed online to serve the Alipay's insurance product search...

大模型相关性建模鲁棒学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。