Collaborative Processing for Multi-Tenant Inference on Memory-Constrained Edge TPUs

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Ng, Nathan, Hanafy, Walid A., Kadambi, Prashanthi, Sunil, Balachandra, Gupta, Ayush, Irwin, David, Simmhan, Yogesh, Shenoy, Prashant
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909034704011264
author Ng, Nathan
Hanafy, Walid A.
Kadambi, Prashanthi
Sunil, Balachandra
Gupta, Ayush
Irwin, David
Simmhan, Yogesh
Shenoy, Prashant
author_facet Ng, Nathan
Hanafy, Walid A.
Kadambi, Prashanthi
Sunil, Balachandra
Gupta, Ayush
Irwin, David
Simmhan, Yogesh
Shenoy, Prashant
contents IoT applications increasingly rely on on-device AI accelerators to ensure high performance, especially in low-connectivity and safety-critical scenarios. However, the limited on-chip memory of these accelerators forces inference runtimes to swap model segments between host and accelerator memory, incurring significant swapping overheads. While collaborative processing by partitioning model execution across CPU and accelerator resources can reduce accelerator memory pressure and execution overhead, naive partitioning may worsen end-to-end latency by either shifting excessive computation to the CPU or failing to sufficiently reduce swapping, a problem that is further exacerbated in multi-tenant and dynamic environments. To address these issues, we present SwapLess, a system for adaptive, multi-tenant TPU-CPU collaborative inference on memory-constrained Edge TPUs. SwapLess utilizes an analytic queueing model that captures partition-dependent CPU/TPU service times as well as inter- and intra-model swapping overheads across different workload mixes and request rates. Using this model, SwapLess continuously adjusts both the partition point and CPU core allocation online to minimize end-to-end response time with low decision overhead. An implementation on Edge TPU-equipped platforms demonstrates that SwapLess reduces mean latency by up to 63.8% for single-tenant workloads and up to 77.4% for multi-tenant workloads relative to the default Edge TPU compiler.
format Preprint
id arxiv_https___arxiv_org_abs_2602_17808
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Collaborative Processing for Multi-Tenant Inference on Memory-Constrained Edge TPUs
Ng, Nathan
Hanafy, Walid A.
Kadambi, Prashanthi
Sunil, Balachandra
Gupta, Ayush
Irwin, David
Simmhan, Yogesh
Shenoy, Prashant
Distributed, Parallel, and Cluster Computing
Performance
IoT applications increasingly rely on on-device AI accelerators to ensure high performance, especially in low-connectivity and safety-critical scenarios. However, the limited on-chip memory of these accelerators forces inference runtimes to swap model segments between host and accelerator memory, incurring significant swapping overheads. While collaborative processing by partitioning model execution across CPU and accelerator resources can reduce accelerator memory pressure and execution overhead, naive partitioning may worsen end-to-end latency by either shifting excessive computation to the CPU or failing to sufficiently reduce swapping, a problem that is further exacerbated in multi-tenant and dynamic environments. To address these issues, we present SwapLess, a system for adaptive, multi-tenant TPU-CPU collaborative inference on memory-constrained Edge TPUs. SwapLess utilizes an analytic queueing model that captures partition-dependent CPU/TPU service times as well as inter- and intra-model swapping overheads across different workload mixes and request rates. Using this model, SwapLess continuously adjusts both the partition point and CPU core allocation online to minimize end-to-end response time with low decision overhead. An implementation on Edge TPU-equipped platforms demonstrates that SwapLess reduces mean latency by up to 63.8% for single-tenant workloads and up to 77.4% for multi-tenant workloads relative to the default Edge TPU compiler.
title Collaborative Processing for Multi-Tenant Inference on Memory-Constrained Edge TPUs
topic Distributed, Parallel, and Cluster Computing
Performance
url https://arxiv.org/abs/2602.17808