arXiv·· 12 天前AI 评分32
RELER:在嵌入空间用强化学习训练稠密检索模型
Learning to Retrieve via Reinforcement Learning in Embedding Space
AI 导读
RELER 是一个强化学习框架,让现有嵌入模型直接在嵌入空间中学习检索并对齐任务特定奖励。它从以归一化编码器输出为中心的 von Mises-Fisher(vMF)分布中采样单位长度的查询与文档嵌入动作进行训练,以解决对比学习目标不直接优化检索指标或下游任务表现的问题。
来源:arXiv · arxiv.org