首页|一种基于特征知识蒸馏的轻量级图像去噪模型

一种基于特征知识蒸馏的轻量级图像去噪模型

扫码查看
为了构建适用于小型设备的轻量级图像去噪模型,提出了一种基于特征知识蒸馏的新方法.该方法通过学习教师模型的特征图,捕捉深层知识,从而构建轻量级去噪模型,其参数量仅为教师模型的五分之一.实验结果验证了蒸馏算法的有效性,在不同噪声水平及数据集下,都显著提升了学生模型的去噪性能,为轻量级图像去噪模型的构建提供了一种新的方向.
A lightweight image denoising model based on feature knowledge distillation
To construct a lightweight image denoising model suitable for small-scale devices,a novel approach based on feature knowledge distillation was proposed in this paper.Deep-seated knowledge within a teacher model was captured through learning from its feature maps by this method,resulting in the creation of a lightweight denoising model with parameters only one-fifth the size of the teacher model.Experimental results validated the effectiveness of the distillation algorithm,demonstrating significant improvements in denoising performance for the student model across varying noise levels and datasets,which introduced a promising avenue for constructing lightweight image denoising models.

knowledge distillationfeature learningconvolution neural networkimage denoising

沈育、张鼎逆

展开 >

上海师范大学信息与机电工程学院,上海 201418

知识蒸馏 特征学习 卷积神经网络 图像去噪

2024

上海师范大学学报(自然科学版)
上海师范大学

上海师范大学学报(自然科学版)

影响因子:0.255
ISSN:1000-5137
年,卷(期):2024.53(4)