ABC: Adaptive BayesNet Structure Learning for Computational Scalable Multi-task Image Compression

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Yufeng, Dai, Wenrui, Yu, Hang, Liu, Shizhan, Hou, Junhui, Li, Jianguo, Lin, Weiyao
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916799214256128
author Zhang, Yufeng
Dai, Wenrui
Yu, Hang
Liu, Shizhan
Hou, Junhui
Li, Jianguo
Lin, Weiyao
author_facet Zhang, Yufeng
Dai, Wenrui
Yu, Hang
Liu, Shizhan
Hou, Junhui
Li, Jianguo
Lin, Weiyao
contents Neural Image Compression (NIC) has revolutionized image compression with its superior rate-distortion performance and multi-task capabilities, supporting both human visual perception and machine vision tasks. However, its widespread adoption is hindered by substantial computational demands. While existing approaches attempt to address this challenge through module-specific optimizations or pre-defined complexity levels, they lack comprehensive control over computational complexity. We present ABC (Adaptive BayesNet structure learning for computational scalable multi-task image Compression), a novel, comprehensive framework that achieves computational scalability across all NIC components through Bayesian network (BayesNet) structure learning. ABC introduces three key innovations: (i) a heterogeneous bipartite BayesNet (inter-node structure) for managing neural backbone computations; (ii) a homogeneous multipartite BayesNet (intra-node structure) for optimizing autoregressive unit processing; and (iii) an adaptive control module that dynamically adjusts the BayesNet structure based on device capabilities, input data complexity, and downstream task requirements. Experiments demonstrate that ABC enables full computational scalability with better complexity adaptivity and broader complexity control span, while maintaining competitive compression performance. Furthermore, the framework's versatility allows integration with various NIC architectures that employ BayesNet representations, making it a robust solution for ensuring computational scalability in NIC applications. Code is available in https://github.com/worldlife123/cbench_BaSIC.
format Preprint
id arxiv_https___arxiv_org_abs_2506_15228
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle ABC: Adaptive BayesNet Structure Learning for Computational Scalable Multi-task Image Compression
Zhang, Yufeng
Dai, Wenrui
Yu, Hang
Liu, Shizhan
Hou, Junhui
Li, Jianguo
Lin, Weiyao
Image and Video Processing
Multimedia
Neural Image Compression (NIC) has revolutionized image compression with its superior rate-distortion performance and multi-task capabilities, supporting both human visual perception and machine vision tasks. However, its widespread adoption is hindered by substantial computational demands. While existing approaches attempt to address this challenge through module-specific optimizations or pre-defined complexity levels, they lack comprehensive control over computational complexity. We present ABC (Adaptive BayesNet structure learning for computational scalable multi-task image Compression), a novel, comprehensive framework that achieves computational scalability across all NIC components through Bayesian network (BayesNet) structure learning. ABC introduces three key innovations: (i) a heterogeneous bipartite BayesNet (inter-node structure) for managing neural backbone computations; (ii) a homogeneous multipartite BayesNet (intra-node structure) for optimizing autoregressive unit processing; and (iii) an adaptive control module that dynamically adjusts the BayesNet structure based on device capabilities, input data complexity, and downstream task requirements. Experiments demonstrate that ABC enables full computational scalability with better complexity adaptivity and broader complexity control span, while maintaining competitive compression performance. Furthermore, the framework's versatility allows integration with various NIC architectures that employ BayesNet representations, making it a robust solution for ensuring computational scalability in NIC applications. Code is available in https://github.com/worldlife123/cbench_BaSIC.
title ABC: Adaptive BayesNet Structure Learning for Computational Scalable Multi-task Image Compression
topic Image and Video Processing
Multimedia
url https://arxiv.org/abs/2506.15228