h-yu16 / Sparse_Variable_Independence

Official repo for AAAI 2023 paper "Stable Learning via Sparse Variable Independence".

Sparse Variable Independence

Official repo for AAAI 2023 paper Stable Learning via Sparse Variable Independence.

Quick Start

Running SVI

CUDA_VISIBLE_DEVICES=0 python3 exp_svi.py 

Running other algorithms

OLS

CUDA_VISIBLE_DEVICES=0 python3 exp_linear.py --reweighting None --paradigm regr

STG

CUDA_VISIBLE_DEVICES=0 python3 exp_linear.py --reweighting None --paradigm fs

DWR

CUDA_VISIBLE_DEVICES=0 python3 exp_linear.py --reweighting DWR --paradigm regr

SVI $^d$

CUDA_VISIBLE_DEVICES=0 python3 exp_linear.py --reweighting DWR --paradigm fs

Citing

If you find this repo useful for your research, please consider citing the paper.

@article{yu2022stable,
  title={Stable Learning via Sparse Variable Independence},
  author={Yu, Han and Cui, Peng and He, Yue and Shen, Zheyan and Lin, Yong and Xu, Renzhe and Zhang, Xingxuan},
  journal={arXiv preprint arXiv:2212.00992},
  year={2022}
}