GOOD.ood_algorithms.algorithms

This module is for OOD algorithms. It is a good idea to inherit the BaseOOD class for new OOD algorithm designs.

Modules

GOOD.ood_algorithms.algorithms.BaseOOD

Base class for OOD algorithms

GOOD.ood_algorithms.algorithms.CIGA

Implementation of the CIGA algorithm from "Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphs" paper

GOOD.ood_algorithms.algorithms.Coral

Implementation of the Deep Coral algorithm from "Deep CORAL: Correlation Alignment for Deep Domain Adaptation" paper

GOOD.ood_algorithms.algorithms.DANN

Implementation of the DANN algorithm from "Domain-Adversarial Training of Neural Networks" paper

GOOD.ood_algorithms.algorithms.DIR

Implementation of the DIR algorithm from "Discovering Invariant Rationales for Graph Neural Networks" paper

GOOD.ood_algorithms.algorithms.EERM

Implementation of the EERM algorithm from "Handling Distribution Shifts on Graphs: An Invariance Perspective" paper

GOOD.ood_algorithms.algorithms.ERM

Implementation of the baseline ERM

GOOD.ood_algorithms.algorithms.GIL

Implementation of the GIL algorithm from "Learning Invariant Graph Representations for Out-of-Distribution Generalization" paper

GOOD.ood_algorithms.algorithms.GSAT

Implementation of the GSAT algorithm from "Interpretable and Generalizable Graph Learning via Stochastic Attention Mechanism" paper

GOOD.ood_algorithms.algorithms.GroupDRO

Implementation of the GroupDRO algorithm from "Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalization" paper

GOOD.ood_algorithms.algorithms.IRM

Implementation of the IRM algorithm from "Invariant Risk Minimization" paper

GOOD.ood_algorithms.algorithms.Mixup

Implementation of the Mixup algorithm from "Mixup for Node and Graph Classification" paper

GOOD.ood_algorithms.algorithms.SRGNN

Implementation of the SRGNN algorithm from "Shift-Robust GNNs: Overcoming the Limitations of Localized Graph Training Data" paper

GOOD.ood_algorithms.algorithms.VREx

Implementation of the VREx algorithm from "Out-of-Distribution Generalization via Risk Extrapolation (REx)" paper