用最小熵和KL散度改进分类损失,让模型更准
Regularizing cross entropy loss via minimum entropy and K-L divergence
- 在交叉熵上加最小熵与反向KL散度正则项
- 在EMNIST-Letters上使VGG准确率提升至95.93%
- 适合追求高精度的图像分类研究者
本文提出两种新型分类损失函数,扩展标准交叉熵损失,通过最小熵和K-L散度进行正则化。第一种为混合熵损失(MIX-ENT),其正则项等价于最小熵与反向K-L散度之和;第二种为最小熵正则化交叉熵损失(MIN-ENT),直接在标准交叉熵上添加最小熵正则项。两者均使神经网络输出的概率分布熵最小化。在EMNIST-Letters数据集上的实验表明,使用MIX-ENT和MIN-ENT的VGG模型,在paperswithcode排行榜上从第3名跃升至第2名,超越Spinal-VGG。标准交叉熵下VGG准确率为95.86%,Spinal-VGG为95.88%;而使用MIN-ENT的VGG达到95.933%,MIX-ENT为95.927%。预训练模型已开源。
原文摘要 · Abstract (English)
I introduce two novel loss functions for classification in deep learning. The two loss functions extend standard cross entropy loss by regularizing it with minimum entropy and Kullback-Leibler (K-L) divergence terms. The first of the two novel loss functions is termed mixed entropy loss (MIX-ENT for short), while the second one is termed minimum entropy regularized cross-entropy loss (MIN-ENT for short). The MIX-ENT function introduces a regularizer that can be shown to be equivalent to the sum of a minimum entropy term and a K-L divergence term. However, it should be noted that the K-L divergence term here is different from that in the standard cross-entropy loss function, in the sense that it swaps the roles of the target probability and the hypothesis probability. The MIN-ENT function simply adds a minimum entropy regularizer to the standard cross entropy loss function. In both MIX-ENT and MIN-ENT, the minimum entropy regularizer minimizes the entropy of the hypothesis probability distribution which is output by the neural network. Experiments on the EMNIST-Letters dataset shows that my implementation of MIX-ENT and MIN-ENT lets the VGG model climb from its previous 3rd position on the paperswithcode leaderboard to reach the 2nd position on the leaderboard, outperforming the Spinal-VGG model in so doing. Specifically, using standard cross-entropy, VGG achieves 95.86% while Spinal-VGG achieves 95.88% classification accuracies, whereas using VGG (without Spinal-VGG) our MIN-ENT achieved 95.933%, while our MIX-ENT achieved 95.927% accuracies. The pre-trained models for both MIX-ENT and MIN-ENT are at https://github.com/rahmanoladi/minimum entropy project.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。