English

SMART: A Surrogate Model for Predicting Application Runtime in Dragonfly Systems

Machine Learning 2025-11-17 v1 Distributed, Parallel, and Cluster Computing

Abstract

The Dragonfly network, with its high-radix and low-diameter structure, is a leading interconnect in high-performance computing. A major challenge is workload interference on shared network links. Parallel discrete event simulation (PDES) is commonly used to analyze workload interference. However, high-fidelity PDES is computationally expensive, making it impractical for large-scale or real-time scenarios. Hybrid simulation that incorporates data-driven surrogate models offers a promising alternative, especially for forecasting application runtime, a task complicated by the dynamic behavior of network traffic. We present \ourmodel, a surrogate model that combines graph neural networks (GNNs) and large language models (LLMs) to capture both spatial and temporal patterns from port level router data. \ourmodel outperforms existing statistical and machine learning baselines, enabling accurate runtime prediction and supporting efficient hybrid simulation of Dragonfly networks.

Keywords

Cite

@article{arxiv.2511.11111,
  title  = {SMART: A Surrogate Model for Predicting Application Runtime in Dragonfly Systems},
  author = {Xin Wang and Pietro Lodi Rizzini and Sourav Medya and Zhiling Lan},
  journal= {arXiv preprint arXiv:2511.11111},
  year   = {2025}
}

Comments

Accepted at AAAI 2026

R2 v1 2026-07-01T07:37:09.221Z