Insertion-based Decoding with Automatically Inferred Generation Order

Jiatao Gu, Qi Liu, Kyunghyun Cho


Abstract
Conventional neural autoregressive decoding commonly assumes a fixed left-to-right generation order, which may be sub-optimal. In this work, we propose a novel decoding algorithm— InDIGO—which supports flexible sequence generation in arbitrary orders through insertion operations. We extend Transformer, a state-of-the-art sequence generation model, to efficiently implement the proposed approach, enabling it to be trained with either a pre-defined generation order or adaptive orders obtained from beam-search. Experiments on four real-world tasks, including word order recovery, machine translation, image caption, and code generation, demonstrate that our algorithm can generate sequences following arbitrary orders, while achieving competitive or even better performance compared with the conventional left-to-right generation. The generated sequences show that InDIGO adopts adaptive generation orders based on input information.
Anthology ID:
Q19-1042
Volume:
Transactions of the Association for Computational Linguistics, Volume 7
Month:
Year:
2019
Address:
Cambridge, MA
Editors:
Lillian Lee, Mark Johnson, Brian Roark, Ani Nenkova
Venue:
TACL
SIG:
Publisher:
MIT Press
Note:
Pages:
661–676
Language:
URL:
https://aclanthology.org/Q19-1042
DOI:
10.1162/tacl_a_00292
Bibkey:
Cite (ACL):
Jiatao Gu, Qi Liu, and Kyunghyun Cho. 2019. Insertion-based Decoding with Automatically Inferred Generation Order. Transactions of the Association for Computational Linguistics, 7:661–676.
Cite (Informal):
Insertion-based Decoding with Automatically Inferred Generation Order (Gu et al., TACL 2019)
Copy Citation:
PDF:
https://aclanthology.org/Q19-1042.pdf
Data
MS COCO