Skip to content

[ICLR 2022] Official pytorch implementation of "Uncertainty Modeling for Out-of-Distribution Generalization" in International Conference on Learning Representations (ICLR) 2022.

Notifications You must be signed in to change notification settings

lixiaotong97/DSU

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Python 3.6 PyTorch 1.5.0

Uncertainty Modeling for Out-of-Distribution Generalization (DSU)

Official pytorch implementation of "Uncertainty Modeling for Out-of-Distribution Generalization" in International Conference on Learning Representations (ICLR) 2022.

By Xiaotong Li, Yongxing Dai, Yixiao Ge, Jun Liu, Ying Shan, Ling-Yu Duan.

Introduction

In this paper, we improve the network generalization ability by modeling the uncertainty of domain shifts with synthesized feature statistics during training. Specifically, we hypothesize that the feature statistic, after considering the potential uncertainties, follows a multivariate Gaussian distribution. Hence, each feature statistic is no longer a deterministic value, but a probabilistic point with diverse distribution possibilities. With the uncertain feature statistics, the models can be trained to alleviate the domain perturbations and achieve better robustness against potential domain shifts.

Overview

Requirements

We use the following versions of OS and softwares:

  • OS: Ubuntu 16.04
  • Python: 3.6.8
  • PyTorch: 1.5.0
  • CUDA: 10.1

Following the instructions of individual repositories to install the required environments.

Experiments

The experiments include: instance retrieval (person re-identification on DukeMTMC and Market1501), multi-domain generalization (PACS and Office-Home), and semantic segmentation (from GTA5 to CityScapes). The core code of our method can be found in dsu.py

Instance Retrieval

Multi-domain Generalization

Semantic Segmentation

Citation

If you find our work is useful for your research, please kindly cite our paper.

@inproceedings{
li2022uncertainty,
title={Uncertainty Modeling for Out-of-Distribution Generalization},
author={Xiaotong Li and Yongxing Dai and Yixiao Ge and Jun Liu and Ying Shan and LINGYU DUAN},
booktitle={International Conference on Learning Representations},
year={2022},
url={https://openreview.net/forum?id=6HN7LHyzGgC}
}

Contact

If you have any questions, you can contact me from the email: [email protected]

About

[ICLR 2022] Official pytorch implementation of "Uncertainty Modeling for Out-of-Distribution Generalization" in International Conference on Learning Representations (ICLR) 2022.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published