Dec, 2023

算法推理的三元边关注

TL;DR该研究通过神经算法推理对神经网络进行研究,旨在开发能够从经典算法中进行学习的神经网络。主要挑战是开发出表现足够强大的图神经网络,能够预测给定的算法输出,并在未知数据上具备良好的泛化能力。在此工作中,我们介绍了一种新的图神经网络层,称为 Triplet Edge Attention (TEA),它是一种边感知的图注意力层。我们通过精确计算边的潜在性能和使用基于边的注意力聚合多个三元组消息的方式来实现我们的算法。我们在 CLRS 基准测试中对 TEA 层进行了实证验证,并取得了平均 5% 的改进。特别是,在与现有模型相比,我们的方法在字符串算法中取得了 30% 的改进。