HAMP-LIC给图像压缩模型做混合精度量化,模型省4.85倍,BD-rate只掉0.59%,还消除跨平台不一致。值得一看。
HAMP-LIC提出一种面向学习图像压缩模型的Hessian感知混合精度后训练量化框架。该方法通过Hessian迹估计逐块敏感性,并联合量化失真与率失真性能进行任务感知修正。在全局模型大小约束下分配位宽,再用小型校准集逐块重建以抑制量化误差。在Minnen2018和Cheng2020上,该方法实现最高4.85倍模型压缩,BD-rate损失仅0.59%。相比现有固定与混合精度PTQ方法,HAMP-LIC在多个数据集上持续领先,并消除跨平台编码解码误差。
HAMP-LIC: Hessian-Aware Mixed-Precision Post-Training Quantization for Learned Image Compression
Use this plain-text version for the arXiv abstract field: Learned image compression (LIC) models achieve strong rate-distortion performance but are hindered by high computational complexity and encoding-decoding mismatches across heterogeneous hardware platforms. Uniform fixed-precision quantization alleviates these issues but suffers severe quality degradation at low bit widths because it ignores differences in the quantization sensitivities of individual layers. To enable efficient and accurate low-bit deployment of pretrained LIC models, we propose HAMP-LIC, a Hessian-aware mixed-precision post-training quantization (PTQ) framework with a four-stage optimization strategy. First, block-wise sensitivity is estimated from the Hessian trace to capture second-order importance. Second, a task-aware refinement module adjusts these sensitivities by jointly considering quantization distortion and rate-distortion performance. Third, guided by the refined sensitivity profile, bit widths are allocated under a global model-size constraint to balance efficiency and reconstruction quality. Finally, block-wise reconstruction using a small calibration set further suppresses quantization error. Experiments on representative LIC models, including Minnen2018 and Cheng2020, demonstrate that HAMP-LIC achieves up to 4.85x model compression with as little as 0.59% BD-rate loss. It consistently outperforms existing fixed- and mixed-precision PTQ methods across multiple datasets while completely eliminating cross-platform encoding-decoding errors.