グラフニューラルネットワーク(GNN)は、グラフを学習するための柔軟で強力なアプローチとして登場しました。この成功にもかかわらず、既存のGNNは、ローカルのメッセージパッシングアーキテクチャによって制約されており、表現力が制限されていることが証明されています。この作業では、新しいGNNアーキテクチャであるニューラルツリーを提案します。ニューラルツリーアーキテクチャは、入力グラフではなく、入力グラフから構築されたHツリーと呼ばれるツリー構造のグラフでメッセージパッシングを実行します。 Hツリーのノードは入力グラフのサブグラフに対応し、Hツリーのノードの親ノードが常に入力グラフのより大きなサブグラフに対応するように階層的に再編成されます。ニューラルツリーアーキテクチャが、無向グラフ上で任意の滑らかな確率分布関数を近似できること、およびジャンクションツリーアルゴリズムをエミュレートできることを示します。また、分布関数のϵ近似を達成するために必要なパラメーターの数は、入力グラフのツリー幅では指数関数的ですが、サイズは線形であることを証明します。ニューラルツリーを3Dシーングラフの半教師ありノード分類に適用し、これらの理論的特性が、従来のGNNアーキテクチャよりも予測精度の大幅な向上につながることを示します。
Graph Neural Networks (GNNs) have emerged as a flexible and powerful approach for learning over graphs. Despite this success, existing GNNs are constrained by their local message-passing architecture and are provably limited in their expressive power. In this work, we propose a new GNN architecture -- the Neural Tree. The neural tree architecture does not perform message passing on the input graph but on a tree-structured graph, called the H-tree, that is constructed from the input graph. Nodes in the H-tree correspond to subgraphs in the input graph, and they are reorganized in a hierarchical manner such that a parent-node of a node in the H-tree always corresponds to a larger subgraph in the input graph. We show that the neural tree architecture can approximate any smooth probability distribution function over an undirected graph, as well as emulate the junction tree algorithm. We also prove that the number of parameters needed to achieve an ϵ-approximation of the distribution function is exponential in the treewidth of the input graph, but linear in its size. We apply the neural tree to semi-supervised node classification in 3D scene graphs, and show that these theoretical properties translate into significant gains in prediction accuracy, over the more traditional GNN architectures.