CI-ICM: Channel Importance-driven Learned Image Coding for Machines

TL;DR

This paper introduces a novel image coding method optimized for machine vision tasks, using channel importance and adaptive bitrate allocation to improve performance on object detection and segmentation.

Contribution

The paper proposes a new channel importance-driven image coding framework with modules for channel ranking, grouping, bit allocation, and task adaptation, tailored for machine vision.

Findings

Achieves 16.25% BD-mAP gain in object detection over baseline

Improves instance segmentation BD-mAP by 13.72%

Validates effectiveness through ablation studies and complexity analysis

Abstract

Traditional human vision-centric image compression methods are suboptimal for machine vision centric compression due to different visual properties and feature characteristics. To address this problem, we propose a Channel Importance-driven learned Image Coding for Machines (CI-ICM), aiming to maximize the performance of machine vision tasks at a given bitrate constraint. First, we propose a Channel Importance Generation (CIG) module to quantify channel importance in machine vision and develop a channel order loss to rank channels in descending order. Second, to properly allocate bitrate among feature channels, we propose a Feature Channel Grouping and Scaling (FCGS) module that non-uniformly groups the feature channels based on their importance and adjusts the dynamic range of each group. Based on FCGS, we further propose a Channel Importance-based Context (CI-CTX) module to allocate bits among feature groups and to preserve higher fidelity in critical channels. Third, to adapt to multiple machine tasks, we propose a Task-Specific Channel Adaptation (TSCA) module to adaptively enhance features for multiple downstream machine tasks. Experimental results on the COCO2017 dataset show that the proposed CI-ICM achieves BD-mAP@50:95 gains of 16.25$\%$ in object detection and 13.72$\%$ in instance segmentation over the established baseline codec. Ablation studies validate the effectiveness of each contribution, and computation complexity analysis reveals the practicability of the CI-ICM. This work establishes feature channel optimization for machine vision-centric compression, bridging the gap between image coding and machine perception.