GNN-Geo: A Graph Neural Network-based Fine-grained IP geolocation Framework

Abstract

Rule-based fine-grained IP geolocation methods are hard to generalize in computer networks which do not follow hypothetical rules. Recently, deep learning methods, like multi-layer perceptron (MLP), are tried to increase generalization capabilities. However, MLP is not so suitable for graph-structured data like networks. MLP treats IP addresses as isolated instances and ignores the connection information, which limits geolocation accuracy. In this work, we research how to increase the generalization capability with an emerging graph deep learning method -- Graph Neural Network (GNN). First, IP geolocation is re-formulated as an attributed graph node regression problem. Then, we propose a GNN-based IP geolocation framework named GNN-Geo. GNN-Geo consists of a preprocessor, an encoder, messaging passing (MP) layers and a decoder. The preprocessor and encoder transform measurement data into the initial node embeddings. MP layers refine the initial node embeddings by modeling the connection information. The decoder maps the refined embeddings to nodes' locations and relieves the convergence problem by considering prior knowledge. The experiments in 8 real-world IPv4/IPv6 networks in North America, Europe and Asia show the proposed GNN-Geo clearly outperforms the state-of-art rule-based and learning-based baselines. This work verifies the great potential of GNN for fine-grained IP geolocation.

This article has been accepted for publication in IEEE Transactions on Network Science and Engineering (TNSE). This is the author's version which has not been fully edited and content may change prior to final publication. Citation information: DOI 10.1109/TNSE.2023.3266752

References in corpus (10)

• Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
• Self-supervised Graph Learning for Recommendation
• Grid Search, Random Search, Genetic Algorithm: A Big Comparison for NAS
• Graph-based Deep Learning for Communication Networks: A Survey
• DisenKGAT: Knowledge Graph Embedding with Disentangled Graph Attention Network
• Message-Passing Neural Networks Learn Little's Law
• Utilizing Edge Features in Graph Neural Networks via Variational Information Maximization
• Applying Graph-based Deep Learning To Realistic Network Scenarios
• Adaptive Visibility Graph Neural Network and its Application in Modulation Classification
• Designing weighted and multiplex networks for deep learning user geolocation in Twitter