NAACL 2021long33 citations

Non-Autoregressive Translation by Learning Target Categorical Codes

Yu Bao, Shujian Huang, Tong Xiao, Dongqi Wang, Xinyu Dai, Jiajun Chen

Abstract

Non-autoregressive Transformer is a promising text generation model. However, current non-autoregressive models still fall behind their autoregressive counterparts in translation quality. We attribute this accuracy gap to the lack of dependency modeling among decoder inputs. In this paper, we propose CNAT, which learns implicitly categorical codes as latent variables into the non-autoregressive decoding. The interaction among these categorical codes remedies the missing dependencies and improves the model capacity. Experiment results show that our model achieves comparable or better performance in machine translation tasks than several strong baselines.

BibTeX
@inproceedings{bao-etal-2021-non,
    title = "Non-Autoregressive Translation by Learning Target Categorical Codes",
    author = "Bao, Yu  and
      Huang, Shujian  and
      Xiao, Tong  and
      Wang, Dongqi  and
      Dai, Xinyu  and
      Chen, Jiajun",
    editor = "Toutanova, Kristina  and
      Rumshisky, Anna  and
      Zettlemoyer, Luke  and
      Hakkani-Tur, Dilek  and
      Beltagy, Iz  and
      Bethard, Steven  and
      Cotterell, Ryan  and
      Chakraborty, Tanmoy  and
      Zhou, Yichao",
    booktitle = "Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies",
    month = jun,
    year = "2021",
    address = "Online",
    publisher = "Association for Computational Linguistics",
    url = "https://aclanthology.org/2021.naacl-main.458/",
    doi = "10.18653/v1/2021.naacl-main.458",
    pages = "5749--5759"
}
Non-Autoregressive Translation by Learning Target Categorical Codes · NAACL 2021