Softmax回归

Softmax回归(Softmax Regression)是逻辑回归在多分类问题上的推广,它将线性模型的输出映射为每个类别的概率分布,使得所有类别的预测概率之和为1。

基本思想

对于 个类别,Softmax回归为每个类别 学习一组权重向量 和偏置 。对于输入 ,先计算每个类别的线性得分:

Softmax函数

Softmax函数将这些得分转换为概率值。对于第 个类别,其预测概率为:

Softmax函数的特性:

  • 输出值在 区间内,所有类别输出之和为1
  • 指数运算使得得分高的类别获得更大的概率
  • 具有”相对比较”的特性,即某个类别的概率取决于所有类别的得分

损失函数

Softmax回归使用交叉熵损失函数。对于单个样本 ,其中 为真实类别,损失为:

总损失是所有样本损失的平均值。优化的目标是最小化交叉熵损失,使模型对正确类别的预测概率尽可能接近1。

与逻辑回归的关系

时,Softmax回归退化为逻辑回归。在二分类情况下,两个类别的 Softmax 输出等价于逻辑回归的 Sigmoid 输出。

应用

Softmax回归广泛应用于多分类问题,如手写数字识别(MNIST)、图像分类、文本分类等场景,同时也是深度神经网络中输出层最常用的激活函数。

链接到