arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1909.05023quant-phcs.CLcs.DScs.LG
收藏

一种用于自然语言处理的量子搜索解码器

A Quantum Search Decoder for Natural Language Processing

University of Cambridge(剑桥大学);University of Bristol(布里斯托大学);Heilbronn Institute for Mathematical Research(海尔布隆数学研究所)

Johannes Bausch, Sathyawageeswar Subramanian, Stephen Piddock

摘要:

概率语言模型,例如基于LSTM的模型,经常面临从一组令牌上的随机变量序列中找到高概率预测的问题。这通常使用诸如波束搜索的贪婪解码的形式来解决,其中保持解码器的有限数量的最高似然路径(波束宽度),并且在最后选择最大似然路径。在这项工作中,我们构建了一个量子算法,以找到具有高恒定成功概率的全局最优解析(即无限光束宽度)。当解码器的输入分布为指数为$k>0$的幂律时,我们的算法具有运行时$R^{n f(R,k)}$,其中$R$是字母表大小,$n$是输入长度;这里是$f<1/2$,并且随着$k$的增加,$f\rightarrow 0$以指数方式快速,因此使我们的算法总是比经典算法快二倍以上。我们进一步修改我们的程序,以恢复一个有限的波束宽度变量,这使得一个更强的经验加速,同时仍然保持更高的精度比可能的经典。最后,我们将这个量子束搜索解码器应用于Mozilla实现的百度DeepSpeech神经网络,我们显示它表现出这样的幂律词排名频率。

英文摘要:

Probabilistic language models, e.g. those based on an LSTM, often face the problem of finding a high probability prediction from a sequence of random variables over a set of tokens. This is commonly addressed using a form of greedy decoding such as beam search, where a limited number of highest-likelihood paths (the beam width) of the decoder are kept, and at the end the maximum-likelihood path is chosen. In this work, we construct a quantum algorithm to find the globally optimal parse (i.e. for infinite beam width) with high constant success probability. When the input to the decoder is distributed as a power-law with exponent $k>0$, our algorithm has runtime $R^{n f(R,k)}$, where $R$ is the alphabet size, $n$ the input length; here $f<1/2$, and $f\rightarrow 0$ exponentially fast with increasing $k$, hence making our algorithm always more than quadratically faster than its classical counterpart. We further modify our procedure to recover a finite beam width variant, which enables an even stronger empirical speedup while still retaining higher accuracy than possible classically. Finally, we apply this quantum beam search decoder to Mozilla's implementation of Baidu's DeepSpeech neural net, which we show to exhibit such a power law word rank frequency.

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

↑