Ch03 AI数学

信息熵 (Information Entropy)

概述

信息熵(Information Entropy)是信息论的核心概念,由 Claude Shannon 于 1948 年提出。熵衡量一个随机变量的不确定性或信息量。在机器学习中,信息熵及其衍生概念(交叉熵、KL 散度)是理解损失函数、模型评估和信息理论的基础。几乎所有分类问题的损失函数(交叉熵损失)都源于信息熵的概念。

信息熵的定义

对于一个离散随机变量 ,其概率分布为 ,信息熵定义为:

直观理解

  • 低熵:分布集中(如 P=[0.99, 0.01]),不确定性低,H 约等于 0
  • 高熵:分布均匀(如 P=[0.5, 0.5]),不确定性高,H 达到最大
  • 熵的单位取决于对数的底数:log2 为 bits,log e 为 nats

核心衍生概念

交叉熵 (Cross-Entropy)

交叉熵衡量用分布 Q 来编码分布 P 所需的平均编码长度:

在机器学习中,P 是真实分布(ground truth),Q 是模型预测分布。交叉熵损失是分类任务的标准损失函数:

KL 散度 (Kullback-Leibler Divergence)

KL 散度衡量两个分布之间的差异,即用 Q 替代 P 所增加的编码开销:

关键性质:

  • ,当且仅当 P=Q 时取等号
  • 非对称:
  • 不满足三角不等式,不是真正的”距离”

三者关系

概念公式含义机器学习中的应用
信息熵 真实分布的不确定性数据的内在噪音水平
交叉熵 用 Q 编码 P 的代价分类损失函数
KL 散度 P 与 Q 的差异模型蒸馏、VAE 正则化
条件熵 已知 X 后 Y 的不确定性特征选择、决策树分裂
互信息 X 和 Y 共享的信息量特征重要性评估

在机器学习中的关键应用

1. 分类损失函数

多分类交叉熵损失(Categorical Cross-Entropy):

其中 是 one-hot 真值, 是 softmax 输出概率。当模型预测完全正确时,损失为 0;预测完全错误时,损失趋近无穷。

2. 决策树分裂准则

信息增益(Information Gain)用于选择分裂特征:

选择使信息增益最大化的特征进行分裂,等价于选择使子节点熵加权和最小的特征。

3. 变分自编码器 (VAE)

VAE 的损失函数中的 KL 项用于约束隐变量分布接近标准正态分布:

4. 知识蒸馏

在模型蒸馏中,KL 散度用于衡量学生模型输出与教师模型输出的差异:

常见分布的熵值

分布熵公式最大熵条件
伯努利分布 p=0.5 时最大
分类分布均匀分布时最大
正态分布 方差越大熵越大
均匀分布 区间越大熵越大

相关笔记

来源

关联8

知识图谱

局部图谱 · 点节点跳转