SpaPool: Soft Partition Assignment Pooling for__Graph Neural Networks

#### Контекст Графовые нейронные сети (Graph Neural Networks, GNNs) широко используются для решения задач на основе графов в различных областях, таких как социальные сети, биология и технические приложения. Однако, при работе с большими графами, эффективность обучения и вывода может существенно ухудшиться из-за высокого порога вычислительных и памяти затрат. Это приводит к необходимости в методах, эффективно сокращающих размер графа при сохранении структурных свойств и информации. Существуют существующие методы пулинга, такие как SAGPool и ASAP, которые решают эту проблему, но имеют ограничения в точности и эффективности. Существует мотивация для разработки более универсального метода, который мог бы комбинировать два подхода: дружественных к масштабу и точности. #### Метод SpaPool (Soft Partition Assignment Pooling) представляет собой новый метод пулинга, который объединяет преимущества двух доминирующих подходов: денсного и спарсеного. Метод разбивает вершины графа на адаптивное число кластеров, используя нейронную сеть для определения принадлежности к кластерам. Подход гарантирует оптимальный баланс между сохранением структурной информации и эффективностью. Архитектура SpaPool состоит из двух этапов: начальной фазы кластеризации и финальной фазы синтеза. В первой фазе, нейронная сеть вычисляет вероятности принадлежности каждой вершине к каждому кластеру. Во второй фазе, эти вероятности используются для создания оптимальных кластеров, которые будут сохранять самую важную информацию графа. #### Результаты Эксперименты проводились на двух различных типах данных, включая данные реальных графов и синтетические графы. Метод SpaPool сравнивался с двумя центральными методами: SAGPool и ASAP. Результаты показали, что SpaPool демонстрирует более высокую точность и эффективность на малых графах, а также существенно уменьшает вычислительные затраты на больших графах. Эти результаты подтверждают, что SpaPool является более гибким и эффективным методом, чем существующие. #### Значимость SpaPool может быть применен в различных областях, таких как социальные сети, биология и технические приложения, где необходимо обработать большие графы с ограниченными ресурсами. Уникальный подход SpaPool также может быть использован для повышения эффективности и точности в приложениях, использующих машинное обучение на графах. Это делает SpaPool важной концепцией в развитии новых технологий в области GNNs. #### Выводы Разработанный SpaPool является первым методом, который эффективно комбинирует два подхода к пулингу. На основании резуль