vix.ing · top · new · best · stats

The Evolved Transformer

2019/01/30 by David R. So, Chen Liang, Liang Chen +4 · 1 voice · 55 citations
Computer Science · Materials Science · Mathematics · #Computation and Language (cs.CL) #FOS: Computer and information sciences #Machine Learning (cs.LG) #Machine Learning (stat.ML) #Magnetic Properties and Applications #Neural and Evolutionary Computing (cs.NE) #cs.CL #cs.LG #cs.NE #stat.ML

paper · pdf · doi:10.48550/arxiv.1901.11117

ICML version with SOTA results

openalex publication_date 2019/01/30 · arxiv published 2019/01/30 · arxiv created 2019/05/17 · arxiv updated 2019/05/21 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Recent works have highlighted the strength of the Transformer architecture on sequence tasks while, at the same time, neural architecture search (NAS) has begun to outperform human-designed models. Our goal is to apply NAS to search for a better alternative to the Transformer. We first construct a large search space inspired by the recent advances in feed-forward sequence models and then run evolutionary architecture search with warm starting by seeding our initial population with the Transformer. To directly search on the computationally expensive WMT 2014 English-German translation task, we develop the Progressive Dynamic Hurdles method, which allows us to dynamically allocate more resources to more promising candidate models. The architecture found in our experiments -- the Evolved Transformer -- demonstrates consistent improvement over the Transformer on four well-established language tasks: WMT 2014 English-German, WMT 2014 English-French, WMT 2014 English-Czech and LM1B. At a big model size, the Evolved Transformer establishes a new state-of-the-art BLEU score of 29.8 on WMT'14 English-German; at smaller sizes, it achieves the same quality as the original "big" Transformer with 37.6% less parameters and outperforms the Transformer by 0.7 BLEU at a mobile-friendly model size of 7M parameters.

Cited by

Discussions

Related