← Search

Jiusheng Chen

2 accepted papers

2021

BANG: Bridging Autoregressive and Non-autoregressive Generation with Large Scale Pretraining

ICML 2021spotlight

In this paper, we propose BANG, a new pretraining model to Bridge the gap between Autoregressive (AR) and Non-autoregressive (NAR) Generation. AR and NAR generation can be uniformly regarded as to what extent previous tokens can be attended, and BANG bridges AR and NAR generation through designing a…

2021

EL-Attention: Memory Efficient Lossless Attention for Generation

ICML 2021spotlight

Transformer model with multi-head attention requires caching intermediate results for efficient inference in generation tasks. However, cache brings new memory-related costs and prevents leveraging larger batch size for faster speed. We propose memory-efficient lossless attention (called EL-attentio…