Positional Normalization

NeurIPS 2019, Spotlight

Authors:

• Boyi Li*
• Felix Wu*
• Kilian Q. Weinberger
• Serge Belongie

*: Equal Contribution

Overview

The repo contains TensorFlow and PyTorch Implementation of Positional Normalization (PONO) and Moment Shortcut (MS), described in the paper Positional Normalization.

This image illustrates how PONO differs from some popular nomalization methods: In the figure, each subplot shows a feature map tensor, with B as the batch axis, C as the channel axis, and (H, W) as the spatial axis. The entries colored in green or blue (ours) are normalized by the same mean and standard deviation. Unlike previous methods, our method processes each position independently.

It's suggested to put PONOs right after convolution operations in the encoder (former layers) and MSs right before convolution operations in the decoder (latter layers). One may use a ConvNet taking μ and σ extracted from PONO as inputs to generate β and γ for MS, which we refer as Dynamic Moment Shortcut (DMS). Here we show the illustration.

Please feel free to check our Video for more details.

Explorations beyond our paper

Image Translation

• Cross-domain Correspondence Learning for Exemplar-based Image Translation: Pytorch Version

Image Dehazing:

• Improved AOD-Net Pytorch Version

Image Classification / Detection (Data Augmentation):

• Moment Exchange (MoEx) Pytorch Version

Autoregressive Models:

• Locally Masked Convolution Pytorch Version

Others (Unofficial)

• Semantic Segmentation:

Keras Version

• Underwater Image Enhancement:

Paper

More information and relevant applications will be updated.

If you find this repo useful, please cite:

@inproceedings{li2019positional,
  title={Positional Normalization},
  author={Li, Boyi and Wu, Felix and Weinberger, Kilian Q and Belongie, Serge},
  booktitle={Advances in Neural Information Processing Systems},
  pages={1620--1632},
  year={2019}
}