Sparse Probabilistic Graph Circuits

2508.07763v1 cs.LG, cs.AI 2025-08-13
Авторы:

Martin Rektoris, Milan Papež, Václav Šmídl, Tomáš Pevný

Резюме на русском

#### Контекст Глубокие генерирующие модели (DGMs) для графов достигли впечатляющей выразительности благодаря эффективным и масштабируемым нейронным сетям. Однако, нелинейные операции в этих сетях мешают аналитическому вычислению простой проблемы простой проблемы логических запросов, что делает эти модели "нераспространенными". Недавно разработанные Probabilistic Graph Circuits (PGCs) решают эту проблему, предоставляя возможность распространенного аналитического вычисления проблем логических запросов. Однако, PGCs работают с толстыми представлениями графов, имеющими $\mathcal{O}(n^2)$ сложность, где $n$ — число узлов, а $m$ — число ребер в графе. Данная сложность становится проблемой при работе с большими графами. Чтобы улучшить эффективность, мы предлагаем Sparse Probabilistic Graph Circuits (SPGCs) — новую классу трактуемых моделей генерирующих графы, работающих с узкими представлениями графа, снижающими сложность до $\mathcal{O}(n + m)$, что значительно эффективнее для случаев $m \ll n^2$. #### Метод SPGCs оперируют спарсевыми представлениями графов, которые содержат только активные ребра, сокращая необходимую сложность. Для реализации используется архитектура глубоких нейронных сетей, адаптированная для эффективного вычисления стандартных запросов логического типа. Мы используем техники сжатия графов, что позволяет ускорить вычисления и сократить используемую память. Во время обучения, SPGCs используют потери, которые принуждают модель к точному вычислению количественных запросов. #### Результаты Мы проводили эксперименты на различных графовых задачах, включая задачу нового лекарственного дизайна. SPGCs доказали способность выполнять точные аналитические запросы с большей эффективностью по сравнению с интраккунтными DGMs. Мы показали, что SPGCs обеспечивают эффективность в памяти, эффективность вычислений и совпадают с интраккунтными DGMs по ключевым метрикам. Наши результаты подтвердили то, что SPGCs хорошо работают в скорости и точности, даже при работе с большими графами. #### Значимость SPGCs могут быть применены в различных областях, включая классификацию графов, разметку, новый дизайн лекарств и другие задачи, где требуется эффективный и точный вычислительный подход. Они предлагают выгоды в скорости и эффективности памяти по сравнению с интракктивными DGMs. Наш подход может положительно сказаться на развитии глубоких моделей графов, позволяя их применять в реальных задачах, где требуется высокая эффективность. #### Выводы Мы представили новую классу трактуемых моделе

Abstract

Deep generative models (DGMs) for graphs achieve impressively high expressive power thanks to very efficient and scalable neural networks. However, these networks contain non-linearities that prevent analytical computation of many standard probabilistic inference queries, i.e., these DGMs are considered \emph{intractable}. While recently proposed Probabilistic Graph Circuits (PGCs) address this issue by enabling \emph{tractable} probabilistic inference, they operate on dense graph representations with $\mathcal{O}(n^2)$ complexity for graphs with $n$ nodes and \emph{$m$ edges}. To address this scalability issue, we introduce Sparse PGCs, a new class of tractable generative models that operate directly on sparse graph representation, reducing the complexity to $\mathcal{O}(n + m)$, which is particularly beneficial for $m \ll n^2$. In the context of de novo drug design, we empirically demonstrate that SPGCs retain exact inference capabilities, improve memory efficiency and inference speed, and match the performance of intractable DGMs in key metrics.

Ссылки и действия