用透明设备模拟优化大模型训练,成本降56%
Maya: Optimizing Deep Learning Training Workloads using GPU Runtime Emulation
- 通过拦截GPU接口调用,无修改代码捕获完整训练行为
- 预测误差低于5%,识别出可节省56%训练成本的配置
- 无需改代码或写特定语言,适合大规模模型训练优化
训练大型基础模型成本高达数亿美元,部署优化至关重要。当前方法依赖工程师在昂贵计算集群上手动调试训练方案,效率低下。虽有性能建模系统支持配置探索,但需将工作负载转换为定制化描述语言,造成实际代码与建模表示之间的语义鸿沟。这一鸿沟导致三难困境:要么支持场景有限,要么规格复杂难以落地,要么牺牲预测精度。我们提出Maya,一种通过透明设备模拟消除该困境的性能建模系统。Maya在训练框架与加速器设备之间窄接口处运行,拦截未修改训练代码的设备API调用,直接观测底层操作,实现无需代码改动或翻译即可完整捕捉工作负载行为。评估显示,Maya在多种模型和优化策略下预测误差均低于5%,所识别配置相比现有方法可降低训练成本达56%。
原文摘要 · Abstract (English)
Training large foundation models costs hundreds of millions of dollars, making deployment optimization critical. Current approaches require machine learning engineers to manually craft training recipes through error-prone trial-and-error on expensive compute clusters. To enable efficient exploration of training configurations, researchers have developed performance modeling systems. However, these systems force users to translate their workloads into custom specification languages, introducing a fundamental semantic gap between the actual workload and its representation. This gap creates an inherent tradeoff: systems must either support a narrow set of workloads to maintain usability, require complex specifications that limit practical adoption, or compromise prediction accuracy with simplified performance models. We present Maya, a performance modeling system that eliminates these tradeoffs through transparent device emulation. By operating at the narrow interface between training frameworks and accelerator devices, Maya can capture complete workload behavior without requiring code modifications or translations. Maya intercepts device API calls from unmodified training code to directly observe low-level operations, enabling accurate performance prediction while maintaining both ease of use and generality. Our evaluation shows Maya achieves less than 5% prediction error across diverse models and optimization strategies, identifying configurations that reduce training costs by up to 56% compared to existing approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。