arXiv reaDer
配布外の一般化における個別および構造グラフ情報のボトルネック
Individual and Structural Graph Information Bottlenecks for Out-of-Distribution Generalization
分布外 (OOD) グラフの一般化は、現実世界の多くのアプリケーションにとって重要です。既存の方法では、ラベルとは無関係な入力のスプリアスまたはノイズの多い特徴を破棄することが無視されています。さらに、それらは主にインスタンスレベルのクラス不変グラフ学習を実行し、グラフインスタンス間の構造的なクラス関係を利用できません。この研究では、個別および構造グラフ情報ボトルネック (IS-GIB) と呼ばれる統一フレームワークでこれらの問題に対処するよう努めています。分布シフトによって引き起こされるクラスのスプリアス特徴を除去するために、入力グラフとその埋め込み間の相互情報を最小化することによって無関係な情報を破棄する個別グラフ情報ボトルネック (I-GIB) を提案します。ドメイン内およびドメイン間の構造的な相関関係を活用するために、構造グラフ情報ボトルネック (S-GIB) を提案します。特に複数のドメインを含むグラフのバッチの場合、S-GIB はまず、ペアごとの入力間相関、埋め込み間埋め込み、およびラベル間の相関を計算します。次に、入力グラフと埋め込みペアの間の相互情報を最小化し、埋め込みペアとラベル ペアの間の相互情報を最大化します。 S-GIB の重要な洞察は、複数の分布シフトの下でクラス関係を維持することによって、偽の特徴を破棄し、不変の特徴を高次の観点から同時に学習することです。特に、提案されている I-GIB と S-GIB を統合して、補完的なフレームワーク IS-GIB を形成します。ノードレベルとグラフレベルのタスクの両方で行われた広範な実験により、IS-GIB の優れた一般化能力が一貫して実証されています。コードは https://github.com/YangLing0818/GraphOOD で入手できます。
Out-of-distribution (OOD) graph generalization are critical for many real-world applications. Existing methods neglect to discard spurious or noisy features of inputs, which are irrelevant to the label. Besides, they mainly conduct instance-level class-invariant graph learning and fail to utilize the structural class relationships between graph instances. In this work, we endeavor to address these issues in a unified framework, dubbed Individual and Structural Graph Information Bottlenecks (IS-GIB). To remove class spurious feature caused by distribution shifts, we propose Individual Graph Information Bottleneck (I-GIB) which discards irrelevant information by minimizing the mutual information between the input graph and its embeddings. To leverage the structural intra- and inter-domain correlations, we propose Structural Graph Information Bottleneck (S-GIB). Specifically for a batch of graphs with multiple domains, S-GIB first computes the pair-wise input-input, embedding-embedding, and label-label correlations. Then it minimizes the mutual information between input graph and embedding pairs while maximizing the mutual information between embedding and label pairs. The critical insight of S-GIB is to simultaneously discard spurious features and learn invariant features from a high-order perspective by maintaining class relationships under multiple distributional shifts. Notably, we unify the proposed I-GIB and S-GIB to form our complementary framework IS-GIB. Extensive experiments conducted on both node- and graph-level tasks consistently demonstrate the superior generalization ability of IS-GIB. The code is available at https://github.com/YangLing0818/GraphOOD.
updated: Wed Jun 28 2023 03:52:41 GMT+0000 (UTC)
published: Wed Jun 28 2023 03:52:41 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アソシエイト