深度学习:交叉熵损失(Cross Entropy Loss)

交叉熵损失(Cross Entropy Loss)

定义和数学表达

交叉熵损失是一种常用于评估概率分类模型性能的损失函数。它衡量的是模型预测的概率分布与真实分布之间的差异。交叉熵损失特别适用于分类任务中,尤其是多类分类问题。

数学上,交叉熵可以定义为:

$C(\\mathbf{y}, \\mathbf{\\hat{y}}) = -\\sum_{i=1}\^{N} y_i \\log(\\hat{y}_i) $

其中:

  • ( y \mathbf{y} y ) 是真实的标签分布,通常表示为one-hot编码向量。
  • ( \\mathbf{\\hat{y}} ) 是模型预测的概率分布,由模型的输出层经过softmax函数转换得到。
  • ( N ) 是类别的总数。
  • ( y i y_i yi ) 是实际标签在第 ( i i i ) 类的值(0或1),( \\hat{y}_i KaTeX parse error: Can't use function '\\)' in math mode at position 1: \\̲)̲ 是预测为第 \\( i ) 类的概率。
特性

交叉熵损失的核心特性包括:

  • 敏感性:这个函数对正确分类的概率非常敏感。如果实际类别的预测概率低(即接近于0),那么损失将会非常高。
  • 非对称性:这种损失在处理极端概率(接近0或1)时表现出明显的非对称性。特别是当预测概率趋近于0时,损失会迅速增加。
交叉熵与信息论

在信息论中,交叉熵衡量的是使用错误的概率分布(模型预测)来编码事件(实际发生的类别)所需的额外信息量。理想情况下,我们希望模型的预测分布尽可能接近真实分布,这样交叉熵就最小,表示预测非常准确。

实例解释

考虑一个简单的三类分类问题,比如预测一张图片是猫、狗还是鸟。假设对于一个实例,真实标签是狗,模型的预测输出(经过softmax)为:

$ \\hat{y} = \[0.1, 0.7, 0.2\]$

对应的真实标签的one-hot编码为:

$ y = \[0, 1, 0\]$

交叉熵损失计算为:

$C(y, \\hat{y}) = -(0 \\times \\log(0.1) + 1 \\times \\log(0.7) + 0 \\times \\log(0.2)) = -\\log(0.7) $

C ( y , y \^ ) ≈ 0.3567 C(y, \\hat{y}) \\approx 0.3567 C(y,y\^)≈0.3567

这表明模型对真实类别(狗)的预测概率为0.7时的损失为0.3567。如果模型对狗的预测概率更高,比如0.9,则损失会更低,显示为:

C ( y , y \^ ) = − log ⁡ ( 0.9 ) ≈ 0.1054 C(y, \\hat{y}) = -\\log(0.9) \\approx 0.1054 C(y,y\^)=−log(0.9)≈0.1054

结论

交叉熵损失函数是监督学习中非常重要的工具,特别是在处理分类问题时。它不仅提供了一种衡量模型性能的方法,还通过梯度下降等优化算法指导了模型的学习过程。优化交叉熵损失可以帮助模型更好地学习区分不同类别,提高分类的准确率。

相关推荐
duyinbi7517几秒前
【环境监测】河流目标检测:建筑物、树木和水轮机识别实战
人工智能·目标检测·计算机视觉
cute_ming2 分钟前
浅谈提示词工程:企业级系统化实践与自动化架构(三)
人工智能·ubuntu·机器学习·架构·自动化
PHOSKEY4 分钟前
QM系列闪测仪在医用核心原材料(TC4/PEEK/ 硅胶)质控方案
人工智能·机器学习
WJSKad12356 分钟前
【狮子目标检测】YOLO13-GhostDynamicConv改进模型实现与性能分析
人工智能·目标检测·计算机视觉
Toky丶9 分钟前
【文献阅读】ARB-LLM: Alternating Refined Binarizations for Large Language Models
人工智能·语言模型·自然语言处理
斯外戈的小白12 分钟前
【NLP】Hugging Face使用指南
人工智能·自然语言处理
victory043113 分钟前
关于深度学习的重要库 transformer
人工智能·深度学习·transformer
love530love14 分钟前
Flash Attention 2.8.3 在 Windows + RTX 3090 上成功编译与运行复盘笔记(2026年1月版)
人工智能·windows·笔记·python·flash_attn·flash attention·z-image
虹科网络安全14 分钟前
艾体宝洞察 | “顶会”看安全(四):Black hat-揭示 PyTorch 中 TorchScript 引擎的潜在风险
人工智能·pytorch·安全
yumgpkpm15 分钟前
Cloudera CDH5、CDH6、CDP7现状及替代方案
数据库·人工智能·hive·hadoop·elasticsearch·数据挖掘·kafka