深度学习中常见的损失函数
不同的损失函数适用于不同的任务,回归任务中通常使用均方误差(MSE)和绝对误差(MAE),分类任务中使用交叉熵损失,物体检测和分割中使用IoU和Dice损失,而生成模型中使用对抗损失和重构损失。根据具体任务的特点选择合适的损失函数对于模型的性能有很大影响。
在机器学习和深度学习中,损失函数用于衡量模型预测值与真实值之间的差异。根据任务的类型(如回归、分类等),可以使用不同的损失函数。下面列举了一些常见的损失函数:
1. 回归问题中的损失函数
回归任务的目标是预测连续值,常用的损失函数有:
-
均方误差(Mean Squared Error, MSE):
LMSE=1N∑i=1N(yi−y^i)2 L_{\text{MSE}} = \frac{1}{N} \sum_{i=1}^{N} (y_i - \hat{y}_i)^2 LMSE=N1i=1∑N(yi−y^i)2- 这是最常用的回归损失函数之一。它对误差的平方进行加权,较大的误差对损失的贡献更大。
-
均方根误差(Root Mean Squared Error, RMSE):
LRMSE=1N∑i=1N(yi−y^i)2 L_{\text{RMSE}} = \sqrt{\frac{1}{N} \sum_{i=1}^{N} (y_i - \hat{y}_i)^2} LRMSE=N1i=1∑N(yi−y^i)2- RMSE是MSE的平方根形式,通常用于更直观地表示误差。
-
平均绝对误差(Mean Absolute Error, MAE):
LMAE=1N∑i=1N∣yi−y^i∣ L_{\text{MAE}} = \frac{1}{N} \sum_{i=1}^{N} |y_i - \hat{y}_i| LMAE=N1i=1∑N∣yi−y^i∣- MAE对误差的线性差异进行惩罚,它对异常值的影响不如MSE大。
-
Huber损失(Huber Loss):
Lδ(a)={12a2if ∣a∣≤δδ(∣a∣−12δ)if ∣a∣>δ L_{\delta}(a) = \begin{cases} \frac{1}{2} a^2 & \text{if } |a| \leq \delta \\ \delta (|a| - \frac{1}{2} \delta) & \text{if } |a| > \delta \end{cases} Lδ(a)={21a2δ(∣a∣−21δ)if ∣a∣≤δif ∣a∣>δ- 这是MSE和MAE的折中,适用于存在异常值的数据集。它对小误差使用MSE,对大误差使用MAE。
2. 分类问题中的损失函数
分类任务的目标是将数据点分配到离散类别中,常用的损失函数包括:
-
交叉熵损失(Cross Entropy Loss):
-
对于二分类问题:
LCE=−1N∑i=1N[yilog(y^i)+(1−yi)log(1−y^i)] L_{\text{CE}} = - \frac{1}{N} \sum_{i=1}^{N} \left[ y_i \log(\hat{y}_i) + (1 - y_i) \log(1 - \hat{y}_i) \right] LCE=−N1i=1∑N[yilog(y^i)+(1−yi)log(1−y^i)]- ( y_i ) $$ 是真实类别标签(0或1),( \hat{y}_i ) 是预测值(0到1之间的概率值)。
-
对于多分类问题(Softmax交叉熵损失):
LCE=−1N∑i=1N∑k=1Kyiklog(y^ik) L_{\text{CE}} = - \frac{1}{N} \sum_{i=1}^{N} \sum_{k=1}^{K} y_{ik} \log(\hat{y}_{ik}) LCE=−N1i=1∑Nk=1∑Kyiklog(y^ik)- 这里yiky_{ik}yik 是类别kkk 的真实标签(one-hot编码),y^ik\hat{y}_{ik}y^ik 是模型输出的类别概率。
-
-
Hinge Loss(用于SVM):
Lhinge=max(0,1−yi⋅y^i) L_{\text{hinge}} = \max(0, 1 - y_i \cdot \hat{y}_i) Lhinge=max(0,1−yi⋅y^i)- 常用于支持向量机(SVM)分类器中,确保正确类别的得分与其他类别的得分相差至少1。
-
Kullback-Leibler散度(Kullback-Leibler Divergence, KL Divergence):
DKL(P∣∣Q)=∑iP(i)log(P(i)Q(i)) D_{\text{KL}}(P || Q) = \sum_{i} P(i) \log\left(\frac{P(i)}{Q(i)}\right) DKL(P∣∣Q)=i∑P(i)log(Q(i)P(i))- 用于衡量两个概率分布 PPP 和 QQQ 之间的差异,通常用于分类任务的概率分布预测中。
3. 用于检测和分割的损失函数
这些损失函数常用于图像处理任务,如物体检测和图像分割:
-
IoU损失(Intersection over Union Loss):
LIoU=1−Intersection AreaUnion Area L_{\text{IoU}} = 1 - \frac{\text{Intersection Area}}{\text{Union Area}} LIoU=1−Union AreaIntersection Area- 用于目标检测,衡量预测框与真实框之间的重叠面积比例。
-
Dice损失(Dice Loss):
LDice=1−2∣A∩B∣∣A∣+∣B∣ L_{\text{Dice}} = 1 - \frac{2 |A \cap B|}{|A| + |B|} LDice=1−∣A∣+∣B∣2∣A∩B∣- 多用于图像分割任务,衡量预测与真实分割区域之间的重叠。
4. 生成模型中的损失函数
生成模型如GANs(生成对抗网络)和自编码器中使用的损失函数包括:
-
对抗损失(Adversarial Loss):
- GAN中用于生成器和判别器的损失。生成器的目标是“欺骗”判别器,判别器的目标是正确地区分真实数据和生成数据。
Lgen=−log(D(G(z))) L_{\text{gen}} = -\log(D(G(z))) Lgen=−log(D(G(z)))
Ldis=−[log(D(x))+log(1−D(G(z)))] L_{\text{dis}} = -\left[\log(D(x)) + \log(1 - D(G(z)))\right] Ldis=−[log(D(x))+log(1−D(G(z)))]
- GAN中用于生成器和判别器的损失。生成器的目标是“欺骗”判别器,判别器的目标是正确地区分真实数据和生成数据。
-
重构损失(Reconstruction Loss):
- 常用于自编码器和变分自编码器(VAE),衡量输入数据和重构数据之间的差异,常使用MSE或交叉熵。
5. 其他常见损失函数
-
标签平滑损失(Label Smoothing Loss):
Lsmooth=−∑i=1N∑k=1K[(1−ϵ)yik+ϵK]log(y^ik) L_{\text{smooth}} = - \sum_{i=1}^{N} \sum_{k=1}^{K} \left[(1 - \epsilon) y_{ik} + \frac{\epsilon}{K}\right] \log(\hat{y}_{ik}) Lsmooth=−i=1∑Nk=1∑K[(1−ϵ)yik+Kϵ]log(y^ik)- 在交叉熵损失中引入平滑标签,避免过拟合或过度自信的预测。
-
对比损失(Contrastive Loss):
Lcontrastive=(1−y)⋅12(Dw)2+y⋅12{max(0,m−Dw)}2 L_{\text{contrastive}} = (1 - y) \cdot \frac{1}{2} (D_w)^2 + y \cdot \frac{1}{2} \{ \max(0, m - D_w) \}^2 Lcontrastive=(1−y)⋅21(Dw)2+y⋅21{max(0,m−Dw)}2- 常用于度量学习任务(如Siamese网络),衡量相似样本和不相似样本之间的距离。
总结
不同的损失函数适用于不同的任务,回归任务中通常使用均方误差(MSE)和绝对误差(MAE),分类任务中使用交叉熵损失,物体检测和分割中使用IoU和Dice损失,而生成模型中使用对抗损失和重构损失。根据具体任务的特点选择合适的损失函数对于模型的性能有很大影响。
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐

所有评论(0)