Federated Learning for Indoor Localization via Model Reliability With Dropout
Junha Park, Jiseon Moon, Taekyoon Kim, Peng Wu, Tales Imbiriba, Pau Closas, Sunwoo Kim
Abstract
In this letter, we propose a novel model weight update method that accounts for the reliability of the local clients in FL-based indoor localization. FL shows degraded localization performance than centralized learning because of the non-independent and identically distributed (non-IID) data configuration. Thus, we aim to improve the localization performance by applying the reliability of the local clients, which is quantified by the model uncertainty of the local models. Bayesian models provide a framework for capturing model uncertainty but usually requires a substantial computational cost as well, particularly for high-dimensional learning problems. In order to resolve this computational issue, the proposed scheme applies Monte Carlo (MC) dropout to approximate the Bayesian uncertainty quantification with enhanced computational efficiency. Our simulation results show that the proposed learning method improves localization performance compared to the existing model, federated averaging (FedAvg), and close to the centralized learning performance.