基于组−信息蒸馏残差网络的轻量级图像超分辨率重建OA北大核心CSTPCD
目前,基于深度学习的超分辨算法已经取得了很好性能,但这些方法通常具有较大内存消耗和较高计算复杂度,很难应用到低算力或便携式设备上.为了解决这个问题,设计一种轻量级的组−信息蒸馏残差网络(Group-information distillation residual network,G-IDRN)用于快速且精确的单图像超分辨率任务.具体地,提出一个更加有效的组−信息蒸馏模块(Group-information distillation block,G-IDB)作为网络特征提取基本块.同时,引入密集快捷连接,对多个基本块进行组合,构建组−信息蒸馏残差组(Group-information distillation residual group,G-IDRG),捕获多层级信息和有效重利用特征.另外,还提出一个轻量的非对称残差Non-local模块,对长距离依赖关系进行建模,进一步提升超分性能.最后,设计一个高频损失函数,去解决像素损失带来图像细节平滑的问题.大量实验结果表明,该算法相较于其他先进方法,可以在图像超分辨率性能和模型复杂度之间取得更好平衡,其在公开测试数据集B100上,4倍超分速率达到56 FPS,比残差注意力网络快15倍.
王云涛;赵蔺;刘李漫;陶文兵;
中南民族大学生物医学工程学院,武汉430074华中科技大学人工智能与自动化学院,武汉430074
计算机与自动化
残差网络超分辨率特征蒸馏高频损失
《自动化学报》 2024 (010)
P.2063-2078 / 16
国家自然科学基金(61976227,62176096);湖北省自然科学基金(2019CFB622)资助。
评论