Graph convolutional networks for spatial interpolation of correlated data
Marianne Abémgnigni Njifon,Dominic Schuhmacher
DOI: https://doi.org/10.1016/j.spasta.2024.100822
IF: 2.125
2024-03-20
Spatial Statistics
Abstract:Several deep learning methods for spatial data have been developed that report good performance in a big data setting. These methods typically require the choice of an appropriate kernel and some tuning of hyperparameters, which are contributing reasons for poor performance on smaller data sets. In this paper we present a general formalism for graph based neural networks for spatial prediction. Specifically, we propose a method which explores higher order neighborhood structures and combines several kernels via convolution, yielding higher flexibility in capturing spatial correlations. In this paper, we propose a mathematical construction of a graph-based neural network for spatial prediction that substantially generalizes the KCN model in [Appleby, Liu and Liu (2020). Kriging convolutional networks. In Proc. AAAI Conf. AI 34, pp. 3187–3194]. In particular, our model, referred to as SPONGE, allows for integrated learning of the convolutional kernel, admits higher order neighborhood structures and can make use of the distance between locations in the neighborhood and between labels of neighboring nodes. All of this yields higher flexibility in capturing spatial correlations. We investigate in simulation studies including small, medium and (reasonably) large data sets in what situations and to what extent graph based neural networks manage to close the mean squared error gap to classic (geo)statistical methods, in particular universal Kriging, whose prediction comes with a guarantee of optimality but at a hefty price of O(n3) in computational complexity. SPONGE comes close to or (if the conditions for optimality are violated) even beats universal Kriging, whose predictions incur a high computational cost if n is large. Furthermore we study the improvement for general SPONGE in comparison with the usual KCN. Finally, we compare various graph based graph-based neural network models on various larger real world data sets and apply our method to the prediction of soil organic carbon in the Southern southern part of Malawi.
remote sensing,geosciences, multidisciplinary,mathematics, interdisciplinary applications,statistics & probability