generalization topic
pomdp-baselines
Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022
self-adaptive-training
[TPAMI2022 & NeurIPS2020] Official implementation of Self-Adaptive Training
pgdrive
PGDrive: an open-ended driving simulator with infinite scenes from procedural generation
Sharpness-Aware-Minimization-TensorFlow
Implements sharpness-aware minimization (https://arxiv.org/abs/2010.01412) in TensorFlow 2.
Recurrent-Independent-Mechanisms
Implementation of the paper Recurrent Independent Mechanisms (https://arxiv.org/pdf/1909.10893.pdf)
DropAttack
Code of paper: "DropAttack: A Masked Weight Adversarial Training Method to Improve Generalization of Neural Networks"
transformer_generalization
The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We significantly improve the systematic generalization of transform...
3DShapeGen
Code for 3D Reconstruction of Novel Object Shapes from Single Images paper
DIR-GNN
Official code of "Discovering Invariant Rationales for Graph Neural Networks" (ICLR 2022)
SLADD
Official code for Self-supervised Learning of Adversarial Example: Towards Good Generalizations for Deepfake Detection (CVPR 2022 oral)