arXiv reaDer
ポアンカレ ResNet
Poincaré ResNet
この論文では、双曲空間のポアンカレ球モデルで完全に動作するエンドツーエンドの残差ネットワークを紹介します。双曲線学習は最近、視覚的な理解の大きな可能性を示していますが、現在のところ、ディープ ネットワークの最後から 2 番目のレイヤーでのみ実行されます。すべての視覚的表現は、標準のユークリッド ネットワークを通じて引き続き学習されます。この論文では、視覚データの双曲線表現をピクセルレベルから直接学習する方法を調査します。ポアンカレ 2D 畳み込みからポアンカレ残差接続まで、有名な残差ネットワークの双曲線版であるポアンカレ ResNet を提案します。畳み込みネットワークを完全に双曲線空間でトレーニングするための 3 つの障害を特定し、それぞれの解決策を提案します。多くのレイヤーで規範を維持するアイデンティティベースの初期化を提供します。 (ii) 残差ネットワークは、双曲線空間での高価なフレシェ平均計算を伴うバッチ正規化に大きく依存しています。ポアンカレ中点バッチ正規化を、より高速で同等に効果的な代替手段として紹介します。 (iii) ポアンカレ層の多くの中間演算により、深層学習ライブラリの計算グラフが爆発し、深層双曲線ネットワークでトレーニングする能力が制限されることが最終的にわかりました。管理しやすい計算グラフを維持するために、核となる双曲線演算の後方導出を手動で提供します。
This paper introduces an end-to-end residual network that operates entirely on the Poincaré ball model of hyperbolic space. Hyperbolic learning has recently shown great potential for visual understanding, but is currently only performed in the penultimate layer(s) of deep networks. All visual representations are still learned through standard Euclidean networks. In this paper we investigate how to learn hyperbolic representations of visual data directly from the pixel-level. We propose Poincaré ResNet, a hyperbolic counterpart of the celebrated residual network, starting from Poincaré 2D convolutions up to Poincaré residual connections. We identify three roadblocks for training convolutional networks entirely in hyperbolic space and propose a solution for each: (i) Current hyperbolic network initializations collapse to the origin, limiting their applicability in deeper networks. We provide an identity-based initialization that preserves norms over many layers. (ii) Residual networks rely heavily on batch normalization, which comes with expensive Fréchet mean calculations in hyperbolic space. We introduce Poincaré midpoint batch normalization as a faster and equally effective alternative. (iii) Due to the many intermediate operations in Poincaré layers, we lastly find that the computation graphs of deep learning libraries blow up, limiting our ability to train on deep hyperbolic networks. We provide manual backward derivations of core hyperbolic operations to maintain manageable computation graphs.
updated: Tue Dec 19 2023 15:15:50 GMT+0000 (UTC)
published: Fri Mar 24 2023 14:37:07 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アソシエイト