The Case for Hierarchical Deep Learning Inference at the Network Edge

Al-Atat, Ghina; Fresa, Andrea; Behera, Adarsh Prasad; Moothedath, Vishnu Narayanan; Gross, James; Champati, Jaya Prakash

doi:10.1145/3597062.3597278

Cited by 5 publications

(2 citation statements)

References 11 publications

Supporting

Mentioning

Contrasting

Order By: Relevance

“…In our recent work [16], we have provided a more general definition of HI and provided multiple use cases, and also compared HI with existing DL inference approaches at the edge. However, in [16], we used a fixed threshold for offloading without the online learning aspects that we do in this work. Further, unlike [16], in this work we rely heavily on analytical results and discuss how close the solution is to an offline optimum.…”

mentioning

confidence: 99%

See 1 more Smart Citation

Getting the Best Out of Both Worlds: Algorithms for Hierarchical Inference at the Edge

Moothedath,

Champati,

Gross

2024

Trans. Mach. Learn. Comm. Netw.

Self Cite

View full text Add to dashboard Cite

We consider a resource-constrained Edge Device (ED), such as an IoT sensor or a microcontroller unit, embedded with a small-size ML model (S-ML) for a generic classification application and an Edge Server (ES) that hosts a large-size ML model (L-ML). Since the inference accuracy of S-ML is lower than that of the L-ML, offloading all the data samples to the ES results in high inference accuracy, but it defeats the purpose of embedding S-ML on the ED and deprives the benefits of reduced latency, bandwidth savings, and energy efficiency of doing local inference. In order to get the best out of both worlds, i.e., the benefits of doing inference on the ED and the benefits of doing inference on ES, we explore the idea of Hierarchical Inference (HI), wherein S-ML inference is only accepted when it is correct, otherwise the data sample is offloaded for L-ML inference. However, the ideal implementation of HI is infeasible as the correctness of the S-ML inference is not known to the ED. We thus propose an online meta-learning framework that the ED can use to predict the correctness of the S-ML inference. In particular, we propose to use the probability corresponding to the maximum probability class output by S-ML for a data sample and decide whether to offload it or not. The resulting online learning problem turns out to be a Prediction with Expert Advice (PEA) problem with continuous expert space. For a full feedback scenario, where the ED receives feedback on the correctness of the S-ML once it accepts the inference, we propose the HIL-F algorithm and prove a sublinear regret bound n ln(1/λ min )/2 without any assumption on the smoothness of the loss function, where n is the number of data samples and λ min is the minimum difference between any two distinct maximum probability values across the data samples. For a no-local feedback scenario, where the ED does not receive the ground truth for the classification, we propose the HIL-N algorithm and prove that it has O n 2/3 ln 1/3 (1/λ min ) regret bound. We evaluate and benchmark the performance of the proposed algorithms for image classification application using four datasets, namely, Imagenette and Imagewoof [1], MNIST [2], and CIFAR-10 [3]. INDEX TERMS Hierarchical inference, edge computing, regret bound, continuous experts I. Introduction

show abstract

mentioning

confidence: 99%

“…However, in [16], we used a fixed threshold for offloading without the online learning aspects that we do in this work. Further, unlike [16], in this work we rely heavily on analytical results and discuss how close the solution is to an offline optimum.…”

mentioning

confidence: 99%