
摘要
我们介绍了不变风险最小化(Invariant Risk Minimization, IRM),这是一种学习范式,旨在估计多个训练分布之间的不变相关性。为了实现这一目标,IRM 学习一种数据表示方法,使得在这种数据表示基础上的最优分类器在所有训练分布中都保持一致。通过理论分析和实验验证,我们展示了 IRM 学到的不变性如何与数据的因果结构相关联,并促进模型在未见分布上的泛化能力。
代码仓库
parshakova/IRM_ICP-IC_ERM
pytorch
GitHub 中提及
thuml/Transfer-Learning-Library
pytorch
GitHub 中提及
facebookresearch/DomainBed
pytorch
GitHub 中提及
hasanjawad001/cglearn
pytorch
GitHub 中提及
kakaobrain/irm-empirical-study
pytorch
GitHub 中提及
reiinakano/invariant-risk-minimization
pytorch
GitHub 中提及
aniquetahir/jax_ood
jax
GitHub 中提及
facebookresearch/InvariantRiskMinimization
官方
pytorch
GitHub 中提及
claudiashi57/nice
pytorch
GitHub 中提及
ycq091044/manydg
pytorch
GitHub 中提及
chunyangx/IRM_research
pytorch
GitHub 中提及
rwchakra/exmap
pytorch
GitHub 中提及
fastforwardlabs/causality-for-ml
pytorch
GitHub 中提及
katoro8989/irm_variants_calibration
pytorch
GitHub 中提及
oddskool/invariant_risk_minimization_reproduced
tf
GitHub 中提及
lingxiaoyuan/ood_mechanics
pytorch
GitHub 中提及
siddarth-c/FedGMA
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| image-classification-on-colored-mnist-with | MLP-IRM | Accuracy : 66.9 |