Verify Distributed Deep Learning Model Implementation Refinement with Iterative Relation Inference

2508.09505v1 cs.DC, cs.AI 2025-08-15
Авторы:

Zhanghan Wang, Ding Ding, Hang Zhu, Haibin Lin, Aurojit Panda

Резюме на русском

**Резюме** В современной AI распределенное обучение и интерпретация моделей стали неотъемлемой частью работы с большими моделями, которые превышают возможности единичного GPU. Однако распределенные реализации моделей часто приводят к багам, когда выходные данные распределенной модели отличаются от своей последовательной версии. Это может привести к ошибкам в приложениях, использующих эти модели. В статье предлагается метод для статического выявления таких ошибок с помощью проверки модельного рефинейма — то есть, можно ли восстановить выходы последовательной модели из выходов распределенной. Реализованный в GraphGuard, этот подход использует итеративную переписываемость для доказательства рефинейма модели. Он протестирован на таких моделях, как GPT и Llama-3, и показал свою эффективность в больших развертываниях. Более того, метод предоставляет детальный анализ, который помогает локализовать ошибки. Это значительно облегчает диагностику и улучшение распределенных моделей.

Abstract

Distributed machine learning training and inference is common today because today's large models require more memory and compute than can be provided by a single GPU. Distributed models are generally produced by programmers who take a sequential model specification and apply several distribution strategies to distribute state and computation across GPUs. Unfortunately, bugs can be introduced in the process, and a distributed model implementation's outputs might differ from the sequential model's outputs. In this paper, we describe an approach to statically identify such bugs by checking model refinement, that is, can the sequential model's outputs be reconstructed from the distributed model's outputs? Our approach, implemented in GraphGuard, uses iterative rewriting to prove model refinement. Our approach can scale to today's large models and deployments: we evaluate it using GPT and Llama-3. Further, it provides actionable output that aids in bug localization.

Ссылки и действия