arXiv reaDer
ContraNorm: オーバースムージングとその先に関する対照的な学習の視点
ContraNorm: A Contrastive Learning Perspective on Oversmoothing and Beyond
オーバースムージングは、さまざまなグラフ ニューラル ネットワーク (GNN) およびトランスフォーマーで一般的な現象であり、レイヤーの数が増えるとパフォーマンスが低下します。表現が一点に収束する完全な崩壊の観点から過度の平滑化を特徴付ける代わりに、表現が狭い円錐にある次元の崩壊のより一般的な視点に飛び込みます。したがって、次元崩壊の防止における対照学習の有効性に触発されて、ContraNorm と呼ばれる新しい正規化レイヤーを提案します。直観的には、ContraNorm は埋め込み空間内の表現を暗黙のうちに粉砕し、より均一な分布とわずかな次元の崩壊につながります。理論的な分析では、ContraNorm が特定の条件下で完全な崩壊と次元の崩壊の両方を軽減できることを証明します。提案された正規化レイヤーは、ごくわずかなパラメーター オーバーヘッドで GNN とトランスフォーマーに簡単に統合できます。さまざまな実世界のデータセットでの実験は、提案された ContraNorm の有効性を示しています。私たちの実装は、https://github.com/PKU-ML/ContraNorm で入手できます。
Oversmoothing is a common phenomenon in a wide range of Graph Neural Networks (GNNs) and Transformers, where performance worsens as the number of layers increases. Instead of characterizing oversmoothing from the view of complete collapse in which representations converge to a single point, we dive into a more general perspective of dimensional collapse in which representations lie in a narrow cone. Accordingly, inspired by the effectiveness of contrastive learning in preventing dimensional collapse, we propose a novel normalization layer called ContraNorm. Intuitively, ContraNorm implicitly shatters representations in the embedding space, leading to a more uniform distribution and a slighter dimensional collapse. On the theoretical analysis, we prove that ContraNorm can alleviate both complete collapse and dimensional collapse under certain conditions. Our proposed normalization layer can be easily integrated into GNNs and Transformers with negligible parameter overhead. Experiments on various real-world datasets demonstrate the effectiveness of our proposed ContraNorm. Our implementation is available at https://github.com/PKU-ML/ContraNorm.
updated: Sun Mar 12 2023 04:04:51 GMT+0000 (UTC)
published: Sun Mar 12 2023 04:04:51 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アソシエイト