arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.11700cs.CVeess.IV

率-失真-认知可控的多功能神经图像压缩

Rate-Distortion-Cognition Controllable Versatile Neural Image Compression

Ningbo Institute of Digital Twin, Eastern Institute of Technology(东方理工大学宁波数字孪生研究院);Shanghai Jiao Tong University(上海交通大学);University of Science and Technology of China(中国科学技术大学)

Jinming Liu, Ruoyu Feng, Yunpeng Qi, Qiuyu Chen, Zhibo Chen, Wenjun Zeng, Xin Jin

AI总结:

提出一种单模型可调码率、重建质量与机器任务精度的多功能图像压缩方法,通过认知损失与辅助残差分支及插值策略实现灵活控制。

摘要:

近年来,得益于基于学习的图像压缩与分析技术的快速发展,面向机器的图像编码(ICM)领域引起了更高的关注并取得了显著进展。以往的研究通常需要训练多个独立的编解码器以支持不同的码率级别、机器任务和网络,因此缺乏灵活性和实用性。为解决这些挑战,我们提出了一种率-失真-认知可控的多功能图像压缩方法,该方法允许用户通过单个神经模型调整码率(即率)、图像重建质量(即失真)和机器任务精度(即认知),实现超可控性。具体而言,我们首先在主压缩分支中引入一个面向认知的损失函数,以训练一个适用于多种机器任务的编解码器。该分支通过潜在码通道调节量化程度来实现可变码率。为了进一步提升重建图像的质量,我们采用一个辅助分支,通过可扩展比特流补充残差信息。最终,两个分支使用“βx + (1 - β)y”插值策略,以实现认知与失真之间的平衡权衡。大量实验表明,我们的方法在ICM性能上表现令人满意,并实现了灵活的率-失真-认知控制。

英文摘要:

Recently, the field of Image Coding for Machines (ICM) has garnered heightened interest and significant advances thanks to the rapid progress of learning-based techniques for image compression and analysis. Previous studies often require training separate codecs to support various bitrate levels, machine tasks, and networks, thus lacking both flexibility and practicality. To address these challenges, we propose a rate-distortion-cognition controllable versatile image compression, which method allows the users to adjust the bitrate (i.e., Rate), image reconstruction quality (i.e., Distortion), and machine task accuracy (i.e., Cognition) with a single neural model, achieving ultra-controllability. Specifically, we first introduce a cognition-oriented loss in the primary compression branch to train a codec for diverse machine tasks. This branch attains variable bitrate by regulating quantization degree through the latent code channels. To further enhance the quality of the reconstructed images, we employ an auxiliary branch to supplement residual information with a scalable bitstream. Ultimately, two branches use a `$βx + (1 - β) y$' interpolation strategy to achieve a balanced cognition-distortion trade-off. Extensive experiments demonstrate that our method yields satisfactory ICM performance and flexible Rate-Distortion-Cognition controlling.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑