GraphPPD: Posterior Predictive Modelling for Graph-Level Inference

2508.16995v1 stat.ML, cs.LG 2025-08-27
Авторы:

Soumyasundar Pal, Liheng Ma, Amine Natik, Yingxue Zhang, Mark Coates

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

## Контекст Одной из ключевых задач в области базовых и прикладных исследований является моделирование и квантификация неопределенности в моделях глубокого обучения. Для повышения надежности моделей необходимо гарантировать, что они могут делать безопасные решения при неполных или неоднозначных данных. Информация о неопределенности позволяет пользователю оценить уверенность модели в своих прогнозах. Несмотря на то, что в последние годы графовые нейронные сети (GNN) получили огромный интерес, большинство техник, оценивающих неопределенность, разработаны для узлов или связей, а не для задач на уровне графов. Это создает проблему, так как поиск решений на уровне графа требует более сложной модели неопределенности. В этой статье предлагается новая фреймворм для моделирования постериорного предиктивного распределения (PPD), которая может применяться в задачах на уровне графа. ## Метод Методология предложенного подхода основывается на использовании граф-уровневой репрезентации, полученной из существующих GNN. Наш фреймворк, GraphPPD, позволяет учитывать индивидуальные характеристики каждого графа и учитывать характеристики распределения на уровне графа. Мы используем вариационную модель для того, чтобы установить граф-уровневую репрезентацию. Наша модель генерирует неопределенность в моделировании, учитывая не только статистические характеристики, но и конкретные характеристики каждого графа. Это позволяет получить более точные и индивидуальные предсказания. ## Результаты Мы проверили нашу модель на нескольких бенчмарк-датасетах, включая задачи классификации графов. Мы сравнили нашу модель с другими подходами, которые оценивают неопределенность в GNN. Наши результаты показали, что GraphPPD превосходит другие техники в характеристиках точности и уверенности предсказаний. Были проведены эксперименты для измерения точности неопределенности и сравнения с другими моделями. Эти эксперименты показали, что GraphPPD может достигать более высокого уровня аккуратности в прогнозировании. ## Значимость Модель GraphPPD может применяться в различных областях, включая биоинформатику, химическое моделирование, информационные системы и другие области, где графы играют ключевую роль. Наш подход предоставляет более точные и надежные предсказания, что может быть полезно для принятия решений в сложных ситуациях, где неопределенность играет важную роль. Этот фреймворк может помочь улучшить надежность и понятность GNN-моделей в задачах на уровне графа. ## Выводы Результаты наших исследований показали, что GraphPPD является эффективным подходом для моде

Abstract

Accurate modelling and quantification of predictive uncertainty is crucial in deep learning since it allows a model to make safer decisions when the data is ambiguous and facilitates the users' understanding of the model's confidence in its predictions. Along with the tremendously increasing research focus on \emph{graph neural networks} (GNNs) in recent years, there have been numerous techniques which strive to capture the uncertainty in their predictions. However, most of these approaches are specifically designed for node or link-level tasks and cannot be directly applied to graph-level learning problems. In this paper, we propose a novel variational modelling framework for the \emph{posterior predictive distribution}~(PPD) to obtain uncertainty-aware prediction in graph-level learning tasks. Based on a graph-level embedding derived from one of the existing GNNs, our framework can learn the PPD in a data-adaptive fashion. Experimental results on several benchmark datasets exhibit the effectiveness of our approach.

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