marrlab / DomainLab

modular domain generalization: https://pypi.org/project/domainlab/
https://marrlab.github.io/DomainLab/
MIT License
40 stars 2 forks source link

implement invariant risk minimization #819

Closed smilesun closed 1 month ago

smilesun commented 2 months ago

paper: https://arxiv.org/pdf/1907.02893 paper section to read: 3.2 Implementation details When estimating the objective (IRMv1) using mini-batches for stochastic gradient descent, one can obtain an unbiased estimate of the squared gradient norm as

reference code:

https://github.com/facebookresearch/DomainBed/blob/93ebb6bccdb46d40471ce0713a165aaf7b5e21b6/domainbed/algorithms.py#L312

smilesun commented 2 months ago

https://github.com/marrlab/DomainLab/pull/824

smilesun commented 1 month ago

Merge pull request https://github.com/marrlab/DomainLab/pull/824 from marrlab/irm c5646fa · 3 minutes ago HistoryMerge pull request https://github.com/marrlab/DomainLab/pull/824 from marrlab/irm c5646fa · 3 minutes ago History