Der Artikel diskutiert die Herausforderungen beim Training von Graph Neural Networks (GNNs) auf großen Graphen und stellt den GPU Initiated Direct Storage Access (GIDS) Dataloader vor. Dieser ermöglicht effizientes Training von GNNs auf großen Graphen, indem er GPU-orientierte Ansätze nutzt und die Speicherressourcen effektiv verwaltet. Der GIDS-Dataloader verbessert die Leistung und Skalierbarkeit von GNNs, indem er die Graphstrukturdaten im CPU-Speicher und die Merkmalsdaten im Speicher speichert, die dynamische Speicherzugriffsakkumulator-Technik verwendet, einen konstanten CPU-Puffer einsetzt und das Window-Buffering zur Optimierung der GPU-Cache-Nutzung nutzt.
Para outro idioma
do conteúdo fonte
arxiv.org
Principais Insights Extraídos De
by Jeongmin Bri... às arxiv.org 03-08-2024
https://arxiv.org/pdf/2306.16384.pdfPerguntas Mais Profundas