Search papers, labs, and topics across Lattice.
This paper introduces HAMP-LIC, a Hessian-aware mixed-precision post-training quantization framework designed to enhance learned image compression (LIC) models by addressing their high computational demands and encoding-decoding mismatches on diverse hardware. By employing a four-stage optimization strategy that incorporates block-wise sensitivity estimation, task-aware refinement, and calibrated block-wise reconstruction, HAMP-LIC achieves significant model compression while maintaining minimal quality loss. Experimental results show that HAMP-LIC can compress models by up to 4.85x with only a 0.59% increase in BD-rate, outperforming existing quantization methods and eliminating cross-platform errors.
Achieving nearly 5x model compression with minimal quality loss, HAMP-LIC sets a new standard for efficient learned image compression across heterogeneous hardware.
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.
