中文

NAFNet基线的比较研究:图像修复中的简单而高效网络

计算机视觉与模式识别 2025-06-25 v1 机器学习

摘要

我们研究了NAFNet(无非线性激活网络),这是一种用于图像修复的简单且高效的深度学习基线模型。通过使用带噪声和模糊的CIFAR10图像,我们对NAFNet的核心组件进行了消融研究。我们的基线模型实现了SimpleGate激活、简化通道激活(SCA)以及层归一化。我们将该基线与替换或移除各个组件的不同变体进行比较。定量结果(PSNR、SSIM)以及示例图展示了每种修改如何影响修复性能。我们的发现支持NAFNet的设计:SimpleGate和简化注意力机制的优于常规激活函数和注意力机制,而层归一化对于稳定训练至关重要。我们以模型设计建议、潜在改进及未来工作为结论。

关键词

引用

@article{arxiv.2506.19845,
  title  = {A Comparative Study of NAFNet Baselines for Image Restoration},
  author = {Vladislav Esaulov and M. Moein Esfahani},
  journal= {arXiv preprint arXiv:2506.19845},
  year   = {2025}
}