arXiv:2601.09823cs.CV2026-01被引 3

轻量级扩散模型NanoSD实现移动端实时图像修复。

NanoSD: Edge Efficient Foundation Model for Real Time Image Restoration

  • 通过网络手术与结构化设计,全链路优化去噪U-Net和VAE。
  • 参数量130M–315M,移动端推理最快仅20ms,支持实时处理。
  • 适合需要低延迟部署的边缘视觉任务开发者使用。

基于Stable Diffusion 1.5,我们提出NanoSD,一系列通过网络手术、特征级生成蒸馏与结构化架构缩放联合优化的轻量级扩散基础模型。该方法对U-Net与VAE编码器-解码器进行全链路协同设计,在保留生成先验的同时,使模型在准确率-延迟-大小权衡曲线上占据多个帕累托最优点(参数量130M–315M),在移动级NPU上实现最低20ms的实时推理。实验表明,参数量减少并不直接带来硬件效率提升;架构平衡、特征路由与潜在空间保持共同决定实际部署延迟。作为即插即用主干网络,NanoSD在图像超分、去模糊、人脸修复与单目深度估计任务中均达到领先性能,优于现有轻量化扩散模型,在感知质量与可部署性上表现更优。该工作为边缘设备上的实时视觉生成与修复提供了通用解决方案。

原文摘要 · Abstract (English)

Latent diffusion models such as Stable Diffusion 1.5 offer strong generative priors that are highly valuable for image restoration, yet their full pipelines remain too computationally heavy for deployment on edge devices. Existing lightweight variants predominantly compress the denoising U-Net or reduce the diffusion trajectory, which disrupts the underlying latent manifold and limits generalization beyond a single task. We introduce NanoSD, a family of Pareto-optimal diffusion foundation models distilled from Stable Diffusion 1.5 through network surgery, feature-wise generative distillation, and structured architectural scaling jointly applied to the U-Net and the VAE encoder-decoder. This full-pipeline co-design preserves the generative prior while producing models that occupy distinct operating points along the accuracy-latency-size frontier (e.g., 130M-315M parameters, achieving real-time inference down to 20ms on mobile-class NPUs). We show that parameter reduction alone does not correlate with hardware efficiency, and we provide an analysis revealing how architectural balance, feature routing, and latent-space preservation jointly shape true on-device latency. When used as a drop-in backbone, NanoSD enables state-of-the-art performance across image super-resolution, image deblurring, face restoration, and monocular depth estimation, outperforming prior lightweight diffusion models in both perceptual quality and practical deployability. NanoSD establishes a general-purpose diffusion foundation model family suitable for real-time visual generation and restoration on edge devices.

扩散模型边缘计算图像修复轻量化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。