Spotting Virus from Satellites: Modeling the Circulation of West Nile Virus Through Graph Neural Networks

Lorenzo Bonicelli,Angelo Porrello,Stefano Vincenzi,Carla Ippoliti,Federica Iapaolo,Annamaria Conte,Simone Calderara
DOI: https://doi.org/10.1109/TGRS.2023.3293270
2023-07-06
Abstract:The occurrence of West Nile Virus (WNV) represents one of the most common mosquito-borne zoonosis viral infections. Its circulation is usually associated with climatic and environmental conditions suitable for vector proliferation and virus replication. On top of that, several statistical models have been developed to shape and forecast WNV circulation: in particular, the recent massive availability of Earth Observation (EO) data, coupled with the continuous advances in the field of Artificial Intelligence, offer valuable opportunities. In this paper, we seek to predict WNV circulation by feeding Deep Neural Networks (DNNs) with satellite images, which have been extensively shown to hold environmental and climatic features. Notably, while previous approaches analyze each geographical site independently, we propose a spatial-aware approach that considers also the characteristics of close sites. Specifically, we build upon Graph Neural Networks (GNN) to aggregate features from neighbouring places, and further extend these modules to consider multiple relations, such as the difference in temperature and soil moisture between two sites, as well as the geographical distance. Moreover, we inject time-related information directly into the model to take into account the seasonality of virus spread. We design an experimental setting that combines satellite images - from Landsat and Sentinel missions - with ground truth observations of WNV circulation in Italy. We show that our proposed Multi-Adjacency Graph Attention Network (MAGAT) consistently leads to higher performance when paired with an appropriate pre-training stage. Finally, we assess the importance of each component of MAGAT in our ablation studies.
Computer Vision and Pattern Recognition,Artificial Intelligence,Machine Learning
What problem does this paper attempt to address?
This paper aims to solve the problem of West Nile Virus (WNV) transmission prediction. Specifically, the author attempts to predict the spread of WNV by leveraging environmental and climatic features in satellite images. Traditional statistical models usually analyze each geographical location point independently when predicting WNV transmission, while this paper proposes a spatially - aware method that not only considers the features of the target location but also incorporates information from neighboring locations to improve the prediction accuracy. ### Problems the paper attempts to solve 1. **Improve the accuracy of WNV transmission prediction**: Traditional prediction methods are usually based on data from a single geographical location point, ignoring the influence of neighboring regions. In this paper, by introducing Graph Neural Networks (GNN), the information of neighboring locations is incorporated into the model, thus more comprehensively considering the influence of environmental and climatic factors on WNV transmission. 2. **Utilize multi - source satellite data**: This paper combines multispectral satellite images from Sentinel and Landsat missions, as well as ground - observation data, to construct a supervised binary - classification task for predicting the spread of WNV. 3. **Consider temporal and spatial factors**: In addition to the spatial proximity relationship, the model also injects time - related information to capture the impact of seasonal changes on WNV transmission. ### Method overview 1. **Feature extractor**: Use ResNet18 as the backbone network to extract high - dimensional features from multispectral satellite images. To better adapt to multispectral data, the author adopts a pre - training strategy, initializing the model parameters through a colorization pretext task. 2. **Graph aggregation module**: Construct a graph neural network, where each node represents a geographical location point and the edges represent the relationships between different locations (such as geographical distance, temperature difference, soil humidity difference). The graph neural network aggregates the features of neighboring nodes through the Multi - Adjacency Graph Attention Network (MAGAT). 3. **Final classifier**: Input the features after graph aggregation into the classifier to perform binary - classification prediction of WNV transmission. ### Key innovation points - **Multi - adjacency graph attention mechanism**: MAGAT can handle multiple real - valued adjacency matrices, considering not only geographical distance but also the similarity of environmental factors such as temperature and soil humidity. - **Conditional batch normalization**: Through the Conditional Batch Normalization (CBN) layer, the time information (such as month) is integrated into the model to capture the impact of seasonal changes on WNV transmission. - **Spatial awareness**: Not only analyze a single geographical location point, but also consider the information of neighboring locations, which improves the robustness and prediction accuracy of the model. Through these innovations, the method proposed in the paper achieves higher performance in predicting WNV transmission, and the importance of each component is verified through ablation experiments.