Heterogeneous Graph Neural Network for WiFi RSSI-Based Indoor Floor Classification

Abstract

Accurate indoor floor classification is essential for wireless positioning systems. However, the performance of conventional received signal strength indictor (RSSI)-based fingerprinting approaches is often limited by signal fluctuations and insufficient feature representation. To address these challenges, this paper introduces a heterogeneous graph neural network (GNN) framework that models WiFi signals using two types of nodes: reference points and Media Access Control (MAC) address. The edges between reference points and MAC addresses are weighted by normalized RSSI values, allowing the model to capture signal strength interactions and perform relation-aware message passing. Through this graph-based representation, the model can learn spatial and signal dependencies more effectively than conventional vector-based approaches. The proposed model was extensively evaluated under both benchmark and practical settings. On small-scale datasets, it achieved performance comparable to that of a conventional convolutional neural network trained on large-scale datasets, confirming its effectiveness with limited samples. In addition, the proposed model consistently outperformed other models under noisy conditions, achieving 93.88% accuracy on the widely used UJIIndoorLoc dataset and 97.3% accuracy in real-time experiments conducted at a test site. These values are significantly higher than those achieved using conventional machine learning (ML) baselines, highlighting the ability of the proposed model to handle real-world signal variations. These findings highlight that the heterogeneous GNN effectively captures spatial and signal-level dependencies, offering a robust and scalable solution for accurate indoor floor classification. Overall, this work presents a promising pathway for improving the performance and reliability of future wireless positioning systems.