4 个月前

不变风险最小化

不变风险最小化

摘要

我们介绍了不变风险最小化(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 中提及
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 中提及
lingxiaoyuan/ood_mechanics
pytorch
GitHub 中提及
siddarth-c/FedGMA
pytorch
GitHub 中提及

基准测试

基准方法指标
image-classification-on-colored-mnist-withMLP-IRM
Accuracy : 66.9

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
不变风险最小化 | 论文 | HyperAI超神经