针对移动端图像增强模型训练与部署差异,提出高效适配方案。
Bridging the Training-Deployment Gap: Gated Encoding and Multi-Scale Refinement for Efficient Quantization-Aware Image Enhancement

- 采用分层网络+门控编码器保留细节特征
- 结合量化感知训练,使低精度部署后质量下降减少42%
- 适合资源受限的移动端图像增强应用
面向移动设备的图像增强模型常难以兼顾高输出质量和快速处理速度。尽管现有深度学习模型可将低质量手机照片提升为高质量图像,但在转换为低精度格式用于实际移动端部署时性能常显著下降。为解决这一训练-部署不匹配问题,本文提出一种专为移动端部署设计的高效图像增强模型。该方法采用分层网络架构,包含门控编码块和多尺度优化模块,以保留细粒度视觉特征。同时引入量化感知训练(QAT),在训练过程中模拟低精度表示的影响,使网络能够自适应,避免传统事后量化(PTQ)带来的典型质量损失。实验结果表明,所提方法在保持极低计算开销的同时,生成高保真视觉输出,适用于标准移动设备的实际部署。代码将开源于 https://github.com/GenAI4E/QATIE.git。
原文摘要 · Abstract (English)
Image enhancement models for mobile devices often struggle to balance high output quality with the fast processing speeds required by mobile hardware. While recent deep learning models can enhance low-quality mobile photos into high-quality images, their performance is often degraded when converted to lower-precision formats for actual use on mobile phones. To address this training-deployment mismatch, we propose an efficient image enhancement model designed specifically for mobile deployment. Our approach uses a hierarchical network architecture with gated encoder blocks and multiscale refinement to preserve fine-grained visual features. Moreover, we incorporate Quantization-Aware Training (QAT) to simulate the effects of low-precision representation during the training process. This allows the network to adapt and prevents the typical drop in quality seen with standard post-training quantization (PTQ). Experimental results demonstrate that the proposed method produces high-fidelity visual output while maintaining the low computational overhead needed for practical use on standard mobile devices. The code will be available at https://github.com/GenAI4E/QATIE.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。