arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2110.03389cs.CLcs.AI
收藏

用于神经响应生成的双向策略束搜索

Beam Search with Bidirectional Strategies for Neural Response Generation

IBM(国际商业机器公司);University of Washington(华盛顿大学);Télécom ParisTech, Université Paris Saclay(巴黎萨克雷大学巴黎电信学院);IBM GBS France(IBM全球企业服务法国分部)

Pierre Colombo, Chouchang Yang, Giovanna Varni, Chloé Clavel

摘要:

序列到序列神经网络由于具有学习各种语言模型的灵活能力而被广泛用于基于语言的应用中。然而,当通过训练的神经网络寻求最佳的语言响应时,目前现有的方法,如波束搜索解码器策略,仍然不能达到有前途的性能。我们没有开发基于“规则句子顺序”神经网络(通过从左到右顺序输出句子的训练模型)的各种解码器策略,而是利用“反向”顺序作为额外的语言模型(通过从右到左顺序输出句子的训练模型),这可以为路径查找问题提供不同的视角。在本文中,我们提出了双向策略,在搜索路径相结合的两个网络(从左到右和从右到左的语言模型),使双向束搜索成为可能。此外,我们的解决方案允许我们在句子选择标准中使用任何相似性度量。与单向束搜索策略相比,我们的方法表现出更好的性能。

英文摘要:

Sequence-to-sequence neural networks have been widely used in language-based applications as they have flexible capabilities to learn various language models. However, when seeking for the optimal language response through trained neural networks, current existing approaches such as beam-search decoder strategies are still not able reaching to promising performances. Instead of developing various decoder strategies based on a "regular sentence order" neural network (a trained model by outputting sentences from left-to-right order), we leveraged "reverse" order as additional language model (a trained model by outputting sentences from right-to-left order) which can provide different perspectives for the path finding problems. In this paper, we propose bidirectional strategies in searching paths by combining two networks (left-to-right and right-to-left language models) making a bidirectional beam search possible. Besides, our solution allows us using any similarity measure in our sentence selection criterion. Our approaches demonstrate better performance compared to the unidirectional beam search strategy.

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

↑