arXiv:2509.10366cs.CV2025-09被引 2

用知识蒸馏让图像压缩模型更小更快,适合手机等设备使用。

Efficient Learned Image Compression Through Knowledge Distillation

  • 用大模型指导小模型训练,提升小模型压缩性能。
  • 在不同画质与码率下,小模型性能接近大模型。
  • 显著降低计算和能耗,适合移动端部署。

学习型图像压缩结合机器学习与图像处理,近年来基于神经网络的方法表现优于传统编码器。其流程包括:编码器将图像映射到低维隐空间,经量化和熵编码生成二进制比特流,解码端反向重建图像。然而,这些模型需大量算力,难以在资源受限设备上实时运行。本文通过知识蒸馏技术,使小型神经网络在大模型指导下训练,显著降低资源消耗。实验表明该方法在多种模型规模、不同画质/码率权衡下均有效,且能节省处理与能耗。研究还提出新训练设置与超参数,未来可探索不同教师模型及损失函数,也可扩展至基于Transformer的模型。代码已公开:https://github.com/FABallemand/PRIM。

原文摘要 · Abstract (English)

Learned image compression sits at the intersection of machine learning and image processing. With advances in deep learning, neural network-based compression methods have emerged. In this process, an encoder maps the image to a low-dimensional latent space, which is then quantized, entropy-coded into a binary bitstream, and transmitted to the receiver. At the receiver end, the bitstream is entropy-decoded, and a decoder reconstructs an approximation of the original image. Recent research suggests that these models consistently outperform conventional codecs. However, they require significant processing power, making them unsuitable for real-time use on resource-constrained platforms, which hinders their deployment in mainstream applications. This study aims to reduce the resource requirements of neural networks used for image compression by leveraging knowledge distillation, a training paradigm where smaller neural networks, partially trained on the outputs of larger, more complex models, can achieve better performance than when trained independently. Our work demonstrates that knowledge distillation can be effectively applied to image compression tasks: i) across various architecture sizes, ii) to achieve different image quality/bit rate tradeoffs, and iii) to save processing and energy resources. This approach introduces new settings and hyperparameters, and future research could explore the impact of different teacher models, as well as alternative loss functions. Knowledge distillation could also be extended to transformer-based models. The code is publicly available at: https://github.com/FABallemand/PRIM .

图像压缩知识蒸馏轻量模型

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