arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2605.03680cs.CVcs.LG

面向高性能移动NPU的真实图像去噪知识蒸馏

Real Image Denoising with Knowledge Distillation for High-Performance Mobile NPUs

Computer Vision Lab, CAIDAS & IFI, University of Würzburg, Germany(计算机视觉实验室,CAIDAS与IFI,乌尔姆大学,德国)

Faraz Kayani, Sarmad Kayani, Asad Ahmed, Radu Timofte, Dmitry Ignatov

AI总结:

提出NPU感知的硬件-算法协同设计方法,通过知识蒸馏将高容量教师网络的知识迁移到轻量级学生网络,在移动NPU上实现高效去噪,PSNR仅损失0.05 dB,推理速度比GPU快3.88倍。

摘要:

虽然基于深度学习的图像恢复已实现前所未有的保真度,但在移动神经处理单元(NPU)上的部署仍受限于算子不兼容和内存访问开销。我们提出了一种NPU感知的硬件-算法协同设计方法,用于移动NPU上的真实图像去噪。该方法采用高容量教师网络监督轻量级学生网络,该学生网络专门设计以利用现代移动SoC的平铺内存架构。通过优先使用NPU原生原语——标准3x3卷积、ReLU激活和最近邻上采样——并采用渐进式上下文扩展策略(高达1024x1024裁剪),该模型在Mobile AI 2026挑战赛的全分辨率(2432x3200)验证基准上达到37.66 dB PSNR / 0.9278 SSIM,在保留测试基准上达到37.58 dB PSNR / 0.9098 SSIM。根据官方挑战规则,在标准全高清(1088x1920)协议下测量推理时间,在联发科天玑9500上为34.0毫秒,在高通骁龙8 Elite NPU上为46.1毫秒。我们进一步揭示了“推理反转”效应,即严格遵循NPU兼容操作可使专用NPU执行速度比集成移动GPU快3.88倍。1.96M参数的学生网络通过高alpha知识蒸馏(alpha=0.9)恢复了教师网络99.8%的恢复质量,实现了21.2倍的参数缩减,同时将PSNR差距从1.63 dB缩小至仅0.05 dB。这些结果确立了硬件感知蒸馏作为统一高保真去噪与跨多种移动NPU架构实际部署的有效策略。所提出的轻量级学生网络(LiteDenoiseNet)及其训练统计数据已在NN数据集中提供,网址为https://github.com/ABrain-One/NN-Dataset。

英文摘要:

While deep-learning-based image restoration has achieved unprecedented fidelity, deployment on mobile Neural Processing Units (NPUs) remains bottlenecked by operator incompatibility and memory-access overhead. We propose an NPU-aware hardware-algorithm co-design approach for real-world image denoising on mobile NPUs. Our approach employs a high-capacity teacher to supervise a lightweight student network specifically designed to leverage the tiled-memory architectures of modern mobile SoCs. By prioritizing NPU-native primitives -- standard 3x3 convolutions, ReLU activations, and nearest-neighbor upsampling -- and employing a progressive context expansion strategy (up to 1024x1024 crops), the model achieves 37.66 dB PSNR / 0.9278 SSIM on the validation benchmark and 37.58 dB PSNR / 0.9098 SSIM on the held-out test benchmark at full resolution (2432x3200) in the Mobile AI 2026 challenge. Following the official challenge rules, the inference runtime is measured under a standardized Full HD (1088x1920) protocol, where it runs in 34.0 ms on the MediaTek Dimensity 9500 and 46.1 ms on the Qualcomm Snapdragon 8 Elite NPU. We further reveal an "Inference Inversion" effect, where strict adherence to NPU-compatible operations enables dedicated NPU execution up to 3.88x faster than the integrated mobile GPU. The 1.96M-parameter student recovers 99.8% of the teacher's restoration quality via high-alpha knowledge distillation (alpha = 0.9), achieving a 21.2x parameter reduction while closing the PSNR gap from 1.63 dB to only 0.05 dB. These results establish hardware-aware distillation as an effective strategy for unifying high-fidelity denoising with practical deployment across diverse mobile NPU architectures. The proposed lightweight student model (LiteDenoiseNet) and its training statistics are provided in the NN Dataset, available at https://github.com/ABrain-One/NN-Dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑