arXiv reaDer
グラフの自己教師あり学習:BT、HSIC、およびVICReg
Graph Self Supervised Learning: the BT, the HSIC, and the VICReg
自己教師あり学習および事前トレーニング戦略は、特に畳み込みニューラルネットワーク(CNN)について、過去数年にわたって開発されてきました。最近、そのような方法の適用は、グラフニューラルネットワーク(GNN)にも注目されています。この論文では、以前にCNNを適用した場合に有望な結果を示した、さまざまな損失関数(Barlow Twins [7]、HSIC [4]、VICReg [1])を使用したグラフベースの自己教師あり学習戦略を使用しました。また、VICRegとHSICの利点を組み合わせたハイブリッド損失関数を提案し、VICRegHSICと呼びました。これらの前述の方法のパフォーマンスは、MUTAGとPROTEINSという2つの異なるデータセットに適用された場合に比較されています。さらに、さまざまなバッチサイズ、プロジェクターの寸法、およびデータ拡張戦略の影響も調査されています。結果は暫定的なものであり、他のデータセットで引き続き調査します。
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[ 7], HSIC[ 4], VICReg[ 1]) 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 two different datasets namely MUTAG and PROTEINS. Moreover, the impact of different batch sizes, projector dimensions and data augmentation strategies have also been explored. The results are preliminary and we will be continuing to explore with other datasets.
updated: Sat May 29 2021 04:51:26 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アソシエイト