Morphling 增强器将 GNN 训练速度提升 20 倍

研究人员开发了 Morphling，这是一种领域特定代码合成器，旨在优化图神经网络 (GNN) 的训练。Morphling 将 GNN 规范编译为适用于 OpenMP、CUDA 和 MPI 等各种平台的便携式、后端专用实现。它包含一个运行时引擎，该引擎根据输入统计信息动态选择密集或稀疏执行路径，从而减少不必要的计算。评估表明，与 PyTorch Geometric 和 Deep Graph Library 等现有框架相比，Morphling 显著提高了训练吞吐量并减少了内存消耗。 AI

影响 加速 GNN 训练并减少内存使用，从而能够实现更大规模的基于图的人工智能应用。

排序理由 这是一篇详细介绍优化 GNN 训练新方法的论文。[lever_c_demoted from research: ic=1 ai=1.0]

在 arXiv cs.LG 阅读 →

• CUDA
• Deep Graph Library
• Graph Neural Networks
• Morphling
• MPI
• OpenMP
• PyTorch Geometric

• 论文
• 基础设施

AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →