Code for reproducing the results of our accpeted CVPR 2020 paper (Oral): An Investigation of Stochasticity of Batch Whitening (arXiv:2003.12327), where we qualitatively and quantitively investigate the stochasticity caused by Batch Whitening (BW), during training and inference.
This repo provides the Pytorch and Tensorflow implementations of different whitening methods.
SBW_Classification_PyTorch: Pytorch project for reproducing the discriminative classification experimentsSBW_GAN_TF: Tensorflow project for reproducing the GAN experiments.
If you find this repo benefits your research, please consider citing:
@inproceedings{2020_CVPR_Huang,
author = {Lei Huang and Lei Zhao and Yi Zhou and Fan Zhu and Li Liu and Ling Shao},
title = {Iterative Normalization: Beyond Standardization towards Efficient Whitening},
booktitle = {IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
year = {2020},
}
Email: huanglei@nlsde.buaa.edu.cn.