Pareto-Optimal Bit Allocation for Collaborative Intelligence
Abstract
In recent studies, collaborative intelligence (CI) has emerged as a promising framework for deployment of Artificial Intelligence (AI)-based services on mobile/edge devices. In CI, the AI model (a deep neural network) is split between the edge and the cloud, and intermediate features are sent from the edge sub-model to the cloud sub-model. In this paper, we study bit allocation for feature coding in multi-stream CI systems. We model task distortion as a function of rate using convex surfaces similar to those found in distortion-rate theory. Using such models, we are able to provide closed-form bit allocation solutions for single-task systems and scalarized multi-task systems. Moreover, we provide analytical characterization of the full Pareto set for 2-stream k-task systems, and bounds on the Pareto set for 3-stream 2-task systems. Analytical results are examined on a variety of DNN models from the literature to demonstrate wide applicability of the results
- Publication:
-
IEEE Transactions on Image Processing
- Pub Date:
- 2021
- DOI:
- 10.1109/TIP.2021.3060875
- arXiv:
- arXiv:2009.12430
- Bibcode:
- 2021ITIP...30.3348A
- Keywords:
-
- Electrical Engineering and Systems Science - Image and Video Processing;
- Computer Science - Artificial Intelligence;
- Computer Science - Machine Learning
- E-Print:
- IEEE Trans. Image Processing, vol. 30, pp. 3348-3361, Feb. 2021