arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.04227cs.LGcs.AI
收藏

Towards Modular Algorithm Induction

Google Brain(谷歌大脑(Google Brain))

Daniel A. Abolafia, Rishabh Singh, Manzil Zaheer, Charles Sutton

英文摘要:

We present a modular neural network architecture Main that learns algorithms given a set of input-output examples. Main consists of a neural controller that interacts with a variable-length input tape and learns to compose modules together with their corresponding argument choices. Unlike previous approaches, Main uses a general domain-agnostic mechanism for selection of modules and their arguments. It uses a general input tape layout together with a parallel history tape to indicate most recently used locations. Finally, it uses a memoryless controller with a length-invariant self-attention based input tape encoding to allow for random access to tape locations. The Main architecture is trained end-to-end using reinforcement learning from a set of input-output examples. We evaluate Main on five algorithmic tasks and show that it can learn policies that generalizes perfectly to inputs of much longer lengths than the ones used for training.

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

↑