As large language models (LLMs) scale to unprecedented sizes, traditional distributed training in homogeneous cluster environments increasingly falls short due to resource constraints and prohibitive computational costs. The integration of cross-regional distributed computing infrastructures for distributed training has become a current research focus. However, existing distributed training methods in heterogeneous environments mainly focus on a single heterogeneous feature, making it challenging to adapt effectively to a dual heterogeneous environment where computing and communication are complexly intertwined. Challenges such as imbalanced training costs and sensitivity to communication delays have significantly hindered the efficiency of distributed training in heterogeneous environments. To address these challenges, we propose a stable hybrid parallel distributed training architecture in dual-heterogeneous environments, SHPTA, which integrates heterogeneous devices across multiple regions to ensure stable and efficient distributed training of LLMs. From a global perspective, we introduce a HPO algorithm that comprehensively considers factors such as different bandwidths, delays, device performance, and quantity, accurately evaluates the parallel cost of the pipeline stage, and determines the optimal training device combination. Following this, the HAS mechanism is introduced from a fine-grained perspective during the training process. By dynamically reconstructing the communication topology structure, the optimal data parallel communication group is quickly formed, effectively reducing the synchronization overhead caused by device delays. Experimental results demonstrate that, compared to existing state-of-the-art methods, our method SHPTA reduces training time by 17.4% and achieves throughput increase of 26.8%.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

SHPTA: Stable Hybrid Parallel Distributed Training Architecture in Dual-Heterogeneous Environments

  • Yuxin Wang,
  • Chuantao Li,
  • Chunxiao Wang,
  • Fulai Liu,
  • Zhigang Zhao,
  • Jintao Li,
  • Guangdong Zhang

摘要

As large language models (LLMs) scale to unprecedented sizes, traditional distributed training in homogeneous cluster environments increasingly falls short due to resource constraints and prohibitive computational costs. The integration of cross-regional distributed computing infrastructures for distributed training has become a current research focus. However, existing distributed training methods in heterogeneous environments mainly focus on a single heterogeneous feature, making it challenging to adapt effectively to a dual heterogeneous environment where computing and communication are complexly intertwined. Challenges such as imbalanced training costs and sensitivity to communication delays have significantly hindered the efficiency of distributed training in heterogeneous environments. To address these challenges, we propose a stable hybrid parallel distributed training architecture in dual-heterogeneous environments, SHPTA, which integrates heterogeneous devices across multiple regions to ensure stable and efficient distributed training of LLMs. From a global perspective, we introduce a HPO algorithm that comprehensively considers factors such as different bandwidths, delays, device performance, and quantity, accurately evaluates the parallel cost of the pipeline stage, and determines the optimal training device combination. Following this, the HAS mechanism is introduced from a fine-grained perspective during the training process. By dynamically reconstructing the communication topology structure, the optimal data parallel communication group is quickly formed, effectively reducing the synchronization overhead caused by device delays. Experimental results demonstrate that, compared to existing state-of-the-art methods, our method SHPTA reduces training time by 17.4% and achieves throughput increase of 26.8%.