Collective Communication Profiling of Modern-day Machine Learning Workloads

📅 2025-07-03
📈 Citations: 0
Influential: 0
📄 PDF

career value

268K/year
🤖 AI Summary
In large-scale distributed ML training, collective communication primitives (e.g., AllReduce, AllGather, Broadcast) generate high-bandwidth, bursty traffic, causing network congestion and packet loss. Method: This work systematically characterizes the communication behavior of mainstream LLMs—including DeepSeek-V3, GPT, and Llama—under diverse parallelism strategies, scales, and network topologies. Leveraging fine-grained empirical analysis of NVIDIA NCCL logs, we quantitatively identify how operation type, message size, and request distribution affect network anomalies. We propose a resource-coordinated optimization framework tailored to LLM communication patterns, jointly optimizing collective primitive scheduling and network topology adaptation. Contribution/Results: Experiments demonstrate that our approach significantly mitigates congestion, improving both communication efficiency and stability in distributed training and inference—without modifying model architecture or training algorithms.

Technology Category

Application Category

📝 Abstract
Machine Learning jobs, carried out on large number of distributed high performance systems, involve periodic communication using operations like AllReduce, AllGather, and Broadcast. These operations may create high bandwidth and bursty traffic patterns, leading to network congestion and packet loss, thus impacting the performance of these jobs. Hence it is imperative to analyze these patterns, which can be helpful in provisioning network resources depending on the type of machine learning workloads. In this poster we carry out extensive analysis of the collective communication behavior seen in a wide variety of models (ex. DeepSeek, GPT, Llama, etc.) To achieve this we instrument Nvidia Collective Communication Library logging functionality for richer context about the collectives and workloads. We adjust configuration parameters that influence collective communication behavior, such as parallelism, number of nodes, and model type. This overview presents and discusses some of the results on the collective communication behavior for the open source DeepSeek V3 inferencing model, which includes operation type and count, transfer sizes per operation, and request size distribution. Our analysis shows that it makes sense to rethink current collective communication frameworks and network topologies so as to accommodate the effect of network anomalies on the mentioned workloads.
Problem

Research questions and friction points this paper is trying to address.

Analyze bursty traffic patterns in ML collective communication
Optimize network resources for diverse ML workloads
Improve collective communication frameworks for network anomalies
Innovation

Methods, ideas, or system contributions that make the work stand out.

Instrument NCCL logging for communication analysis
Adjust parameters like parallelism and node count
Analyze operation types and transfer sizes
🔎 Similar Papers
2024-06-07International Symposium on High-Performance Computer ArchitectureCitations: 5
J
Jit Gupta
Juniper Networks
A
Andrew Li
Stanford University
T
Tarun Banka
Juniper Networks
A
Ariel Cohen
Juniper Networks
T
T. Sridhar
Juniper Networks
R
Raj Yavatkar
Juniper Networks