arXiv:2506.15228eess.IVcs.MM2025-06被引 1

提出自适应贝叶斯网络结构学习框架,实现图像压缩全流程计算可扩展。

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

  • 用异构双部贝叶斯网控制主干网络计算量
  • 用同构多部贝叶斯网优化自回归单元处理效率
  • 动态调整结构适配设备、数据和任务需求

神经图像压缩(NIC)在率失真性能和多任务能力上取得突破,支持人眼感知与机器视觉。但其广泛应用受限于高计算开销。现有方法仅对特定模块优化或设定固定复杂度,缺乏全局计算复杂度控制。本文提出ABC(自适应贝叶斯网结构学习框架),通过贝叶斯网结构学习实现所有NIC组件的计算可扩展性。创新包括:(i) 异构双部贝叶斯网(跨节点结构)管理神经主干计算;(ii) 同构多部贝叶斯网(节点内结构)优化自回归单元处理;(iii) 自适应控制模块,根据设备能力、输入数据复杂度和下游任务需求动态调整结构。实验表明,ABC实现全链路计算可扩展,具备更优复杂度适应性与更广控制范围,同时保持竞争力压缩性能。该框架可集成至多种采用贝叶斯网表示的NIC架构中,为NIC应用提供稳健的计算可扩展解决方案。代码已开源。

原文摘要 · Abstract (English)

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.

图像压缩贝叶斯网络计算可扩展多任务

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。