深度学习笔记:详解MSE Loss均方误差损失函数[通俗易懂]

欢迎Star我的Machine Learning Blog:https://github.com/purepisces/Wenqing-Machine_Learning_Blog。

均方误差损失

均方误差（MSE，Mean Squared Error）是回归问题中评估预测误差的一种广泛使用的指标。在回归中，目标是预测连续值，例如根据房屋的特征（如面积、位置、卧室数量等）估计房屋的价格。

MSE 损失前向方程

计算从计算模型预测（$A$）与实际真值（$Y$）之间的平方误差（$SE$）开始：

$$SE(A,Y)=(A-Y)\odot (A-Y)$$

接下来，我们确定平方误差之和（$SSE$）。这里，${\iota }_N$ 和 ${\iota }_C$ 分别表示大小为 $N$ 和 $C$ 的、填充有 1 的列向量：

$$SSE(A,Y)={\iota }_N^T\cdot SE(A,Y)\cdot {\iota }_C$$

这个操作将 $SE(A,Y)$ 矩阵中的所有元素求和，该矩阵的维度为 $N\times C$。通过 ${\iota }_N^T$ 的乘法在行间聚合错误，随后通过 ${\iota }_C$ 的乘法将这些在列间求和，产生一个单一标量的总误差。

然后计算每个组件的均方误差（$MSE$）损失：

$$MSELoss(A,Y)=\frac{SSE(A,Y)}{N\cdot C}$$

MSE 损失反向方程

在反向传播过程中，需要计算 MSE 损失相对于模型输出（$A$）的梯度，以更新模型参数：

$$MSELoss.backward()=2\cdot \frac{(A-Y)}{N\cdot C}$$

MSE 损失的导数

MSE 损失函数定义为：

$$MSELoss(A,Y)=\frac{1}{N\cdot C}\sum _{i=1}^N\sum _{j=1}^C(A_{ij}-Y_{ij}{)}^2$$

其中：

$A$：模型预测值。

$Y$：真实值。

$N$：批次中的样本数量。

$C$：每个样本的输出维度，回归任务中通常为 1。

为了更新模型参数（在这个例子中，通过反向传播），我们需要知道 $A$ 的变化如何影响损失。这由损失函数相对于 $A$ 的导数给出，表示为 $\frac{\partial MSELoss}{\partial A}$。

$$\frac{\partial MSELoss}{\partial A}=2\cdot \frac{(A-Y)}{N\cdot C}$$

梯度 $\frac{\partial MSELoss}{\partial A}$ 指向损失函数最陡增加的方向。通过向相反方向移动（即，从预测 $A$ 中减去这个梯度），我们可以减少损失，这是训练模型的目标。

总之，$MSELoss.backward()$ 的 $2\cdot \frac{(A-Y)}{N\cdot C}$ 公式是通过对预测 $A$ 求 MSE 损失函数的导数得到的，考虑了平方误差和平均操作。这个梯度在优化过程中被用来调整模型参数，以最小化损失。

import numpy as np

class MSELoss:
    def forward(self, A, Y):
        # 为反向计算存储预测值（A）和真实值（Y）
        self.A = A
        self.Y = Y
        # 计算预测值和真实值之间的平方误差
        se = (A - Y) ** 2
        # 对平方误差求和以得到总的平方误差
        sse = np.sum(se)
        # 通过将总的平方误差除以元素数量来计算均方误差
        mse = sse / (A.shape[0] * A.shape[1])
        return mse

    def backward(self):
        # 计算损失相对于预测值（A）的梯度
        dLdA = 2 * (self.A - self.Y) / (self.A.shape[0] * self.A.shape[1])
        return dLdA

示例

让我们通过一个具体的例子来了解在回归场景中如何应用均方误差（MSE）损失。假设我们正试图基于一些特征来预测房屋价格。为了简单起见，我们将考虑一个案例，其中我们的模型基于多个特征（例如平方英尺面积和卧室数量）来预测两栋房屋的价格，因此我们的批量大小 $N$ 为 2，特征数量 $C$ 也为 2。

给定数据：

• 模型输出 ($A$)：预测的两栋房屋的价格和卧室数量。假设模型对每栋房屋的预测如下（价格以美元计，卧室以数量计）。这可以表示为一个 2×2 矩阵（因为 $N=2$ 且 $C=2$）：

$$A=\left[\begin{array}{cc}300,000& 3\\ 500,000& 4\end{array}\right]$$

这里，第一列代表两栋房屋的预测价格，第二列代表预测的卧室数量。

• 真实值 ($Y$)：两栋房屋的实际价格和卧室数量。这也是一个 2×2 矩阵：

$$Y=\left[\begin{array}{cc}350,000& 4\\ 450,000& 3\end{array}\right]$$

正向传播（计算 MSE 损失）：

1. 计算平方误差 ($SE$)：

$$\begin{array}{rl}SE(A,Y)& =(A-Y)\odot (A-Y)\\ & =\left[\begin{array}{cc}(300,000-350,000{)}^2& (3-4{)}^2\\ (500,000-450,000{)}^2& (4-3{)}^2\end{array}\right]\\ & =\left[\begin{array}{cc}2500\times 10^6& 1\\ 2500\times 10^6& 1\end{array}\right]\end{array}$$

2. 平方误差之和 ($SSE$)：
   将 $SE$ 中的所有元素相加：

$$SSE(A,Y)=\sum SE(A,Y)=2\times (2500\times 10^6)+2\times 1=5000\times 10^6+2$$

3. 均方误差 ($MSE$)：

$$MSELoss(A,Y)=\frac{SSE(A,Y)}{N\cdot C}=\frac{5000\times 10^6+2}{2\times 2}=\frac{2500\times 10^6+1}{2}$$

反向传播（计算梯度）：

可以计算损失相对于预测值 ($A$) 的梯度：

$$\begin{array}{rl}\frac{\partial MSELoss}{\partial A}& =2\cdot \frac{(A-Y)}{N\cdot C}\\ & =2\cdot \frac{\left[\begin{array}{cc}300,000-350,000& 3-4\\ 500,000-450,000& 4-3\end{array}\right]}{2\times 2}\\ & =\frac{1}{2}\left[\begin{array}{cc}-50,000& -1\\ 50,000& 1\end{array}\right]\end{array}$$

这个梯度矩阵提供了如何调整每个预测（价格和卧室数量）以最小化损失的指导。负值表示需要增加预测值，正值表明需要减少预测值以减少误差。

参考：

• CMU_11785_深度学习导论

今天的文章深度学习笔记:详解MSE Loss均方误差损失函数[通俗易懂]分享到此就结束了，感谢您的阅读。