arXiv reaDer
グラフの自己教師あり学習:BT、HSIC、およびVICReg
Graph Self Supervised Learning: the BT, the HSIC, and the VICReg
自己教師あり学習および事前トレーニング戦略は、特に畳み込みニューラルネットワーク(CNN)について、過去数年にわたって開発されてきました。最近、そのような方法の適用は、グラフニューラルネットワーク(GNN)にも注目されています。この論文では、さまざまな損失関数(Barlow Twins [Zbontar et al。、2021]、HSIC [Tsai et al。、2021]、VICReg [Bardes et al。、2021])を使用したグラフベースの自己教師あり学習戦略を使用しました。 )以前にCNNを適用した場合に有望な結果を示しています。また、VICRegとHSICの利点を組み合わせたハイブリッド損失関数を提案し、VICRegHSICと呼びました。これらの前述のメソッドのパフォーマンスは、MUTAG、PROTEINS、IMDB-Binaryなどのさまざまなデータセットに適用された場合に比較されています。さらに、さまざまなバッチサイズ、プロジェクターの寸法、およびデータ拡張戦略の影響も調査されています。
Self-supervised learning and pre-training strategies have developed over the last few years especially for Convolutional Neural Networks (CNNs). Recently application of such methods can also be noticed for Graph Neural Networks (GNNs) . In this paper, we have used a graph based self-supervised learning strategy with different loss functions (Barlow Twins[Zbontar et al., 2021], HSIC[Tsai et al., 2021], VICReg[Bardes et al., 2021]) which have shown promising results when applied with CNNs previously. We have also proposed a hybrid loss function combining the advantages of VICReg and HSIC and called it as VICRegHSIC. The performance of these aforementioned methods have been compared when applied to different datasets such as MUTAG, PROTEINS and IMDB-Binary. Moreover, the impact of different batch sizes, projector dimensions and data augmentation strategies have also been explored
updated: Thu Jul 01 2021 03:18:43 GMT+0000 (UTC)
published: Tue May 25 2021 22:34:19 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アソシエイト