paperbot · PL 论文追踪

全部论文 待读 收藏

P2 TetriServe: Efficiently Serving Mixed DiT Workloads

TetriServe:高效服务混合 DiT 工作负载
ASPLOS Proceedings of the 31st ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 22026匹配度 2/10 · 论文研究 DiT 模型服务中的序列并行与调度优化,属于机器学习系统/并发调度方向,但未涉及静态分析、程序语言理论、浮点程序分析、贝叶斯程序分析或 LLM/Agent 与程序分析的交叉。它仅与“机器学习系统”和“并发、并行”标签有边缘关联,与用户核心研究画像匹配度低。引用 4 inbox
Runyu Lu, Shiqi He, Wenxuan Tan, Shenggui Li, Ruofan Wu, Jeff J. Ma, Ang Chen, Mosharaf Chowdhury
crossref: 2026-09-12dblp: 2026-09-12arxiv: 2026-09-12

AI 速览

一句话:提出步级序列并行与轮次调度,提升混合 DiT 负载的 SLO 达成率与 GPU 利用率。

扩散 Transformer(DiT)通过迭代去噪生成高质量图像,但在严格 SLO 下服务计算开销大,尤其高分辨率场景。现有系统采用固定度序列并行,难以应对混合分辨率与截止时间的异构负载,导致 GPU 利用率低、SLO 达成率差。TetriServe 提出步级序列并行,按请求截止时间动态调整并行度;并设计轮次调度机制,将时间离散为固定轮次,使截止时间感知调度可行,同时在步级别适配并行度、最小化 GPU 小时消耗,并联合打包请求以减少延迟完成。在先进 DiT 模型上评估显示,SLO 达成率最高提升 32%,且图像质量不下降。

问题
DiT 服务在严格 SLO 下因高计算成本与混合分辨率、截止时间的异构负载难以高效调度,固定序列并行导致 GPU 利用率与 SLO 达成率低。
方法
提出步级序列并行和轮次调度,按截止时间动态调整每请求并行度,并联合打包请求以最小化延迟完成。
评估
在先进 DiT 模型上对比现有方案,SLO 达成率最高提升 32%,且图像质量未下降。
原文摘要(Abstract)

Diffusion Transformer (DiT) models excel at generating high-quality images through iterative denoising steps, but serving them under strict Service Level Objectives (SLOs) is challenging due to their high computational cost, particularly at larger resolutions. Existing serving systems use fixed-degree sequence parallelism, which is inefficient for heterogeneous workloads with mixed resolutions and deadlines, leading to poor GPU utilization and low SLO attainment. In this paper, we propose step-level sequence parallelism to dynamically adjust the degree of parallelism of individual requests according to their deadlines. We present TetriServe1, a DiT serving system that implements this strategy for highly efficient image generation. Specifically, TetriServe introduces a novel round-based scheduling mechanism that improves SLO attainment by (1) discretizing time into fixed rounds to make deadline-aware scheduling tractable, (2) adapting parallelism at the step level and minimizing GPU hour consumption, and (3) jointly packing requests to minimize late completions. Extensive evaluation on state-of-the-art DiT models shows that TetriServe achieves up to 32% higher SLO attainment compared to existing solutions without degrading image quality. TetriServe is available at https://github.com/DiT-Serving/TetriServe.

链接与引用

DOI 原文 · arXiv · PDF(开放获取) · DBLP

BibTeX
@inproceedings{LuHTLWM0C26,
  title = {TetriServe: Efficiently Serving Mixed DiT Workloads},
  author = {Runyu Lu and Shiqi He and Wenxuan Tan and Shenggui Li and Ruofan Wu and Jeff J. Ma and Ang Chen and Mosharaf Chowdhury},
  booktitle = {ASPLOS},
  year = {2026},
  doi = {10.1145/3779212.3790233}
}