UQGNN: Uncertainty Quantification of Graph Neural Networks for Multivariate Spatiotemporal Prediction

#### Контекст Spatiotemporal prediction является ключевым компонентом многих реальных приложений, таких как урбанизация, оптимизация транспортных систем, быстродействие при чрезвычайных ситуациях и контроль эпидемий. Однако, существующие модели детерминированными характеризуются, то есть, они предсказывают только среднее значение без оценки неопределенности. Это может привести к неточным и небезопасным результатам. Несмотря на то, что недавние исследования предложили модели, которые оценивают неопределенность, они ограничиваются одним феноменом, таким как транспорт, криминал или экономические данные. Нет же общих подходов для корреляции различных городских феноменов. Мы предлагаем **UQGNN (Uncertainty Quantification of Graph Neural Networks)**, которая адресует эти проблемы, объединяя прогнозирование и оценку неопределенности в одной модели. #### Метод UQGNN основывается на **Graph Neural Network (GNN)** и включает два ключевых модуля: (i) **Interaction-aware Spatiotemporal Embedding Module**, который использует многомерные диффузионные сверточные сети и сети временной конвульюции для достаточного извлечения зависимостей между пространственными и временными факторами; (ii) **Multivariate Probabilistic Prediction Module**, который оценивает ожидаемые значения и соответствующие неопределенности. Мы используем диффузионную графовую сеть для представления пространственных связей, а временную сеть конвульюции для учета исторических трендов. Модель распределенно построена для предсказания нескольких переменных в пространственно-временном пространстве. #### Результаты Мы проверили UQGNN на четырех реальных датасетах: Shenzhen, New York City, и Chicago, включающих в себя несколько связанных феноменов, таких как транспортные потоки, климат и демография. Наша модель показала значительный выигрыш в прогностической точности и оценке неопределенности по сравнению с состоятельными моделями. На датасете из Shenzhen, например, UQGNN достигла 5% усовершенствований в обеих метриках. Эти результаты подтверждают эффективность модели в объединении учета неопределенности и прогнозирования сложных пространственно-временных явлений. #### Значимость Модель UQGNN может применяться в различных областях, таких как машинное обучение, городская планирования, оптимизация транспорта, и мониторинг экологических явлений. Она предоставляет точные прогнозы и оценивает неопределенность, что улучшает надежность решений. Это значительное улучшение по сравнению с детерминированными моделями. Будущие исследования будут сосредоточены на расширении модели для обработки более высокомасштабных данных и включении более разнообразных феноменов. #### Выводы Мы представили UQGNN, первый под