1.3 softmax回归
通常,机器学习实践者用分类这个词来描述两个有微妙差别的问题:
- 我们只对样本的“硬性”类别感兴趣,即属于哪个类别;
- 我们希望得到“软性”类别,即得到属于每个类别的概率。 这两者的界限往往很模糊。其中的一个原因是:即使我们只关心硬类别,我们仍然使用软类别的模型。
softmax 含义
Softmax 是一种数学函数,它将一个 K 维的实数向量转换为一个 K 维的概率分布。换句话说,它将任意实数值转换为 0 到 1 之间的概率值,并且所有输出的和为 1。
对于一个输入向量Softmax 函数定义为:
其中:
:第 个输入值(logit,未归一化的分数) :对 取指数(确保结果为正) - 分母:所有
的和(归一化项)
分类问题
从一个图像分类问题开始。假设每次输入是一个
对于如何表示标签,我们有两个明显的选择:
最直接的想法是选择
,其中整数分别代表 。这是在计算机上存储此类信息的有效方法。如果类别间有一些自然顺序,比如说我们试图预测 ,那么将这个问题转变为回归问题,并且保留这种格式是有意义的。 如果分类问题并不与类别之间的自然顺序有关,那么采用独热编码(one-hot encoding)是更合适的。独热编码是一个向量,它的分量和类别一样多。类别对应的分量设置为 1,其他所有分量设置为 0。在我们的例子中,标签
将是一个三维向量,其中 对应于“猫”、 对应于“鸡”、 对应于“狗”:
网络架构
为了估计所有可能类别的条件概率,我们需要一个有多个输出的模型,每个类别对应一个输出。
为了解决线性模型的分类问题,我们需要和输出一样多的仿射函数(affine function)。每个输出对应于它自己的仿射函数。
在我们的例子中,由于我们有 4 个特征和 3 个可能的输出类别,我们将需要 12 个标量来表示权重(带下标的
下面我们为每个输入计算三个未规范化的预测(logit):
我们可以用下面的神经网络图来描述这个计算过程。与线性回归一样,softmax 回归也是一个单层神经网络。由于计算每个输出
向量形式表达为
全连接层的参数开销
一个标准的全连接层,如果具有
可以使用分解全连接层的方法解决这个问题,也就是将大的权重矩阵分解为多个小的矩阵相乘,从而减少参数数量;
传统全连接层:
softmax 运算
现在我们将优化参数以最大化观测数据的概率。为了得到预测结果,我们将设置一个阈值,如选择具有最大概率的标签。
我们希望模型的输出
然而不能将未规范化的预测
要将输出视为概率,我们必须保证在任何数据上的输出都是非负的且总和为 1。softmax 函数能够将未规范化的预测变换为非负数并且总和为 1,同时让模型保持可导的性质。为了确保输出非负,我们首先对每个未规范化的预测求幂。为了确保最终输出的概率值总和为 1,再让每个求幂后的结果除以它们的总和。如下式:
这里,对于所有的
softmax 运算不会改变未规范化的预测
因此,在预测过程中,我们仍然可以用下式来选择最有可能的类别。
尽管 softmax 是一个非线性函数,但 softmax 回归的输出仍然由输入特征的仿射变换决定。因此,softmax 回归是一个线性模型(linear model)。
模型的校准
校准(calibration):指的是模型的预测概率与真实发生概率的一致性。
对于一个分类器,如果预测为某一类的概率为
小批量样本的矢量化
为了提高计算效率并且充分利用 GPU,我们通常会对小批量样本的数据执行矢量计算。假设我们读取了一个批量的样本
那么小批量样本的特征为
softmax 回归的矢量计算表达式为:
相对于一次处理一个样本,小批量样本的矢量化加快了
在上面的式子中,
损失函数
接下来,我们需要一个损失函数来度量预测的效果。我们将使用最大似然估计,这与在线性回归中的方法相同。
对数似然
softmax 函数给出了一个向量
例如,
假设整个数据集
根据最大似然估计,我们最大化
在上面的式子中,真实标签是一个独热向量,而模型的预测输出是一个总和为 1 的概率向量,那么对该样本的预测与事实一致的概率,就是概率向量中对应的分量的值;
在数学上可以统一写成交叉熵损失(cross-entropy loss)的形式,将取到当前情况的概率的负对数作为损失,最小化损失函数即是最大似然估计,交叉熵损失函数是softmax最常用的损失函数;
如果正确地预测实际标签,即如果实际标签
softmax 及其导数
由于 softmax 和相关的损失函数很常见,因此我们需要更好地理解它的计算方式。将
将上面的式子对
换句话说,导数是我们 softmax 模型分配的概率与实际发生的情况(由独热标签向量表示)之间的差异。
从这个意义上讲,这与我们在回归中看到的非常相似,其中梯度是观测值
这不是巧合,在任何指数族分布模型中,对数似然的梯度正是由此得出的。这使梯度计算在实践中变得容易很多。
交叉熵损失
对于标签
唯一的区别是,我们现在用一个概率向量表示,如
而不是仅包含二元项的向量
我们使用交叉熵损失函数,它是所有标签分布的预期的损失值。它是分类问题最常用的损失之一。
信息论基础
信息论(information theory)涉及编码、解码、发送以及尽可能简洁地处理信息或数据。
熵
信息论的核心思想是量化数据中的信息内容。
在信息论中,该数值被称为分布
信息论的基本定理之一指出,为了对从分布
信息量
压缩与预测有什么关系呢?想象一下,我们有一个要压缩的数据流。如果我们很容易预测下一个数据,那么这个数据就很容易压缩。为什么呢?举一个极端的例子,假如数据流中的每个数据完全相同,这会是一个非常无聊的数据流。由于它们总是相同的,我们总是知道下一个数据是什么。所以,为了传递数据流的内容,我们不必传输任何信息。也就是说,“下一个数据是xx”这个事件毫无信息量。
但是,如果我们不能完全预测每一个事件,那么我们有时可能会感到”惊异”。克劳德·香农决定用信息量
在上面式子中定义的熵,是当分配的概率真正匹配数据生成过程时的信息量的期望。
重新审视交叉熵
如果把熵
交叉熵从
我们可以把交叉熵想象为“主观概率为
当
在这种情况下,从
简而言之,我们可以从两方面来考虑交叉熵分类目标:
- 最大化观测数据的似然;
- 最小化传达标签所需的惊异。
模型预测和评估
在训练softmax回归模型后,给出任何样本特征,我们可以预测每个输出类别的概率。通常我们使用预测概率最高的类别作为输出类别。如果预测与实际类别(标签)一致,则预测是正确的。在接下来的实验中,我们将使用精度(accuracy)来评估模型的性能。精度等于正确预测数与预测总数之间的比率。
- 标题: 1.3 softmax回归
- 作者: HuxJiang
- 创建于 : 2026-02-07 17:00:28
- 更新于 : 2026-06-25 22:37:11
- 链接: https://github.com/HuxJiang/2026/02/07/23-softmax-regression-z2bnn4d/
- 版权声明: 本文章采用 CC BY-NC-SA 4.0 进行许可。