Abstract | ||
---|---|---|
We propose regularizing neural networks by penalizing low entropy output distributions. We show that penalizing low entropy output distributions, which has been shown to improve exploration in reinforcement learning, acts as a strong regularizer in supervised learning. We connect our confidence penalty to label smoothing through the direction of the KL divergence between networks output distribution and the uniform distribution. We exhaustively evaluate our proposed confidence penalty and label smoothing (uniform and unigram) on 6 common benchmarks: image classification (MNIST and Cifar-10), language modeling (Penn Treebank), machine translation (WMTu002714 English-to-German), and speech recognition (TIMIT and WSJ). We find that both label smoothing and our confidence penalty improve state-of-the-art models across benchmarks without modifying existing hyper-parameters. |
Year | Venue | Field |
---|---|---|
2017 | ICLR | TIMIT,MNIST database,Pattern recognition,Computer science,Supervised learning,Smoothing,Artificial intelligence,Deep learning,Artificial neural network,Kullback–Leibler divergence,Machine learning,Language model |
DocType | Volume | Citations |
Journal | abs/1701.06548 | 42 |
PageRank | References | Authors |
1.24 | 22 | 5 |
Name | Order | Citations | PageRank |
---|---|---|---|
Gabriel Pereyra | 1 | 42 | 1.58 |
George Tucker | 2 | 160 | 15.17 |
Jan Chorowski | 3 | 673 | 33.81 |
Łukasz Kaiser | 4 | 2307 | 89.08 |
geoffrey e hinton | 5 | 40435 | 4751.69 |