arXiv reaDer
バイナリグラフニューラルネットワーク
Binary Graph Neural Networks
グラフニューラルネットワーク(GNN)は、不規則なデータの表現学習のための強力で柔軟なフレームワークとして登場しました。グリッド上の従来のCNNの操作を任意のトポロジに一般化すると、GNNはユークリッド対応の実装の課題の多くももたらします。モデルサイズ、メモリフットプリント、およびエネルギー消費は、多くの実際のアプリケーションに共通の懸念事項です。ネットワークの2値化により、ネットワークパラメータとアクティベーションに1ビットが割り当てられるため、メモリ要件が大幅に削減され(単精度浮動小数点パラメータと比較して最大32倍)、最新のハードウェアの高速SIMD命令の利点が最大化され、測定可能な速度が向上します。ただし、古典的なCNNの二値化に関する大量の作業にもかかわらず、この領域は幾何学的な深層学習ではほとんど未踏のままです。この論文では、グラフニューラルネットワークの二値化のためのさまざまな戦略を提示し、評価します。モデルの注意深い設計とトレーニングプロセスの制御を通じて、バイナリグラフニューラルネットワークは、挑戦的なベンチマークで中程度の精度のコストでトレーニングできることを示します。特に、ハミング空間で最初の動的グラフニューラルネットワークを紹介します。これは、バイナリベクトルの効率的なk-NN検索を活用して、動的グラフの構築を高速化できます。さらに、バイナリモデルが組み込みデバイスを大幅に節約できることを確認します。
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 network parameters and activations, thus dramatically reducing the memory requirements (up to 32x compared to single-precision floating-point parameters) and maximizing the benefits of fast SIMD instructions of 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.
updated: Thu Dec 31 2020 18:48:58 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アソシエイト