arXiv reaDer
球状 CNN のスケーリング
Scaling Spherical CNNs
球面 CNN は、球面畳み込みを主な線形演算として使用することにより、CNN を球面上の関数に一般化します。球面畳み込みを計算する最も正確かつ効率的な方法はスペクトル領域での方法です (畳み込み定理による)。これでも通常の平面畳み込みよりもコストがかかります。このため、球面 CNN の応用はこれまで、低いモデル能力でアプローチできる小さな問題に限定されてきました。この研究では、より大きな問題に対して球状 CNN をどのように拡張できるかを示します。これを達成するために、共通モデル コンポーネントの新しいバリアント、ハードウェア アクセラレータの特性を活用するコア操作の実装、モデルのプロパティを活用するアプリケーション固有の入力表現など、重要な改善を行っています。実験では、当社のより大きな球状 CNN が、以前は等変グラフ ニューラル ネットワークによって支配されていた QM9 分子ベンチマークのいくつかのターゲットで最先端の水準に達し、複数の天気予報タスクで競争力のあるパフォーマンスを達成することが示されています。コードは https://github.com/google-research/spherical-cnn で入手できます。
Spherical CNNs generalize CNNs to functions on the sphere, by using spherical convolutions as the main linear operation. The most accurate and efficient way to compute spherical convolutions is in the spectral domain (via the convolution theorem), which is still costlier than the usual planar convolutions. For this reason, applications of spherical CNNs have so far been limited to small problems that can be approached with low model capacity. In this work, we show how spherical CNNs can be scaled for much larger problems. To achieve this, we make critical improvements including novel variants of common model components, an implementation of core operations to exploit hardware accelerator characteristics, and application-specific input representations that exploit the properties of our model. Experiments show our larger spherical CNNs reach state-of-the-art on several targets of the QM9 molecular benchmark, which was previously dominated by equivariant graph neural networks, and achieve competitive performance on multiple weather forecasting tasks. Our code is available at https://github.com/google-research/spherical-cnn.
updated: Thu Jun 08 2023 17:59:08 GMT+0000 (UTC)
published: Thu Jun 08 2023 17:59:08 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アソシエイト