arrow
返回

swPredictor: A data-driven performance model for distributed data parallelism training on large-scale HPC clusters

delete2025-11-01
delete0
PRE
AI
X
Xianyu Zhu
R
Ruohan Wu
J
Junshi Chen *
安
安虹 (Hong An)
DOI:10.1016/j.peva.2025.102530delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
鉴于异构架构和多节点协作的复杂性,大规模HPC(高性能计算)集群在分布式数据并行(DDP)训练的资源利用和性能优化方面面临挑战。性能建模旨在识别应用瓶颈并指导算法设计,但现有性能模型很少考虑系统架构对通信性能的影响,或提供分布式训练的系统化分析。为解决这些问题,本文提出swPredictor,一种面向DDP训练性能精确预测的数据驱动性能模型。首先,基于运行时各种通信模式开发原始性能数据集,以避免系统误差。随后,提出一种新型多分支模块FNO-Inception,结合FNO(傅里叶神经网络算子)层与Inception结构,以同时利用多种频率特征。最后,通过引入FNO-Inception模块,构建新型回归模型FI-Net,以拟合复杂非线性关系。实验结果表明,FI-Net能够准确预测Sunway OceanLight超级计算机上DDP训练的性能,总体MAPE为0.93%,优于其他基准模型。
Keyword:
High-performance computing
Performance modeling
Deep learning
Distributed training

期刊

P
Performance Evaluation
IF:
0.8
论文数:
38
被引数:
851

机构

C
chinese academy of sciences
学者数:
56.7W
论文数: 45.0W
被引数: 704
引用论文

引用论文

BaGuaLu
err2022-03-28
err0
errOAAI
errZixuan Ma; Jiaao He; Jiezhong Qiu; Huanqi Cao; Yuanwei Wang; Zhenbo Sun; Liyan Zheng; Haojie Wang; Shizhi Tang; Tianyu Zheng; Junyang Lin; Guanyu Feng; Zeqiang Huang; Jie Gao; Aohan Zeng; Jianwei Zhang; Runxin Zhong; Tianhui Shi; Sha Liu; Weimin Zheng; Jie Tang; Hongxia Yang; Xin Liu; Jidong Zhai; Wenguang Chen
err分享
err收藏
FBNet: Hardware-Aware Efficient ConvNet Design via Differentiable Neural Architecture Search
err2019-06-01
err0
errOAAI
errBichen Wu; Kurt Keutzer; Xiaoliang Dai; Peizhao Zhang; Yanghan Wang; Fei Sun; Yiming Wu; Yuandong Tian; Peter Vajda; Yangqing Jia
err分享
err收藏
LogP: towards a realistic model of parallel computation
err1993-01-01
err0
PREAI
errDavid Culler; Richard Karp; David Patterson; Abhijit Sahay; Klaus Erik Schauser; Eunice Santos; Ramesh Subramonian; Thorsten von Eicken
err分享
err收藏
Nyquist-Shannon sampling theorem applied to refinements of the atomic pair distribution function
err2011-10-18
err78
errOAAI
errFarrow, Christopher L.; Shaw, Margaret; Kim, Hyunjeong; Juhas, Pavol; Billinge, Simon J. L.
err分享
err收藏
Establishing high performance AI ecosystem on Sunway platform
err2021-09-28
err0
PREAI
errSha Liu; Jie Gao; Xin Liu; Zeqiang Huang; Tianyu Zheng
err分享
err收藏
err分享
err收藏
err分享
err收藏
学者 查看更多内容