Inference Offloading for Cost-Sensitive Binary Classification at the EdgeShow others and affiliations
2026 (English)In: Fortieth AAAI Conference on Artificial Intelligence, Thirty-Eighth Conference on Innovative Applications of Artificial Intelligence, Sixteenth Symposium on Educational Advances in Artificial Intelligence, AAAI 2026, Association for the Advancement of Artificial Intelligence (AAAI) , 2026, p. 24449-24457Conference paper, Published paper (Refereed)
Abstract [en]
We investigate a binary classification problem in an edge intelligence system where false negatives are more costly than false positives. The system features a compact, locally deployed model, supplemented by a larger, remote model that is accessible via the network, albeit at an offloading cost. For each sample, our system first uses the locally deployed model for inference. Based on the output of the local model, the sample may be offloaded to the remote model. This work aims to understand the fundamental trade-off between classification accuracy and the offloading costs within such a hierarchical inference (HI) system. To optimise this system, we propose an online learning framework that continuously adapts a pair of thresholds on the local model’s confidence scores. These thresholds determine the prediction of the local model and whether a sample is classified locally or offloaded to the remote model. We present a closed-form solution for the setting where the local model is calibrated. For the more general case of uncalibrated models, we introduce H2T2, an online two-threshold hierarchical inference policy, and prove it achieves sublinear regret. H2T2 is model-agnostic, requires no training, and learns during the inference phase using limited feedback. Simulations on real-world datasets show that H2T2 consistently outperforms naive and single-threshold HI policies, sometimes even surpassing single-threshold offline optima. The policy also demonstrates robustness to distribution shifts and adapts effectively to mismatched classifiers.
Place, publisher, year, edition, pages
Association for the Advancement of Artificial Intelligence (AAAI) , 2026. p. 24449-24457
National Category
Computer Sciences Computer Systems
Identifiers
URN: urn:nbn:se:kth:diva-380161DOI: 10.1609/aaai.v40i29.39627Scopus ID: 2-s2.0-105034874054OAI: oai:DiVA.org:kth-380161DiVA, id: diva2:2055267
Conference
40th AAAI Conference on Artificial Intelligence, AAAI 2026, Singapore, Singapore, Jan 20 2026 - Jan 27 2026
Note
QC 20260423
2026-04-232026-04-232026-04-23Bibliographically approved