Saved in:
Bibliographic Details
Main Authors: Cha, Seohyeon, Chan, Kevin, de Veciana, Gustavo, Vikalo, Haris
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2508.13380
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912543568560128
author Cha, Seohyeon
Chan, Kevin
de Veciana, Gustavo
Vikalo, Haris
author_facet Cha, Seohyeon
Chan, Kevin
de Veciana, Gustavo
Vikalo, Haris
contents The growing demand for intelligent services on resource-constrained edge devices has spurred the development of collaborative inference systems that distribute workloads across end devices, edge servers, and the cloud. While most existing frameworks focus on single-task, single-model scenarios, many real-world applications (e.g., autonomous driving and augmented reality) require concurrent execution of diverse tasks including detection, segmentation, and depth estimation. In this work, we propose a unified framework to jointly decide which multi-task models to deploy (onload) at clients and edge servers, and how to route queries across the hierarchy (offload) to maximize overall inference accuracy under memory, compute, and communication constraints. We formulate this as a mixed-integer program and introduce J3O (Joint Optimization of Onloading and Offloading), an alternating algorithm that (i) greedily selects models to onload via Lagrangian-relaxed submodular optimization and (ii) determines optimal offloading via constrained linear programming. We further extend J3O to account for batching at the edge, maintaining scalability under heterogeneous task loads. Experiments show J3O consistently achieves over $97\%$ of the optimal accuracy while incurring less than $15\%$ of the runtime required by the optimal solver across multi-task benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2508_13380
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Batching-Aware Joint Model Onloading and Offloading for Hierarchical Multi-Task Inference
Cha, Seohyeon
Chan, Kevin
de Veciana, Gustavo
Vikalo, Haris
Machine Learning
The growing demand for intelligent services on resource-constrained edge devices has spurred the development of collaborative inference systems that distribute workloads across end devices, edge servers, and the cloud. While most existing frameworks focus on single-task, single-model scenarios, many real-world applications (e.g., autonomous driving and augmented reality) require concurrent execution of diverse tasks including detection, segmentation, and depth estimation. In this work, we propose a unified framework to jointly decide which multi-task models to deploy (onload) at clients and edge servers, and how to route queries across the hierarchy (offload) to maximize overall inference accuracy under memory, compute, and communication constraints. We formulate this as a mixed-integer program and introduce J3O (Joint Optimization of Onloading and Offloading), an alternating algorithm that (i) greedily selects models to onload via Lagrangian-relaxed submodular optimization and (ii) determines optimal offloading via constrained linear programming. We further extend J3O to account for batching at the edge, maintaining scalability under heterogeneous task loads. Experiments show J3O consistently achieves over $97\%$ of the optimal accuracy while incurring less than $15\%$ of the runtime required by the optimal solver across multi-task benchmarks.
title Batching-Aware Joint Model Onloading and Offloading for Hierarchical Multi-Task Inference
topic Machine Learning
url https://arxiv.org/abs/2508.13380