├── InvariantRiskMinimization ├── LICENSE ├── colored_mnist │ ├── README.md │ ├── REx-HD.py │ ├── covariate_shifts │ │ ├── README.md │ │ ├── class_imbalance.py │ │ ├── color_imbalance.py │ │ └── digit_imbalance.py │ ├── main.py │ ├── main_waterfall_experiment.py │ └── make_Figure2.py └── experiment_synthetic │ ├── README.md │ ├── models.py │ ├── plot.py │ ├── run_all.sh │ ├── run_toy_SEMs.py │ ├── sem.py │ └── synthetic_results.pt ├── README.md ├── discrete_example.py ├── domain_generalization ├── LICENSE ├── README.md ├── READ_BEFORE_RUNNING_EXPERIMENT.txt ├── models │ ├── __init__.py │ ├── alexnet.py │ ├── caffenet.py │ ├── mnist.py │ ├── model_factory.py │ ├── model_utils.py │ ├── patch_based.py │ ├── resnet.py │ └── resnet_gap.py ├── optimizer │ ├── __init__.py │ └── optimizer_helper.py ├── run_rex_pacs.sh ├── test │ ├── __init__.py │ └── test_alexNetCaffe.py ├── train_DA_jigsaw.py ├── train_jigsaw.py ├── train_jigsaw_evaljig.py └── utils │ ├── Logger.py │ ├── __init__.py │ ├── tf_logger.py │ └── vis.py ├── finance_experiments ├── README.md ├── data │ ├── 2014_clean.csv │ ├── 2015_clean.csv │ ├── 2016_clean.csv │ ├── 2017_clean.csv │ └── 2018_clean.csv ├── load.py ├── main.py ├── plot_grid.py ├── run-exp.sh └── utils.py └── reinforcement_learning_with_spurious_correlations ├── README.md ├── agent ├── __init__.py ├── actor.py ├── critic.py └── sac.py ├── conda_env.yml ├── config ├── agent │ ├── rrex.yaml │ └── sac.yaml └── train.yaml ├── decoder.py ├── dmc2gym ├── __init__.py └── wrappers.py ├── encoder.py ├── graph_utils.py ├── logger.py ├── replay_buffer.py ├── run_all.sh ├── run_cluster.sh ├── run_local.sh ├── runs.ipynb ├── train.py ├── utils.py └── video.py /InvariantRiskMinimization/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/LICENSE -------------------------------------------------------------------------------- /InvariantRiskMinimization/colored_mnist/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/colored_mnist/README.md -------------------------------------------------------------------------------- /InvariantRiskMinimization/colored_mnist/REx-HD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/colored_mnist/REx-HD.py -------------------------------------------------------------------------------- /InvariantRiskMinimization/colored_mnist/covariate_shifts/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/colored_mnist/covariate_shifts/README.md -------------------------------------------------------------------------------- /InvariantRiskMinimization/colored_mnist/covariate_shifts/class_imbalance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/colored_mnist/covariate_shifts/class_imbalance.py -------------------------------------------------------------------------------- /InvariantRiskMinimization/colored_mnist/covariate_shifts/color_imbalance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/colored_mnist/covariate_shifts/color_imbalance.py -------------------------------------------------------------------------------- /InvariantRiskMinimization/colored_mnist/covariate_shifts/digit_imbalance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/colored_mnist/covariate_shifts/digit_imbalance.py -------------------------------------------------------------------------------- /InvariantRiskMinimization/colored_mnist/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/colored_mnist/main.py -------------------------------------------------------------------------------- /InvariantRiskMinimization/colored_mnist/main_waterfall_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/colored_mnist/main_waterfall_experiment.py -------------------------------------------------------------------------------- /InvariantRiskMinimization/colored_mnist/make_Figure2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/colored_mnist/make_Figure2.py -------------------------------------------------------------------------------- /InvariantRiskMinimization/experiment_synthetic/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/experiment_synthetic/README.md -------------------------------------------------------------------------------- /InvariantRiskMinimization/experiment_synthetic/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/experiment_synthetic/models.py -------------------------------------------------------------------------------- /InvariantRiskMinimization/experiment_synthetic/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/experiment_synthetic/plot.py -------------------------------------------------------------------------------- /InvariantRiskMinimization/experiment_synthetic/run_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/experiment_synthetic/run_all.sh -------------------------------------------------------------------------------- /InvariantRiskMinimization/experiment_synthetic/run_toy_SEMs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/experiment_synthetic/run_toy_SEMs.py -------------------------------------------------------------------------------- /InvariantRiskMinimization/experiment_synthetic/sem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/experiment_synthetic/sem.py -------------------------------------------------------------------------------- /InvariantRiskMinimization/experiment_synthetic/synthetic_results.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/InvariantRiskMinimization/experiment_synthetic/synthetic_results.pt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/README.md -------------------------------------------------------------------------------- /discrete_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/discrete_example.py -------------------------------------------------------------------------------- /domain_generalization/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/LICENSE -------------------------------------------------------------------------------- /domain_generalization/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/README.md -------------------------------------------------------------------------------- /domain_generalization/READ_BEFORE_RUNNING_EXPERIMENT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/READ_BEFORE_RUNNING_EXPERIMENT.txt -------------------------------------------------------------------------------- /domain_generalization/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /domain_generalization/models/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/models/alexnet.py -------------------------------------------------------------------------------- /domain_generalization/models/caffenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/models/caffenet.py -------------------------------------------------------------------------------- /domain_generalization/models/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/models/mnist.py -------------------------------------------------------------------------------- /domain_generalization/models/model_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/models/model_factory.py -------------------------------------------------------------------------------- /domain_generalization/models/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/models/model_utils.py -------------------------------------------------------------------------------- /domain_generalization/models/patch_based.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/models/patch_based.py -------------------------------------------------------------------------------- /domain_generalization/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/models/resnet.py -------------------------------------------------------------------------------- /domain_generalization/models/resnet_gap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/models/resnet_gap.py -------------------------------------------------------------------------------- /domain_generalization/optimizer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /domain_generalization/optimizer/optimizer_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/optimizer/optimizer_helper.py -------------------------------------------------------------------------------- /domain_generalization/run_rex_pacs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/run_rex_pacs.sh -------------------------------------------------------------------------------- /domain_generalization/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /domain_generalization/test/test_alexNetCaffe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/test/test_alexNetCaffe.py -------------------------------------------------------------------------------- /domain_generalization/train_DA_jigsaw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/train_DA_jigsaw.py -------------------------------------------------------------------------------- /domain_generalization/train_jigsaw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/train_jigsaw.py -------------------------------------------------------------------------------- /domain_generalization/train_jigsaw_evaljig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/train_jigsaw_evaljig.py -------------------------------------------------------------------------------- /domain_generalization/utils/Logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/utils/Logger.py -------------------------------------------------------------------------------- /domain_generalization/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /domain_generalization/utils/tf_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/utils/tf_logger.py -------------------------------------------------------------------------------- /domain_generalization/utils/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/domain_generalization/utils/vis.py -------------------------------------------------------------------------------- /finance_experiments/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/finance_experiments/README.md -------------------------------------------------------------------------------- /finance_experiments/data/2014_clean.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/finance_experiments/data/2014_clean.csv -------------------------------------------------------------------------------- /finance_experiments/data/2015_clean.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/finance_experiments/data/2015_clean.csv -------------------------------------------------------------------------------- /finance_experiments/data/2016_clean.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/finance_experiments/data/2016_clean.csv -------------------------------------------------------------------------------- /finance_experiments/data/2017_clean.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/finance_experiments/data/2017_clean.csv -------------------------------------------------------------------------------- /finance_experiments/data/2018_clean.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/finance_experiments/data/2018_clean.csv -------------------------------------------------------------------------------- /finance_experiments/load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/finance_experiments/load.py -------------------------------------------------------------------------------- /finance_experiments/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/finance_experiments/main.py -------------------------------------------------------------------------------- /finance_experiments/plot_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/finance_experiments/plot_grid.py -------------------------------------------------------------------------------- /finance_experiments/run-exp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/finance_experiments/run-exp.sh -------------------------------------------------------------------------------- /finance_experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/finance_experiments/utils.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/README.md -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/agent/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/agent/__init__.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/agent/actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/agent/actor.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/agent/critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/agent/critic.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/agent/sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/agent/sac.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/conda_env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/conda_env.yml -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/config/agent/rrex.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/config/agent/rrex.yaml -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/config/agent/sac.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/config/agent/sac.yaml -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/config/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/config/train.yaml -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/decoder.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/dmc2gym/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/dmc2gym/__init__.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/dmc2gym/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/dmc2gym/wrappers.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/encoder.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/graph_utils.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/logger.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/replay_buffer.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/run_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/run_all.sh -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/run_cluster.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/run_cluster.sh -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/run_local.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/run_local.sh -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/runs.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/runs.ipynb -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/train.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/utils.py -------------------------------------------------------------------------------- /reinforcement_learning_with_spurious_correlations/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/capybaralet/REx_code_release/HEAD/reinforcement_learning_with_spurious_correlations/video.py --------------------------------------------------------------------------------