arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.05441cs.LG
收藏

图的分布外泛化的因果不变表示学习

Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphs

The Chinese University of Hong Kong(香港中文大学);Hong Kong Baptist University(香港浸会大学);Tencent AI Lab(腾讯人工智能实验室);The University of Sydney(悉尼大学)

Yongqiang Chen, Yonggang Zhang, Yatao Bian, Han Yang, Kaili Ma, Binghui Xie, Tongliang Liu, Bo Han, James Cheng

摘要:

尽管最近成功地将不变性原理用于欧几里德数据的分布外(OOD)推广(例如,图像),对图形数据的研究仍然有限。与图像不同,图形的复杂性对采用不变性原理提出了独特的挑战。特别是,图上的分布偏移可以以属性和结构等多种形式出现,使得难以识别不变性。此外,域或环境分区,这往往是需要的OOD方法对欧几里德数据,可能是非常昂贵的获得图。为了弥补这一差距,我们提出了一个新的框架,称为因果启发不变图学习(CIGA),捕捉各种分布偏移下的保证OOD推广的图的不变性。具体来说,我们的特点潜在的分布偏移图与因果模型,得出结论,OOD泛化图是可以实现的,当模型只集中在子图包含最多的信息标签的原因。因此,我们提出了一个信息理论的目标,以提取所需的子图,最大限度地保持不变的类内信息。使用这些子图进行学习不受分布偏移的影响。在16个合成或真实世界的数据集上进行了广泛的实验,包括一个具有挑战性的设置-来自AI辅助药物发现的DrugOOD,验证了CIGA的卓越OOD性能。

英文摘要:

Despite recent success in using the invariance principle for out-of-distribution (OOD) generalization on Euclidean data (e.g., images), studies on graph data are still limited. Different from images, the complex nature of graphs poses unique challenges to adopting the invariance principle. In particular, distribution shifts on graphs can appear in a variety of forms such as attributes and structures, making it difficult to identify the invariance. Moreover, domain or environment partitions, which are often required by OOD methods on Euclidean data, could be highly expensive to obtain for graphs. To bridge this gap, we propose a new framework, called Causality Inspired Invariant Graph LeArning (CIGA), to capture the invariance of graphs for guaranteed OOD generalization under various distribution shifts. Specifically, we characterize potential distribution shifts on graphs with causal models, concluding that OOD generalization on graphs is achievable when models focus only on subgraphs containing the most information about the causes of labels. Accordingly, we propose an information-theoretic objective to extract the desired subgraphs that maximally preserve the invariant intra-class information. Learning with these subgraphs is immune to distribution shifts. Extensive experiments on 16 synthetic or real-world datasets, including a challenging setting -- DrugOOD, from AI-aided drug discovery, validate the superior OOD performance of CIGA.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑