Fast Interleaved Bidirectional Sequence Generation

Biao Zhang, Ivan Titov, Rico Sennrich


Abstract
Independence assumptions during sequence generation can speed up inference, but parallel generation of highly inter-dependent tokens comes at a cost in quality. Instead of assuming independence between neighbouring tokens (semi-autoregressive decoding, SA), we take inspiration from bidirectional sequence generation and introduce a decoder that generates target words from the left-to-right and right-to-left directions simultaneously. We show that we can easily convert a standard architecture for unidirectional decoding into a bidirectional decoder by simply interleaving the two directions and adapting the word positions and selfattention masks. Our interleaved bidirectional decoder (IBDecoder) retains the model simplicity and training efficiency of the standard Transformer, and on five machine translation tasks and two document summarization tasks, achieves a decoding speedup of ~2x compared to autoregressive decoding with comparable quality. Notably, it outperforms left-to-right SA because the independence assumptions in IBDecoder are more felicitous. To achieve even higher speedups, we explore hybrid models where we either simultaneously predict multiple neighbouring tokens per direction, or perform multi-directional decoding by partitioning the target sequence. These methods achieve speedups to 4x–11x across different tasks at the cost of <1 BLEU or <0.5 ROUGE (on average)
Anthology ID:
2020.wmt-1.62
Volume:
Proceedings of the Fifth Conference on Machine Translation
Month:
November
Year:
2020
Address:
Online
Editors:
Loïc Barrault, Ondřej Bojar, Fethi Bougares, Rajen Chatterjee, Marta R. Costa-jussà, Christian Federmann, Mark Fishel, Alexander Fraser, Yvette Graham, Paco Guzman, Barry Haddow, Matthias Huck, Antonio Jimeno Yepes, Philipp Koehn, André Martins, Makoto Morishita, Christof Monz, Masaaki Nagata, Toshiaki Nakazawa, Matteo Negri
Venue:
WMT
SIG:
SIGMT
Publisher:
Association for Computational Linguistics
Note:
Pages:
503–515
Language:
URL:
https://aclanthology.org/2020.wmt-1.62
DOI:
Bibkey:
Cite (ACL):
Biao Zhang, Ivan Titov, and Rico Sennrich. 2020. Fast Interleaved Bidirectional Sequence Generation. In Proceedings of the Fifth Conference on Machine Translation, pages 503–515, Online. Association for Computational Linguistics.
Cite (Informal):
Fast Interleaved Bidirectional Sequence Generation (Zhang et al., WMT 2020)
Copy Citation:
PDF:
https://aclanthology.org/2020.wmt-1.62.pdf
Video:
 https://slideslive.com/38939588
Code
 bzhangGo/zero
Data
C4WMT 2014