常用

损失函数

BCE Loss

二值交叉熵（Binary Cross Entropy, 简称 BCE）损失函数是二值图像分割中的常用损失函数之一。它用于衡量模型预测的二值输出与实际标签之间的差异。下面是对 BCE 损失函数的详细解释。

Lbce=−1N∑i=1N(yi⋅log⁡(y^i)+(1−yi)⋅log⁡(1−y^i))\mathcal{L}_{bce} = - \frac{1}{N} \sum_{i=1}^{N} (y_i \cdot \log(\hat{y}_i) + (1 - y_i) \cdot \log(1 - \hat{y}_i))

基于交叉熵的概念，交叉熵是一种衡量两个概率分布之间差异的指标。在二值图像分割中，真实标签 yiy_i 和预测概率 y^i\hat{y}_i 可以被看作两个二元分布。其中

  • yiy_i 是第 ii 个像素的真实标签（0 或 1）。
  • y^i\hat{y}_i 是模型预测的第 ii 个像素属于前景的概率（0 到 1 之间的值）。
  • 当真实标签 yi=1y_i = 1 时，损失函数的第一部分 yi⋅log⁡(y^i)y_i \cdot \log(\hat{y}_i) 起作用，如果 y^i\hat{y}_i 越接近 1，损失值越小。
  • 当真实标签 yi=0y_i = 0 时，损失函数的第二部分 (1−yi)⋅log⁡(1−y^i)(1 - y_i) \cdot \log(1 - \hat{y}_i) 起作用，如果 y^i\hat{y}_i 越接近 0，损失值越小。

因此，当模型预测的概率接近真实值时，BCE 损失会较小；反之，如果预测值与真实值偏离较大，损失值就会增大。

原文链接☞

L2 Norm

v=vmax(∥v∥p,ϵ)v = \frac{v}{max(\Vert v \Vert_p, \epsilon)}

归一化输出，使最终结果更加稳定。

实验结果评估

t-SNE

可视化效果最好的降维算法。可用于绘图。

PSNR (Peak Signal-to-Noise Ratio)

衡量重建图像（如超分辨率、去噪）与原始图像的像素级误差，数值越大越好。

PSNR=10⋅log⁡10(MAXI2MSE)\text{PSNR} = 10 \cdot \log_{10}\left(\frac{\text{MAX}_I^2}{\text{MSE}}\right)

  • MAXI\text{MAX}_I​：图像像素最大值（如 8 位图像为255）
  • MSE\text{MSE}：均方误差

MSE=1MN∑i=1M∑j=1N(I(i,j)−K(i,j))2\text{MSE} = \frac{1}{MN} \sum_{i=1}^{M} \sum_{j=1}^{N} (I(i,j) - K(i,j))^2

特点：

  • 计算简单，但对人类视觉感知不敏感（可能与人眼评价不一致）。
  • 单位：分贝（dB），通常值在 20~40 之间，越高越好。

SSIM (Structural Similarity Index)

评估两图像在亮度、对比度和结构上的相似性，数值范围 [0,1]，越大越好。

SSIM(x,y)=(2μxμy+C1)(2σxy+C2)(μx2+μy2+C1)(σx2+σy2+C2)\text{SSIM}(x,y) = \frac{(2\mu_x \mu_y + C_1)(2\sigma_{xy} + C_2)}{(\mu_x^2 + \mu_y^2 + C_1)(\sigma_x^2 + \sigma_y^2 + C_2)}

其中

  • μx,μy\mu_x, \mu_y：图像 xx 和 yy 的均值（亮度）
  • σx,σy\sigma_x, \sigma_y：标准差（对比度）
  • σxy\sigma_{xy}：协方差（结构相关性）
  • C1,C2C_1, C_2：防止除零的小常数（如 C1=(0.01L)2C_1 = (0.01L)^2, C2=(0.03L)2C_2 = (0.03L)^2，LL 为像素范围）

特点：

  • 更符合人类视觉感知，但对局部失真不敏感。

其他指标

  • FID
  • LPIPS

原文链接 ☞