用神经编译技术对比解析图神经网络的算法推理能力差距。
Mind The Gap: Quantifying Mechanistic Gaps in Algorithmic Reasoning via Neural Compilation
- 通过神经编译直接设置参数,实现算法精确计算
- 发现训练出的模型在路径搜索任务中准确率仅达68%
- 适合研究算法学习机制与模型可解释性的学者
本文旨在理解神经网络如何学习算法推理,聚焦两个问题:当模型表现有效时,其学习到的算法有多忠实?为何神经网络有时无法学会有效算法?为此,我们采用神经编译技术,将源算法直接编码为神经网络参数,使网络能精确执行算法。该方法允许我们比较编译与常规训练所得参数、中间向量及行为差异。研究以图神经网络(GNNs)为中心,选取广度优先搜索(BFS)、深度优先搜索(DFS)和贝尔曼-福特(Bellman-Ford)三种算法,覆盖有效、忠实与无效学习案例。传统方法依赖合成数据上的归纳学习,而本工作提出针对GNN的神经编译方法,通过解析设定参数跳过训练过程。选择GNN基于其与算法推理任务的自然契合性、丰富的归纳学习文献基础以及首次将神经编译应用于GNN的创新性。整体目标是刻画表达能力与可训练性之间的差距——这是算法推理学习中的根本性缺陷。我们假设归纳学习对属于 exttt{NC}类的并行算法最为有效。
原文摘要 · Abstract (English)
This paper aims to understand how neural networks learn algorithmic reasoning by addressing two questions: How faithful are learned algorithms when they are effective, and why do neural networks fail to learn effective algorithms otherwise? To answer these questions, we use neural compilation, a technique that directly encodes a source algorithm into neural network parameters, enabling the network to compute the algorithm exactly. This enables comparison between compiled and conventionally learned parameters, intermediate vectors, and behaviors. This investigation is crucial for developing neural networks that robustly learn complexalgorithms from data. Our analysis focuses on graph neural networks (GNNs), which are naturally aligned with algorithmic reasoning tasks, specifically our choices of BFS, DFS, and Bellman-Ford, which cover the spectrum of effective, faithful, and ineffective learned algorithms. Commonly, learning algorithmic reasoning is framed as induction over synthetic data, where a parameterized model is trained on inputs, traces, and outputs produced by an underlying ground truth algorithm. In contrast, we introduce a neural compilation method for GNNs, which sets network parameters analytically, bypassing training. Focusing on GNNs leverages their alignment with algorithmic reasoning, extensive algorithmic induction literature, and the novel application of neural compilation to GNNs. Overall, this paper aims to characterize expressability-trainability gaps - a fundamental shortcoming in learning algorithmic reasoning. We hypothesize that inductive learning is most effective for parallel algorithms contained within the computational class \texttt{NC}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。