A Flexible Approach to Automated RNN Architecture GenerationDownload PDF

12 Feb 2018 (modified: 03 Apr 2024)ICLR 2018 Workshop SubmissionReaders: Everyone
Keywords: reinforcement learning, architecture search, ranking function, recurrent neural networks, recursive neural networks
TL;DR: We define a flexible DSL for RNN architecture generation that allows RNNs of varying size and complexity and propose a ranking function that represents RNNs as recursive neural networks, simulating their performance to decide on the most promising architectures.
Abstract: The process of designing neural architectures requires expert knowledge and extensive trial and error. While automated architecture search may simplify these requirements, the recurrent neural network (RNN) architectures generated by existing methods are limited in both flexibility and components. We propose a domain-specific language (DSL) for use in automated architecture search which can produce novel RNNs of arbitrary depth and width. The DSL is flexible enough to define standard architectures such as the Gated Recurrent Unit and Long Short Term Memory and allows the introduction of non-standard RNN components such as trigonometric curves and layer normalization. Using two different candidate generation techniques, random search with a ranking function and reinforcement learning, we explore the novel architectures produced by the RNN DSL for language modeling and machine translation domains. The resulting architectures do not follow human intuition yet perform well on their targeted tasks, suggesting the space of usable RNN architectures is far larger than previously assumed.
Data: [Multi30K](https://paperswithcode.com/dataset/multi30k), [WikiText-2](https://paperswithcode.com/dataset/wikitext-2)
1 Reply