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

基于强化学习的面向任务的查询重构

Task-Oriented Query Reformulation with Reinforcement Learning

Rodrigo Nogueira, Kyunghyun Cho

摘要:

搜索引擎在我们的日常生活中扮演着重要的角色,帮助我们找到我们需要的信息。然而,当我们输入一个复杂的查询时,结果往往远远不能令人满意。在这项工作中,我们介绍了一个查询重写系统的基础上,重写一个查询,以最大限度地提高返回的相关文件的数量的神经网络。我们用强化学习来训练这个神经网络。这些操作对应于选择术语来构建重构的查询,而奖励是文档召回。我们在三个数据集上对我们的方法进行了评估,并在召回率方面显示出5-20%的相对改善。此外,我们提出了一个简单的方法来估计一个模型在特定环境下的保守上限性能,并验证了仍然有很大的改进空间。

英文摘要:

Search engines play an important role in our everyday lives by assisting us in finding the information we need. When we input a complex query, however, results are often far from satisfactory. In this work, we introduce a query reformulation system based on a neural network that rewrites a query to maximize the number of relevant documents returned. We train this neural network with reinforcement learning. The actions correspond to selecting terms to build a reformulated query, and the reward is the document recall. We evaluate our approach on three datasets against strong baselines and show a relative improvement of 5-20% in terms of recall. Furthermore, we present a simple method to estimate a conservative upper-bound performance of a model in a particular environment and verify that there is still large room for improvements.

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

↑