Softmax回归
Softmax回归(Softmax Regression)是逻辑回归在多分类问题上的推广,它将线性模型的输出映射为每个类别的概率分布,使得所有类别的预测概率之和为1。
基本思想
对于 个类别,Softmax回归为每个类别 学习一组权重向量 和偏置 。对于输入 ,先计算每个类别的线性得分:
Softmax函数
Softmax函数将这些得分转换为概率值。对于第 个类别,其预测概率为:
Softmax函数的特性:
- 输出值在 区间内,所有类别输出之和为1
- 指数运算使得得分高的类别获得更大的概率
- 具有”相对比较”的特性,即某个类别的概率取决于所有类别的得分
损失函数
Softmax回归使用交叉熵损失函数。对于单个样本 ,其中 为真实类别,损失为:
总损失是所有样本损失的平均值。优化的目标是最小化交叉熵损失,使模型对正确类别的预测概率尽可能接近1。
与逻辑回归的关系
当 时,Softmax回归退化为逻辑回归。在二分类情况下,两个类别的 Softmax 输出等价于逻辑回归的 Sigmoid 输出。
应用
Softmax回归广泛应用于多分类问题,如手写数字识别(MNIST)、图像分类、文本分类等场景,同时也是深度神经网络中输出层最常用的激活函数。