跳到正文
arXiv·· 12 天前AI 评分32

RELER:在嵌入空间用强化学习训练稠密检索模型

Learning to Retrieve via Reinforcement Learning in Embedding Space

AI 导读

RELER 是一个强化学习框架,让现有嵌入模型直接在嵌入空间中学习检索并对齐任务特定奖励。它从以归一化编码器输出为中心的 von Mises-Fisher(vMF)分布中采样单位长度的查询与文档嵌入动作进行训练,以解决对比学习目标不直接优化检索指标或下游任务表现的问题。

来源:arXiv · arxiv.org