Papers

GearUp:利用 DAG 空闲时间隐藏 Serverless 推理加载开销

GearUp: DAG-aware Acceleration for Serverless Inference Serving via Slack-guided Pre-loading

Why we picked it

一篇系统方向的论文:GearUp 把模型加载塞进 DAG 的空闲时段,延迟最高降到十分之一,内存还省九成,做推理服务的人可以看看。

论文提出 GearUp,一个面向 DAG 式 ML 推理工作流的 serverless 服务系统。它通过分析每个节点的加载阶段(库、模型权重、CUDA kernel),把初始化调度进 DAG 暴露的 slack 时间,并用流量自适应的 keep-alive 策略控制实例保留时长。在 AMBER Alert 和 FINRA-ML 工作流、NVIDIA L40S GPU 上,相对 SMIless 中位端到端延迟分别降低 64.5% 和 75.9%;相对 InstaInfer 的急切预加载,GPU 和主机内存分配最多下降 83.9% 和 94.6%。四节点上还实现 4.2 倍强扩展。

Original · arXiv cs.LG

GearUp: DAG-aware Acceleration for Serverless Inference Serving via Slack-guided Pre-loading

Serverless platforms increasingly host machine learning (ML) inference workflows, modeled as directed acyclic graphs (DAGs) of interdependent functions. Beyond container startup, each ML node must load libraries, model weights, and CUDA kernels before it can run. These costs accumulate along dependent DAG paths and place initialization on the end-to-end critical path. Existing systems optimize one half of this problem. DAG-agnostic pre-loading prepares nodes the request may never reach, wasting host and GPU memory. DAG-aware serving routes and scales whole functions, leaving the loading stages inside a node untouched. This paper presents GearUp, a serverless DAG inference serving system that removes ML-node loading from the end-to-end critical path at low resource cost. GearUp profiles the loading stages of each node and schedules them inside the slack time the DAG exposes, hiding initialization behind upstream execution. It reuses initialized instances within and across requests to avoid redundant cold starts. A probabilistic, traffic-adaptive keep-alive policy sets how long idle instances are retained, balancing memory cost against cold-start risk under bursty load. On the AMBER Alert and FINRA-ML workflows with NVIDIA L40S GPUs, GearUp reduces median end-to-end latency by 64.5% and 75.9% against SMIless, for median speedups of 3.1x and 10.0x over cold execution. Integrated GPU and host memory allocation fall by up to 83.9% and 94.6% against eager pre-loading in InstaInfer. GearUp achieves 4.2x strong scaling on four workers.