kth.sePublications KTH
Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
Inference Offloading for Cost-Sensitive Binary Classification at the Edge
KTH, School of Electrical Engineering and Computer Science (EECS), Information Science and Engineering.ORCID iD: 0000-0002-2739-5060
Electrical Engineering, Indian Institute of Technology Bombay, Mumbai, India.
Electrical Engineering, Indian Institute of Technology Bombay, Mumbai, India.
KTH, School of Electrical Engineering and Computer Science (EECS), Information Science and Engineering.ORCID iD: 0000-0001-6682-6559
Show 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

Available from: 2026-04-23 Created: 2026-04-23 Last updated: 2026-04-23Bibliographically approved

Open Access in DiVA

No full text in DiVA

Other links

Publisher's full textScopus

Authority records

Moothedath, Vishnu NarayananGross, James

Search in DiVA

By author/editor
Moothedath, Vishnu NarayananGross, James
By organisation
Information Science and Engineering
Computer SciencesComputer Systems

Search outside of DiVA

GoogleGoogle Scholar

doi
urn-nbn

Altmetric score

doi
urn-nbn
Total: 15 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf