深度学习基础之sofxmax回归

模型输出可以是一个像图像类别的离散值。对于这样的离散值预测问题,可以使用诸如softmax回归在内的分类模型。和线性回归不同,softmax回归的输出单元从一个变成了多个,且引入了softmax运算使输出更适合离散值的预测和训练。

1. 分类问题

考虑⼀个简单的图像分类问题,其输⼊图像的高和宽均为2像素,且色彩为灰度。这样每个像素值都可以用一个标量表示。我们将图像中的4像素分别记为回归图像数据集 图像回归问题_回归。假设训练数据集中图像的真实标签为狗、猫或鸡(假设可以用4像素表示出这3种动物),这些标签分别对应离散值回归图像数据集 图像回归问题_机器学习_02

通常使⽤离散的数值来表示类别,例如回归图像数据集 图像回归问题_深度学习_03。如此,⼀张图像的标签为1、2和3这3个数值中的⼀个。虽然我们仍然可以使⽤回归模型来进⾏建模,并将预测值就近定点化到1、2和3这3个离散值之一,但这种连续值到离散值的转化通常会影响到分类质量。因此我们一般使用更加适合离散值输出的模型来解决分类问题。

2. softmax回归模型

softmax回归跟线性回归一样将输入特征与权重做线性叠加。与线性回归的一个主要不同在于,softmax回归的输出值个数等于标签里的类别数。

回归图像数据集 图像回归问题_回归图像数据集_04
⽤神经⽹络图描绘了上⾯的计算。softmax回归同线性回归⼀样,也是⼀个单层神经⽹络。由于每个输出回归图像数据集 图像回归问题_回归_05的计算都要依赖于所有的输⼊回归图像数据集 图像回归问题_回归_06,softmax回归的输出层也是⼀个全连接层。

回归图像数据集 图像回归问题_机器学习_07

softmax运算

既然分类问题需要得到离散的预测输出,⼀个简单的办法是将输出值回归图像数据集 图像回归问题_机器学习_08当作预测类别是回归图像数据集 图像回归问题_深度学习_09置信度,并将值最大的输出所对应的类作为预测输出,即输出回归图像数据集 图像回归问题_回归图像数据集_10。例如,如果回归图像数据集 图像回归问题_深度学习_11分别为0.1, 10, 0.1,由于回归图像数据集 图像回归问题_深度学习_12最大,那么预测类别为2,其代表猫。然后该方法有很大的问题。⼀方面,由于输出层的输出值的范围不确定,我们难以直观上判断这些值的意义。例如,刚才举的例⼦中的输出值10表示“很置信”图像类别为猫,因为该输出值是其他两类的输出值的100倍。但如果回归图像数据集 图像回归问题_回归图像数据集_13,那么输出值10却⼜表示图像。类别为猫的概率很低。另⼀方面,由于真实标签是离散值,这些离散值与不确定范围的输出值之间的误差难以衡量。

softmax运算符(softmax operator)将输出值变换成值为正且和为1的概率分布,解决了上述问题:
回归图像数据集 图像回归问题_回归_14
其中:
回归图像数据集 图像回归问题_回归_15
容易看出回归图像数据集 图像回归问题_机器学习_16回归图像数据集 图像回归问题_回归图像数据集_17,因此回归图像数据集 图像回归问题_线性回归_18是一个合法的概率分布。如果回归图像数据集 图像回归问题_回归图像数据集_19,不管回归图像数据集 图像回归问题_机器学习_20回归图像数据集 图像回归问题_深度学习_21的值是多少,我们都知道图像类别为猫的概率是80%:
回归图像数据集 图像回归问题_深度学习_22
因此softmax运算不改变预测类别输出。

3. 单样本分类的矢量计算表达式

为了提高计算效率,我们可以将单样本分类通过矢量计算来表达。在上面的图像分类问题中,假设softmax回归的权重和偏差参数分别为:
回归图像数据集 图像回归问题_回归_23
设高和宽分别为2个像素的图像样本回归图像数据集 图像回归问题_深度学习_24的特征为:
回归图像数据集 图像回归问题_深度学习_25
输出层的输出为:
回归图像数据集 图像回归问题_回归图像数据集_26
预测为狗、猫或鸡的概率分布为:
回归图像数据集 图像回归问题_机器学习_27
softmax回归对样本回归图像数据集 图像回归问题_深度学习_24分类的矢量计算表达式为:
回归图像数据集 图像回归问题_回归图像数据集_29

4. ⼩批量样本分类的矢量计算表达式

为了进⼀步提升计算效率,我们通常对小批量数据做⽮量计算。⼴义上讲,给定⼀个小批量样本,其批量大小为回归图像数据集 图像回归问题_回归图像数据集_30,输入个数(特征数)为回归图像数据集 图像回归问题_回归_31,输出个数(类别数)为回归图像数据集 图像回归问题_机器学习_32。设批量特征为回归图像数据集 图像回归问题_回归图像数据集_33。假设softmax回归的权重和偏差参数分别为回归图像数据集 图像回归问题_深度学习_34回归图像数据集 图像回归问题_回归_35。softmax回归的矢量计算表达式为:
回归图像数据集 图像回归问题_深度学习_36
其中的加法运算使⽤了⼴播机制,回归图像数据集 图像回归问题_机器学习_37回归图像数据集 图像回归问题_回归_38且这两个矩阵的第回归图像数据集 图像回归问题_深度学习_24行分别为样本回归图像数据集 图像回归问题_深度学习_24的输出回归图像数据集 图像回归问题_线性回归_41和概率分布回归图像数据集 图像回归问题_线性回归_42

5. 交叉熵损失函数

对于样本回归图像数据集 图像回归问题_深度学习_24,构造向量回归图像数据集 图像回归问题_线性回归_44,使其第回归图像数据集 图像回归问题_线性回归_45(样本回归图像数据集 图像回归问题_深度学习_24类别的离散数值)个元素为1,其余为0。此时训练目标可以设为使预测概率分布回归图像数据集 图像回归问题_回归_47尽可能接近真实的标签概率分布回归图像数据集 图像回归问题_回归_48

交叉熵(cross entropy):
回归图像数据集 图像回归问题_深度学习_49
如果一个样本只有一个标签,于是:
回归图像数据集 图像回归问题_回归图像数据集_50
假设训练数据集的样本数为回归图像数据集 图像回归问题_回归图像数据集_30,交叉熵损失函数定义为:
回归图像数据集 图像回归问题_线性回归_52
其中回归图像数据集 图像回归问题_回归_53代表模型参数。如果每个样本只有一个标签,那么交叉熵损失可以简写成回归图像数据集 图像回归问题_深度学习_54
从另一个角度来说,最小化回归图像数据集 图像回归问题_线性回归_55等价于最大化回归图像数据集 图像回归问题_回归_56,即最小化交叉熵损失函数等价于最⼤化训练数据集所有标签类别的联合预测概率。

6. 模型预测及评价

使用准确率(accuracy)来评价模型的表现。它等于正确预测数量与总预测数量之⽐。