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

图元网络用于处理多样化的神经架构

Graph Metanetworks for Processing Diverse Neural Architectures

MIT CSAIL(麻省理工学院计算机科学与人工智能实验室);Technion(以色列理工学院);NVIDIA(英伟达)

Derek Lim, Haggai Maron, Marc T. Law, Jonathan Lorraine, James Lucas

AI总结:

本文提出图元网络(GMNs),通过将神经网络参数构建为图并用图神经网络处理,以统一处理多种神经架构,具备表达力与置换等变性,并在多任务中验证有效性。

摘要:

神经网络在其参数内高效地编码所学信息。因此,许多任务可以通过将神经网络本身视为输入数据来统一。这样做时,最近的研究证明了考虑参数空间的对称性和几何性的重要性。然而,这些工作开发了针对特定网络(如没有归一化层的MLP和CNN)定制的架构,而将此类架构推广到其他类型的网络可能具有挑战性。在本工作中,我们通过构建新的元网络(即接收其他神经网络权重作为输入的神经网络)来克服这些挑战。简而言之,我们仔细构建表示输入神经网络的图,并使用图神经网络处理这些图。我们的方法,图元网络(GMNs),能够推广到竞争方法难以处理的神经架构,如多头注意力层、归一化层、卷积层、ResNet块和群等变线性层。我们证明了GMNs具有表达能力,并且对保持输入神经网络功能不变的参数置换对称性是等变的。我们在多个元网络任务上验证了我们方法的有效性,这些任务涉及多样化的神经网络架构。

英文摘要:

Neural networks efficiently encode learned information within their parameters. Consequently, many tasks can be unified by treating neural networks themselves as input data. When doing so, recent studies demonstrated the importance of accounting for the symmetries and geometry of parameter spaces. However, those works developed architectures tailored to specific networks such as MLPs and CNNs without normalization layers, and generalizing such architectures to other types of networks can be challenging. In this work, we overcome these challenges by building new metanetworks - neural networks that take weights from other neural networks as input. Put simply, we carefully build graphs representing the input neural networks and process the graphs using graph neural networks. Our approach, Graph Metanetworks (GMNs), generalizes to neural architectures where competing methods struggle, such as multi-head attention layers, normalization layers, convolutional layers, ResNet blocks, and group-equivariant linear layers. We prove that GMNs are expressive and equivariant to parameter permutation symmetries that leave the input neural network functions unchanged. We validate the effectiveness of our method on several metanetwork tasks over diverse neural network architectures.

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

↑