什么是标签平滑(Label Smoothing)?
标签平滑(Label Smoothing)是一种正则化技术,旨在防止模型过度自信(即输出的概率分布过于“尖锐”)。在分类任务中,标准的目标标签是one-hot编码,也就是正确类别的概率为 1,其他类别的概率为 0。而标签平滑通过将正确类别的概率从 1 调整为一个略小的值(如 0.9),并将剩余的概率均匀分配给其他类别,从而使模型的预测更加平滑,降低过拟合风险。
标签平滑的应用场景
-
防止过拟合:
- 模型如果太过自信(即对某些类别的预测概率接近 1),容易导致过拟合。标签平滑可以通过减少模型的“自信”来缓解这个问题。
-
提升模型的泛化能力:
- 通过引入标签平滑,模型不会对训练集中的标签太过依赖,增强了泛化性能,特别是在小数据集上表现显著。
-
改进硬标签的不足:
- One-hot 标签可能不能完全反映类别之间的关系。例如,在一些多类分类任务中,类别之间可能有一定的相似性,而 one-hot 标签忽略了这些信息。标签平滑能让模型学到更多信息,改善预测结果。