TY - GEN
T1 - Tiny Generative Image Compression for Bandwidth-Constrained Sensor Applications
AU - Korber, Nikolai
AU - Siebert, Andreas
AU - Hauke, Sascha
AU - Mueller-Gritschneder, Daniel
N1 - Publisher Copyright:
© 2021 IEEE.
PY - 2021
Y1 - 2021
N2 - Deep image compression algorithms based on Generative Adversarial Networks (GANs) are a promising direction to address the strict communication bandwidth limitations commonly encountered in IoT sensor networks (e.g. Low Power Wide Area Networks). However, current methods do not consider that the sensor nodes, which perform the image encoding, usually only offer very limited computation and memory capabilities, e.g. a resource-constrained tiny device such as a micro-controller. In this paper, we propose the first tiny generative image compression method specifically designed for image compression on micro-controllers. We base our encoder on the well-known MobileNetV2 network architecture, while keeping the decoder side fixed. To cope with the resulting asymmetric design of the compression pipeline, we investigate the impact of different training strategies (end-to-end, knowledge distillation) and integer quantization techniques (post-training, quantization-aware training) on the GAN-training stability. On the Cityscapes dataset, we achieve a compression performance that is very close to the state-of-the-art, while requiring 99% less SRAM size, 97% smaller flash storage and 87% less multiply-add operations. Our findings suggest that tiny generative image compression is particularly well suited for application-specific domains.
AB - Deep image compression algorithms based on Generative Adversarial Networks (GANs) are a promising direction to address the strict communication bandwidth limitations commonly encountered in IoT sensor networks (e.g. Low Power Wide Area Networks). However, current methods do not consider that the sensor nodes, which perform the image encoding, usually only offer very limited computation and memory capabilities, e.g. a resource-constrained tiny device such as a micro-controller. In this paper, we propose the first tiny generative image compression method specifically designed for image compression on micro-controllers. We base our encoder on the well-known MobileNetV2 network architecture, while keeping the decoder side fixed. To cope with the resulting asymmetric design of the compression pipeline, we investigate the impact of different training strategies (end-to-end, knowledge distillation) and integer quantization techniques (post-training, quantization-aware training) on the GAN-training stability. On the Cityscapes dataset, we achieve a compression performance that is very close to the state-of-the-art, while requiring 99% less SRAM size, 97% smaller flash storage and 87% less multiply-add operations. Our findings suggest that tiny generative image compression is particularly well suited for application-specific domains.
KW - Auto-encoder
KW - Generative adversarial network
KW - Image compression
KW - Neural network
KW - TinyML
UR - http://www.scopus.com/inward/record.url?scp=85125876313&partnerID=8YFLogxK
U2 - 10.1109/ICMLA52953.2021.00094
DO - 10.1109/ICMLA52953.2021.00094
M3 - Conference contribution
AN - SCOPUS:85125876313
T3 - Proceedings - 20th IEEE International Conference on Machine Learning and Applications, ICMLA 2021
SP - 564
EP - 569
BT - Proceedings - 20th IEEE International Conference on Machine Learning and Applications, ICMLA 2021
A2 - Wani, M. Arif
A2 - Sethi, Ishwar K.
A2 - Shi, Weisong
A2 - Qu, Guangzhi
A2 - Raicu, Daniela Stan
A2 - Jin, Ruoming
PB - Institute of Electrical and Electronics Engineers Inc.
T2 - 20th IEEE International Conference on Machine Learning and Applications, ICMLA 2021
Y2 - 13 December 2021 through 16 December 2021
ER -