让神经网络在测试时搜索隐式程序,高效适应新任务。
Searching Latent Program Spaces
- 用可微分隐空间表示程序,测试时梯度搜索优化输出
- 在ARC-AGI上使分布外任务性能提升一倍
- 无需预定义语言,适合需要灵活适配的智能系统
通用智能需要高效获取新技能并超越训练分布进行泛化。尽管程序合成方法具有强泛化能力,但其庞大的组合搜索空间导致扩展困难,通常依赖人工设计的领域特定语言(DSL)或预训练先验来缩小范围。而深度学习方法虽有高成功率,却缺乏结构化测试时适应能力,且依赖大量随机采样或昂贵的梯度更新进行微调。本文提出隐式程序网络(Latent Program Network, LPN),将测试时搜索直接嵌入神经模型。LPN通过学习输入到输出的隐式程序潜空间,在测试时利用梯度进行搜索,结合了符号方法的适应性与神经方法的可扩展性。它在测试时对紧凑的潜空间进行搜索,无需预定义的领域特定语言。在一系列编程示例任务中,LPN表现优于或匹配现有上下文学习与测试时训练方法。在ARC-AGI基准测试中,我们验证了LPN既能学习紧凑程序空间,又能在测试时搜索以适应新任务;开启测试时搜索后,其在分布外任务上的性能提升一倍。
原文摘要 · Abstract (English)
General intelligence requires systems that acquire new skills efficiently and generalize beyond their training distributions. Although program synthesis approaches have strong generalization power, they face scaling issues due to the large combinatorial spaces that quickly render them impractical, requiring human-generated DSLs or pre-trained priors to narrow this search space. On the other hand, deep learning methods have had high successes, but they lack structured test-time adaptation and rely on heavy stochastic sampling or expensive gradient updates for fine-tuning. In this work, we propose the Latent Program Network (LPN), a novel architecture that builds in test-time search directly into neural models. LPN learns a latent space of implicit programs -- neurally mapping inputs to outputs -- through which it can search using gradients at test time. LPN combines the adaptability of symbolic approaches and the scalability of neural methods. It searches through a compact latent space at test time and bypasses the need for pre-defined domain-specific languages. On a range of programming-by-examples tasks, LPN either outperforms or matches performance compared to in-context learning and test-time training methods. Tested on the ARC-AGI benchmark, we demonstrate that LPN can both learn a compact program space and search through it at test time to adapt to novel tasks. LPN doubles its performance on out-of-distribution tasks when test-time search is switched on.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。