Jun, 2024

基于动态(图)神经网络的规则学习

TL;DR通过将规则函数从知识中生成并使用这些规则定义基于规则的层,我们提出了一种解决经典神经网络体系结构中难以自然整合额外信息或专家知识进入学习过程的问题的两步方法。我们的研究重点是第二步,即基于规则的层,它们旨在根据输入样本动态地安排可学习参数在权重矩阵和偏差向量中。我们证明了我们的方法通过选择适当的规则来广义化传统前馈层,如全连接层和卷积层。我们提出了基于规则的图神经网络(RuleGNNs)作为一个具体的应用,克服了普通图神经网络的一些限制。我们的实验结果表明,RuleGNNs 的预测性能与使用基于 Weisfeiler-Leman 标签和模式计数的简单规则的最先进图分类器相当。此外,我们引入了新的合成基准图数据集,展示了如何将专家知识整合到 RuleGNNs 中,使它们比普通图神经网络更强大。