线性回归输出是一个连续值,因此适用于回归问题。回归问题在实际中很常见,如预测房屋价格、气温等连续值的问题。与回归问题不同,分类问题中模型的输出是一个离散值。我们所说的图像分类、疾病检测等输出为离散值的问题都属于分类问题的范畴。softmax回归则适用于分类问题
由于线性回归和softmax回归都是单层神经网络,它们涉及的概念和技术同样适用于大多数的深度学习模型。首先以线性回归为例,介绍大多数深度学习模型的基本要素和表示方法


线性回归详解

首先以一个简单的房屋价格预测作为例子来进行切入。目标是预测一栋房子的售出价格(元)。我们知道这个价格取决于很多因素,如房屋状况、地段、市场行情等。为了简单起见,这里我们假设价格只取决于房屋状况的两个因素,即面积(平方米)和房龄(年),因此我们需要探索价格与这两个因素的具体关系

模型定义

设房屋的面积为 线性回归 - 图1,房龄为 线性回归 - 图2 ,售出价格为 线性回归 - 图3 。我们需要建立基于输入 线性回归 - 图4线性回归 - 图5 来计算输出 线性回归 - 图6的表达式,也就是模型(model)。顾名思义,线性回归假设输出与各个输入之间是线性关系:
线性回归 - 图7

其中 线性回归 - 图8线性回归 - 图9 是权重(weight),线性回归 - 图10 是偏差(bias),且均为标量。它们是线性回归模型的参数。模型输出 线性回归 - 图11 是线性回归对真实价格 线性回归 - 图12 的预测或估计

模型训练

接下来需要通过数据来寻找特定的模型参数值,使模型在数据上的误差尽可能小。这个过程叫作模型训练。下面介绍模型训练所涉及的3个要素

(1) 训练数据

我们通常收集一系列的真实数据,例如多栋房屋的真实售出价格和它们对应的面积和房龄。我们希望在这个数据上面寻找模型参数来使模型的预测价格与真实价格的误差最小。在机器学习术语里,该数据集被称为训练数据集(training data set)或训练集(training set),一栋房屋被称为一个样本(sample),其真实售出价格叫作标签(label),用来预测标签的两个因素叫作特征(feature)。特征用来表征样本的特点
假设我们采集的样本数为 线性回归 - 图13,索引为 线性回归 - 图14 的样本的特征为 线性回归 - 图15线性回归 - 图16,标签为 线性回归 - 图17。对于索引为 线性回归 - 图18 的房屋,线性回归模型的房屋价格预测表达式为
线性回归 - 图19

(2) 损失函数

在模型训练中,我们需要衡量价格预测值与真实值之间的误差。通常我们会选取一个非负数作为误差,且数值越小表示误差越小。一个常用的选择是平方函数。它在评估索引为 线性回归 - 图20 的样本误差的表达式为

线性回归 - 图21

其中常数 线性回归 - 图22 使对平方项求导后的常数系数为1,这样在形式上稍微简单一些。显然,误差越小表示预测价格与真实价格越相近,且当二者相等时误差为0。给定训练集,这个误差只与模型参数相关。在机器学习里,将衡量误差的函数称为损失函数(loss function)。这里使用的平方误差函数也称为平方损失(square loss)
通常,我们用训练集中所有样本误差的平均来衡量模型预测的质量,即

线性回归 - 图23

训练中,我们希望找出一组模型参数,记为 线性回归 - 图24 , 线性回归 - 图25 , 线性回归 - 图26 ,来使训练样本平均损失最小:

线性回归 - 图27

(3) 优化算法

当模型和损失函数形式较为简单时,上述误差最小化问题的解可以直接用公式表达出来。这类解叫作解析解(analytical solution)。本节使用的线性回归和平方误差刚好属于这个范畴。然而,大多数深度学习模型并没有解析解,只能通过优化算法有限次迭代模型参数来尽可能降低损失函数的值。这类解叫作数值解(numerical solution)
在求数值解的优化算法中,小批量随机梯度下降(mini-batch stochastic gradient descent)在深度学习中被广泛使用。它的算法很简单:先选取一组模型参数的初始值,如随机选取;接下来对参数进行多次迭代,使每次迭代都可能降低损失函数的值。在每次迭代中,先随机均匀采样一个由固定数目训练数据样本所组成的小批量(mini-batch)线性回归 - 图28,然后求小批量中数据样本的平均损失有关模型参数的导数(梯度),最后用此结果与预先设定的一个正数的乘积作为模型参数在本次迭代的减小量。
在训练本节讨论的线性回归模型的过程中,模型的每个参数将作如下迭代:

线性回归 - 图29

在上式中,线性回归 - 图30 代表每个小批量中的样本个数(批量大小,batch size),线性回归 - 图31 称作学习率并取正数。需要强调的是,这里的批量大小和学习率的值是人为设定的,并不是通过模型训练学出的,因此叫作超参数。我们通常所说的“调参”指的正是调节超参数,在少数情况下,超参数也可以通过模型训练学出

模型预测

模型训练完成后,我们将模型参数 线性回归 - 图32 在优化算法停止时的值分别记作 线性回归 - 图33 。注意,这里得到的并不一定是最小化损失函数的最优解 线性回归 - 图34,而是对最优解的一个近似。然后,我们就可以使用学出的线性回归模型 线性回归 - 图35 来估算训练集以外任意一栋房屋的价格了。这里的估算也叫作模型预测、推断或测试


线性回归的表示方法

我们已经阐述了线性回归的模型表达式、训练和预测。下面我们解释线性回归与神经网络的联系,以及线性回归的矢量计算表达式

神经网络图

在深度学习中,我们可以使用神经网络图直观地表现模型结构。为了更清晰地展示线性回归作为神经网络的结构,下图使用神经网络图表示线性回归模型。神经网络图隐去了模型参数权重和偏差

线性回归 - 图36

如上图所示,输入分别为 线性回归 - 图37线性回归 - 图38,因此输入层的输入个数为2。输入个数也叫特征数或特征向量维度。上图中网络的输出为 线性回归 - 图39,输出层的输出个数为1
需要注意的是,我们直接将神经网络的输出 线性回归 - 图40 作为线性回归的输出,即 线性回归 - 图41 。由于输入层并不涉及计算,按照惯例,上图所示的神经网络的层数为1。所以,线性回归是一个单层神经网络。输出层中负责计算 线性回归 - 图42 的单元又叫神经元。在线性回归中,线性回归 - 图43 的计算依赖于 线性回归 - 图44线性回归 - 图45。也就是说,输出层中的神经元和输入层中各个输入完全连接。因此,这里的输出层又叫全连接层(fully-connected layer)或稠密层(dense layer)

矢量计算表达式

在模型训练或预测时,我们常常会同时处理多个数据样本并用到矢量计算。在介绍线性回归的矢量计算表达式之前,我们先考虑对两个向量相加的两种方法
下面先定义两个1000维的向量

  1. import torch
  2. from time import time
  3. a = torch.ones(1000)
  4. b = torch.ones(1000)

向量相加的一种方法是,将这两个向量按元素逐一做标量加法

  1. start = time()
  2. c = torch.zeros(1000)
  3. for i in range(1000):
  4. c[i] = a[i] + b[i]
  5. print(time() - start)
  6. 0.02039504051208496

向量相加的另一种方法是,将这两个向量直接做矢量加法

  1. start = time()
  2. d = a + b
  3. print(time() - start)
  4. 0.0008330345153808594

结果很明显,后者比前者更省时。因此,我们应该尽可能采用矢量计算,以提升计算效率
再次回到房价预测问题。如果我们对训练集里的3个房屋样本(索引分别为1、2和3)逐一预测价格,将得到

线性回归 - 图46

现在,我们将上面3个等式转化成矢量计算。设
线性回归 - 图47
对3个房屋样本预测价格的矢量计算表达式为线性回归 - 图48, 其中的加法运算使用了广播机制。例如:

  1. a = torch.ones(3)
  2. b = 10
  3. print(a + b)
  4. tensor([11., 11., 11.])

广义上讲,当数据样本数为 线性回归 - 图49,特征数为 线性回归 - 图50 时,线性回归的矢量计算表达式为
线性回归 - 图51
其中模型输出 线性回归 - 图52 批量数据样本特征 线性回归 - 图53,权重 线性回归 - 图54, 偏差 线性回归 - 图55
批量数据样本标签 线性回归 - 图56。设模型参数 线性回归 - 图57,我们可以重写损失函数为
线性回归 - 图58
小批量随机梯度下降的迭代步骤将相应地改写为
线性回归 - 图59
其中梯度是损失有关3个为标量的模型参数的偏导数组成的向量:
线性回归 - 图60