Code for "Invariance Principle Meets Information Bottleneck for Out-of-Distribution Generalization"

We place the three experiments (anti causal colored MNIST, covariate shift colored MNIST and invariance unit tests) and the respective instructions in three sub folders (AC-CMNIST, CS-CMNIST and InvarianceUnitTests).

Requirements

• torch 1.6.0
• torchvision 0.7.0
• numpy 1.19.1
• tqdm 4.41.1