arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.02593cs.LGcs.CLstat.ML

Learning Architectures from an Extended Search Space for Language Modeling

Northeastern University(东北大学);NiuTrans Research(小牛翻译研究院);Institute of Psychology, CAS(中国科学院心理研究所);Kingsoft AI Lab(金山人工智能研究院)

Yinqiao Li, Chi Hu, Yuhao Zhang, Nuo Xu, Yufan Jiang, Tong Xiao, Jingbo Zhu, Tongran Liu, Changliang Li

英文摘要:

Neural architecture search (NAS) has advanced significantly in recent years but most NAS systems restrict search to learning architectures of a recurrent or convolutional cell. In this paper, we extend the search space of NAS. In particular, we present a general approach to learn both intra-cell and inter-cell architectures (call it ESS). For a better search result, we design a joint learning method to perform intra-cell and inter-cell NAS simultaneously. We implement our model in a differentiable architecture search system. For recurrent neural language modeling, it outperforms a strong baseline significantly on the PTB and WikiText data, with a new state-of-the-art on PTB. Moreover, the learned architectures show good transferability to other systems. E.g., they improve state-of-the-art systems on the CoNLL and WNUT named entity recognition (NER) tasks and CoNLL chunking task, indicating a promising line of research on large-scale pre-learned architectures.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑