arXiv reaDer
マルチエージェント強化学習による高速シーケンス生成
Fast Sequence Generation with Multi-Agent Reinforcement Learning
自己回帰シーケンス生成モデルは、機械翻訳や画像キャプションなどの分野で最先端のパフォーマンスを実現しています。これらのモデルは、以前に生成された単語を条件付けして各単語を生成するという点で自己回帰であり、推論中に大きな待ち時間が発生します。最近、すべての単語を並列に生成することによって推論時間を高速化するために、非自己回帰デコードが機械翻訳で提案されています。通常、これらのモデルは、単語レベルのクロスエントロピー損失を使用して、各単語を個別に最適化します。ただし、このような学習プロセスでは文レベルの一貫性を考慮することができないため、これらの非自己回帰モデルの生成品質が低下します。この論文では、新しいトレーニングパラダイムである反事実条件節-クリティカルマルチエージェント学習(CMAL)を使用して、非自己回帰シーケンス生成(NAG)のシンプルで効率的なモデルを提案します。 CMALは、NAGをマルチエージェント強化学習システムとして定式化します。このシステムでは、ターゲットシーケンス内の要素の位置が、文レベルの報酬を協調的に最大化することを学習するエージェントと見なされます。 MSCOCO画像キャプションベンチマークでは、NAGメソッドは、最先端の自己回帰モデルに匹敵するパフォーマンスを実現し、13.9倍のデコード速度を実現します。 WMT14 EN-DE機械翻訳データセットでは、この方法は、クロスエントロピーでトレーニングされたベースラインを6.0 BLEUポイント上回り、最大のデコード速度17.46倍を実現します。
Autoregressive sequence Generation models have achieved state-of-the-art performance in areas like machine translation and image captioning. These models are autoregressive in that they generate each word by conditioning on previously generated words, which leads to heavy latency during inference. Recently, non-autoregressive decoding has been proposed in machine translation to speed up the inference time by generating all words in parallel. Typically, these models use the word-level cross-entropy loss to optimize each word independently. However, such a learning process fails to consider the sentence-level consistency, thus resulting in inferior generation quality of these non-autoregressive models. In this paper, we propose a simple and efficient model for Non-Autoregressive sequence Generation (NAG) with a novel training paradigm: Counterfactuals-critical Multi-Agent Learning (CMAL). CMAL formulates NAG as a multi-agent reinforcement learning system where element positions in the target sequence are viewed as agents that learn to cooperatively maximize a sentence-level reward. On MSCOCO image captioning benchmark, our NAG method achieves a performance comparable to state-of-the-art autoregressive models, while brings 13.9x decoding speedup. On WMT14 EN-DE machine translation dataset, our method outperforms cross-entropy trained baseline by 6.0 BLEU points while achieves the greatest decoding speedup of 17.46x.
updated: Sun Jan 24 2021 12:16:45 GMT+0000 (UTC)
published: Sun Jan 24 2021 12:16:45 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト