arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.03629cs.LGcs.AIcs.NEstat.ML
收藏

神经网络程序的强推广性和有效性

Strong Generalization and Efficiency in Neural Programs

DeepMind

Yujia Li, Felix Gimeno, Pushmeet Kohli, Oriol Vinyals

摘要:

我们研究的问题,学习有效的算法,强烈概括的神经程序归纳的框架。通过仔细设计神经模型的输入/输出接口并通过模仿,我们能够学习对任意输入大小产生正确结果的模型,从而实现强大的泛化能力。此外,通过使用强化学习,我们优化了程序的效率指标,并发现了新的算法,超越了模仿中使用的老师。有了这个,我们的方法可以学习超越各种问题的定制编写的解决方案,因为我们在排序,有序列表中搜索和NP完全0/1背包问题上测试了它,这在神经程序归纳领域树立了一个值得注意的里程碑。作为亮点,我们学习的模型可以在我们测试的任何输入数据大小上完美地执行排序,具有$O(n log n)$复杂性,同时在操作数量上优于手工编码的算法,包括快速排序,即使列表大小远远超过训练期间看到的。

英文摘要:

We study the problem of learning efficient algorithms that strongly generalize in the framework of neural program induction. By carefully designing the input / output interfaces of the neural model and through imitation, we are able to learn models that produce correct results for arbitrary input sizes, achieving strong generalization. Moreover, by using reinforcement learning, we optimize for program efficiency metrics, and discover new algorithms that surpass the teacher used in imitation. With this, our approach can learn to outperform custom-written solutions for a variety of problems, as we tested it on sorting, searching in ordered lists and the NP-complete 0/1 knapsack problem, which sets a notable milestone in the field of Neural Program Induction. As highlights, our learned model can perform sorting perfectly on any input data size we tested on, with $O(n log n)$ complexity, whilst outperforming hand-coded algorithms, including quick sort, in number of operations even for list sizes far beyond those seen during training.

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

↑