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.
Università della Svizzera Italiana(瑞士意大利大学);University of Liechtenstein(利赫泰森大学);Politecnico di Torino(托尼诺理工学院);IBM Research(IBM研究院);Scuola Universitaria Professionale della Svizzera Italiana(瑞士意大利专业大学)
Out-of-distribution Generalization for Total Variation based Invariant Risk Minimization
College of Cyber Security, Jinan University(暨南大学网络安全学院);Pratt School of Engineering, Duke University(杜克大学普拉特工程学院);Sino-French Engineer School, Beihang University(北京航空航天大学中法工程师学院)
Counterfactual Supervision-based Information Bottleneck for Out-of-Distribution Generalization
South China University of Technology(华南理工大学)
学习不变(因果)特征的分布外(OOD)推广最近引起了广泛的关注,其中不变风险最小化(VRM)是一个值得注意的解决方案。尽管线性回归理论上有希望,但在线性分类问题中使用递归的挑战仍然存在。通过将信息瓶颈(IB)原理引入到机器学习中,IB-SVM方法已经证明了它解决这些挑战的能力。本文从两个方面进一步改进了IB-SVM。首先,我们证明了在IB-EQUIPMENT中使用的不变特征的支持重叠的关键假设对于保证OOD泛化是强有力的,并且在没有这个假设的情况下仍然有可能实现最优解。其次,我们说明了两种故障模式,IB-ESTA(和ESTA)可能会失败的学习不变的功能,并解决这样的故障,我们提出了一个\textit{Counterfactual Supervision-based Information Bottleneck (CSIB)}学习算法,可证明恢复不变的功能。通过要求反事实推理,CSIB即使在从单个环境访问数据时也能工作。在多个数据集上的实验验证了我们的理论结果。
Adversarial Label Invariant Graph Data Augmentations for Out-of-Distribution Generalization
Department of Computer Science, Purdue University(普渡大学计算机科学系);Department of Computer Science and Engineering, The Ohio State University(俄亥俄州立大学计算机科学与工程系);Department of Industrial and Systems Engineering, The Ohio State University(俄亥俄州立大学工业与系统工程系)
Winning Prize Comes from Losing Tickets: Improve Invariant Learning by Exploring Variant Parameters for Out-of-Distribution Generalization
Sydney AI Centre, The University of Sydney(悉尼大学悉尼人工智能中心);JD Explore Academy(京东探索研究院);University of Science and Technology of China(中国科学技术大学);Nanjing University of Science and Technology(南京理工大学);Hong Kong Baptist University(香港浸会大学)
Raising the Bar in Graph OOD Generalization: Invariant Learning Beyond Explicit Environment Modeling
The School of Artificial Intelligence, Jilin University(吉林大学人工智能学院);The School of Information and Communication Technology, Griffith University(格里菲斯大学信息与通信技术学院)
Out-of-distribution Generalization with Causal Invariant Transformations
University of Chinese Academy of Sciences(中国科学院大学);Academy of Mathematics and Systems Science, Chinese Academy of Sciences(中国科学院数学与系统科学研究院);Huawei Noah’s Ark Lab(华为诺亚方舟实验室)
Invariant Graph Learning Meets Information Bottleneck for Out-of-Distribution Generalization
School of Cyber Science and Technology, University of Science and Technology of China(中国科学技术大学网络空间安全学院);Tencent(腾讯);School of Artificial Intelligence and Data Science, University of Science and Technology of China(中国科学技术大学人工智能与数据科学学院)
Learning to Learn Domain-invariant Parameters for Domain Generalization
Institute of Computing Technology, Chinese Academy of Sciences(中国科学院计算技术研究所);University of Chinese Academy of Sciences(中国科学院大学);Southeast University(东南大学);Lenovo Research(联想研究院);Lenovo Ltd(联想有限公司)
Improved Generalization Bounds of Group Invariant / Equivariant Deep Networks via Quotient Feature Spaces
RIKEN Center for Advanced Intelligence Project(理化学研究所先进智能项目中心);Keio University(庆应义塾大学);The University of Tokyo(东京大学)
许多不变(或等变)神经网络已经成功地处理不变的数据,如点云和图形。然而,神经网络的推广理论还没有得到很好的发展,因为该理论的几个基本因素,如网络大小和边缘分布,没有深入联系的不变性和等变性。在这项研究中,我们开发了一个新的泛化误差界不变和等变深度神经网络。为了描述不变性和等变性对推广的影响,我们提出了一个\textit{quotient feature space}的概念,它度量了群作用对性质的影响。我们的主要结果证明了商特征空间的体积可以描述泛化误差。此外,该界表明,不变性和等变性显着改善的领先项的界。我们将我们的结果应用于特定的不变和等变网络,例如DeepSets(Zaheer et al.(2017)),并表明它们的泛化界被$\sqrt{n!}$大大提高,其中$n!$是排列的数量。我们还讨论了不变DNN的表达能力,并表明它们可以实现最佳近似率。我们的实验结果支持了我们的理论主张。
Invariant Information Bottleneck for Domain Generalization
Microsoft Research Asia(微软亚洲研究院);Hong Kong University of Science and Technology(香港科技大学);Washington University in St. Louis(圣路易斯华盛顿大学);University of California, Berkeley(加州大学伯克利分校);University of Illinois at Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校)
Domain-invariant Feature Exploration for Domain Generalization
Institute of Computing Technology, Chinese Academy of Sciences(中国科学院计算技术研究所);Microsoft Research Asia(微软亚洲研究院);City University of Hong Kong(香港城市大学);Peng Cheng Laboratory(鹏城实验室)
Class-Invariant Test-Time Augmentation for Domain Generalization
School of Computing and Artificial Intelligence(计算机与人工智能学院);ANGEL Lab(ANGEL实验室);Southwest Jiaotong University(西南交通大学);Nanyang Technological University(南洋理工大学)
Domain Generalization via Conditional Invariant Representation
University of Science and Technology of China(中国科学技术大学);The University of Sydney(悉尼大学);Carnegie Mellon University(卡内基梅隆大学);University of Pittsburgh(匹兹堡大学)