arrow
返回

Execution-Delay-Balanced Pipeline-Parallelism-Based Distributed Model Training for Artificial Intelligence Data Centers Interconnected by Optical Networks

delete2026-06-17
delete0
PRE
AI
J
Jingjie Xin
李
李昕 (Xin Li)
D
Daniel C. Kilper
S
Shanguo Huang
DOI:10.1109/jiot.2026.3704665delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
利用管道并行技术实现人工智能数据中心(AIDCs)的分布式模型训练是可行的,其中基于Transformer的分布式模型训练(TDMT)任务被划分为多个顺序阶段并卸载到合适的AIDCs。然而,由于划分点数量和位置的不可预测性,确定合适的划分和卸载决策以满足延迟要求具有挑战性。本文提出了一种基于执行延迟平衡的管道并行分布式模型训练(EP-DMT)方案,该方案联合调度计算和网络资源,以确定TDMT任务的划分和卸载决策。其目标是最小化气泡时间,从而提高训练效率。开发了一种启发式算法,通过基于资源状态估计基本阶段数量和执行延迟来确定每个阶段。同时优化了AIDCs的选择以及各阶段的路由和波长分配(RWA)方案。仿真结果表明,与基准方法相比,EP-DMT具有更高的接受率和更低的气泡时间。
Keyword:
Artificial intelligence data centers (AIDCs)
distributed model training (DMT)
optical networks
pipeline parallelism
task offloading

期刊

IEEE Internet of Things Journal 封面图
IEEE Internet of Things Journal
IF:
8.9
论文数:
1.4W
被引数:
7.8W

机构

B
beijing university of posts and telecommunications
学者数:
2.3K
论文数: 861
被引数: 0
T
Trinity College Dublin
学者数:
2.4W
论文数: 1.9W
被引数: 2.7W
引用论文

引用论文

A Survey on Efficient Training of Transformers
err2023-08-01
err0
errOAAI
errBohan Zhuang; Jing Liu; Zizheng Pan; Haoyu He; Yuetian Weng; Chunhua Shen
err分享
err收藏
Merak: An Efficient Distributed DNN Training Framework With Automated 3D Parallelism for Giant Foundation Models
err2023-05-01
err18
errOAAI
errLai, Zhiquan; Li, Shengwei; Tang, Xudong; Ge, Keshi; Liu, Weijie; Duan, Yabo; Qiao, Linbo; Li, Dongsheng
err分享
err收藏
Nvidia Hopper GPU: Scaling Performance
err
IF0
err2022-08-21
err0
PREAI
errJack Choquette
err分享
err收藏
Optimizing Multi-GPU Parallelization Strategies for Deep Learning Training
err2019-09-01
err0
errOAAI
errSaptadeep Pal; Eiman Ebrahimi; Arslan Zulfiqar; Yaosheng Fu; Victor Zhang; Szymon Migacz; David Nellans; Puneet Gupta
err分享
err收藏
err分享
err收藏
err
IF0
err
err0
PREAI
err
err分享
err收藏
PyTorch distributed
err2020-09-14
err0
PREAI
errShen Li; Yanli Zhao; Rohan Varma; Omkar Salpekar; Pieter Noordhuis; Teng Li; Adam Paszke; Jeff Smith; Brian Vaughan; Pritam Damania; Soumith Chintala
err分享
err收藏
学者 查看更多内容