arXiv:2604.20291cs.CV2026-04中稿 · the Mobile AI

轻量级INT8图像超分模型,兼顾精度与移动端部署效率。

Efficient INT8 Single-Image Super-Resolution via Deployment-Aware Quantization and Teacher-Guided Training

论文配图:Efficient INT8 Single-Image Super-Resolution via Deployment-Aware Quantization and Teacher-Guided Training
图 1 · 摘自论文原文
  • 采用提取-精炼-上采样架构,低分辨率空间计算为主,轻量化结构设计
  • 三阶段训练提升质量:空间监督、DCT域约束与基于Mamba教师的蒸馏优化
  • 支持移动端INT8部署,动态与固定形状TFLite均达30.00 dB以上性能

高效单图超分辨率(SISR)需在重建保真度、模型紧凑性和低比特部署鲁棒性之间取得平衡,尤其对x3超分更具挑战。本文提出一种面向部署的量化SISR框架,基于提取-精炼-上采样设计,学生模型主要在低分辨率空间进行计算,采用轻量级可重参数化骨干网络和PixelShuffle重建,实现紧凑的推理图。为在不显著增加复杂度的前提下提升质量,采用三阶段训练流程:第一阶段通过空间监督学习基础重建映射;第二阶段使用Charbonnier损失、DCT域监督及来自Mamba教师的置信加权输出级蒸馏进行精度优化;第三阶段直接在融合后的部署图上进行量化感知训练。进一步引入权重裁剪和BatchNorm校准以增强量化稳定性。在MAI 2026量化4K图像超分挑战测试集上,最终提交的AIO MAI方案达到29.79 dB PSNR和0.8634 SSIM,目标移动端INT8部署下得分为1.8。消融实验表明,教师引导监督使动态INT8 TFLite重建性能从29.91 dB/0.853提升至30.0003 dB/0.856,固定形状可部署INT8 TFLite则达30.006 dB/0.857。

原文摘要 · Abstract (English)

Efficient single-image super-resolution (SISR) requires balancing reconstruction fidelity, model compactness, and robustness under low-bit deployment, which is especially challenging for x3 SR. We present a deployment-oriented quantized SISR framework based on an extract-refine-upsample design. The student performs most computation in the low-resolution space and uses a lightweight re-parameterizable backbone with PixelShuffle reconstruction, yielding a compact inference graph. To improve quality without significantly increasing complexity, we adopt a three-stage training pipeline: Stage 1 learns a basic reconstruction mapping with spatial supervision; Stage 2 refines fidelity using Charbonnier loss, DCT-domain supervision, and confidence-weighted output-level distillation from a Mamba-based teacher; and Stage 3 applies quantization-aware training directly on the fused deploy graph. We further use weight clipping and BatchNorm recalibration to improve quantization stability. On the MAI 2026 Quantized 4K Image Super-Resolution Challenge test set, our final AIO MAI submission achieves 29.79 dB PSNR and 0.8634 SSIM, obtaining a final score of 1.8 under the target mobile INT8 deployment setting. Ablation on Stage 3 optimization shows that teacher-guided supervision improves the dynamic INT8 TFLite reconstruction from 29.91 dB/0.853 to 30.0003 dB/0.856, while the fixed-shape deployable INT8 TFLite artifact attains 30.006 dB/0.857.

超分辨率INT8量化移动端部署蒸馏训练

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