🔬 Spotlight на ICML: ускоряем графовые нейросети — 20 июня 2026 г. в 07:30:01.391
🔬 Spotlight на ICML: ускоряем графовые нейросети На конференции ICML 2026, проходящей с 6 по 11 июля, внимание уделено работе Яндекса и ШАДа, удостоенной статуса Spotlight. Эта статья исследует ускорение Graph Neural Networks (GNN), что стало важным шагом в области графовых вычислений. Проблема в том, что графовые неэффективно обрабатываются на GPU, так как они расположены неравномерно в памяти. Из-за этого видеокарты часто простаивают в ожидании загрузки данных. Авторы исследования минимизируют ненужные перемещения данных между памятью и вычислительными блоками, что является ключом к ускорению операций. Они также переносят алгоритмы на тензорные ядра, что способствует ускорению на графах с высокой плотностью. Результаты впечатляют: новое решение обеспечивает ускорение в 3–10 раз в зависимости от архитектуры, а в некоторых сценариях потребление памяти снижается более чем на порядок. 📊 Работа доступна на arXiv и код можно найти здесь.