Noisy Softmax: Improving the Generalization Ability of DCNN via Postponing the Early Softmax Saturation

By Binghui Chen, Weihong Deng, Junping Du. 2017

132 citations recorded by the source corpus.

Over the past few years, softmax and SGD have become a commonly used component and the default training strategy in CNN frameworks, respectively. However, when optimizing CNNs with SGD, the saturation behavior behind softmax always gives us an illusion of training well and then is omitted. In this paper, we first emphasize that the early saturation behavior of softmax will impede the exploration of SGD, which sometimes is a reason for model converging at a bad local-minima, then propose Noisy Softmax to mitigating this early saturation issue by injecting annealed noise in softmax during each iteration. This operation based on noise injection aims at postponing the early saturation and further bringing continuous gradients propagation so as to significantly encourage SGD solver to be more exploratory and help to find a better local-minima. This paper empirically verifies the superiority of the early softmax desaturation, and our method indeed improves the generalization ability of CNN model by regularization. We experimentally find that this early desaturation helps optimization in many tasks, yielding state-of-the-art or competitive results on several popular benchmark datasets.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Qiang Liu
  2. Dilin Wang
  3. Jiani Hu
  4. Tianyue Zheng
  5. Abdelmalik Taleb-Ahmed
  6. Yuki Yamanaka
  7. Yasuhiro Fujiwara
  8. Weihong Deng

In Computer Vision and Pattern Recognition

  1. Qiang Liu
  2. Dilin Wang
  3. Jiani Hu
  4. Weihong Deng
  5. Tianyue Zheng
  6. Abdelmalik Taleb-Ahmed
  7. Yuki Yamanaka
  8. Yasuhiro Fujiwara

Citations in the corpus