arXiv reaDer
エッジ機能グラフニューラルアーキテクチャ検索
Edge-featured Graph Neural Architecture Search
グラフニューラルネットワーク(GNN)は、多くのリレーショナルタスクでグラフの表現を学習するためにうまく適用されています。最近、研究者は人間の専門知識への依存を減らし、より良いGNNアーキテクチャを探索するためにニューラルアーキテクチャ検索(NAS)を研究していますが、エンティティの機能を強調しすぎて、エッジに隠された潜在的な関係情報を無視しています。この問題を解決するために、エッジ機能をグラフ検索スペースに組み込み、エッジ機能を備えたグラフニューラルアーキテクチャ検索を提案して、最適なGNNアーキテクチャを見つけます。具体的には、より一般的なメッセージパッシングメカニズムを伝達する高次の表現を学習するために、豊富なエンティティとエッジの更新操作を設計します。さらに、検索スペースのアーキテクチャトポロジでは、エンティティとエッジの両方の複雑な機能依存性を調査できます。これは、微分可能な検索戦略によって効率的に最適化できます。 6つのデータセットに対する3つのグラフタスクでの実験は、EGNASが現在の最先端の人間が設計および検索ベースのGNNよりも高いパフォーマンスでより優れたGNNを検索できることを示しています。
Graph neural networks (GNNs) have been successfully applied to learning representation on graphs in many relational tasks. Recently, researchers study neural architecture search (NAS) to reduce the dependence of human expertise and explore better GNN architectures, but they over-emphasize entity features and ignore latent relation information concealed in the edges. To solve this problem, we incorporate edge features into graph search space and propose Edge-featured Graph Neural Architecture Search to find the optimal GNN architecture. Specifically, we design rich entity and edge updating operations to learn high-order representations, which convey more generic message passing mechanisms. Moreover, the architecture topology in our search space allows to explore complex feature dependence of both entities and edges, which can be efficiently optimized by differentiable search strategy. Experiments at three graph tasks on six datasets show EGNAS can search better GNNs with higher performance than current state-of-the-art human-designed and searched-based GNNs.
updated: Fri Sep 03 2021 07:53:18 GMT+0000 (UTC)
published: Fri Sep 03 2021 07:53:18 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト