arXiv reaDer
バイナリグラフニューラルネットワーク
Binary Graph Neural Networks
グラフニューラルネットワーク(GNN)は、不規則なデータの表現学習のための強力で柔軟なフレームワークとして登場しました。グリッド上の従来のCNNの操作を任意のトポロジに一般化すると、GNNはユークリッド対応物の実装上の課題の多くももたらします。モデルサイズ、メモリフットプリント、およびエネルギー消費は、多くの実際のアプリケーションに共通の懸念事項です。ネットワークの2値化により、パラメーターとアクティベーションに1ビットが割り当てられるため、メモリ要件が大幅に削減され(単精度浮動小数点数と比較して最大32倍)、最新のハードウェアでの高速SIMD命令の利点が最大化され、測定可能な速度が向上します。ただし、古典的なCNNの二値化に関する多くの作業にもかかわらず、この領域は幾何学的な深層学習ではほとんど未踏のままです。この論文では、グラフニューラルネットワークの二値化のためのさまざまな戦略を提示し、評価します。モデルの注意深い設計とトレーニングプロセスの制御を通じて、バイナリグラフニューラルネットワークは、挑戦的なベンチマークで中程度の精度のコストでトレーニングできることを示します。特に、ハミング空間で最初の動的グラフニューラルネットワークを紹介します。これは、バイナリベクトルの効率的なk-NN検索を活用して、動的グラフの構築を高速化できます。さらに、バイナリモデルが組み込みデバイスを大幅に節約できることを確認します。私たちのコードはGithubで公開されています。
Graph Neural Networks (GNNs) have emerged as a powerful and flexible framework for representation learning on irregular data. As they generalize the operations of classical CNNs on grids to arbitrary topologies, GNNs also bring much of the implementation challenges of their Euclidean counterparts. Model size, memory footprint, and energy consumption are common concerns for many real-world applications. Network binarization allocates a single bit to parameters and activations, thus dramatically reducing the memory requirements (up to 32x compared to single-precision floating-point numbers) and maximizing the benefits of fast SIMD instructions on modern hardware for measurable speedups. However, in spite of the large body of work on binarization for classical CNNs, this area remains largely unexplored in geometric deep learning. In this paper, we present and evaluate different strategies for the binarization of graph neural networks. We show that through careful design of the models, and control of the training process, binary graph neural networks can be trained at only a moderate cost in accuracy on challenging benchmarks. In particular, we present the first dynamic graph neural network in Hamming space, able to leverage efficient k-NN search on binary vectors to speed-up the construction of the dynamic graph. We further verify that the binary models offer significant savings on embedded devices. Our code is publicly available on Github.
updated: Mon Mar 29 2021 23:48:56 GMT+0000 (UTC)
published: Thu Dec 31 2020 18:48:58 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アソシエイト