Collaborative Learning at the Edge for Air Pollution Prediction

• Published in: IEEE transactions on instrumentation and measurement Vol. 73; p. 1
• Main Authors: Wardana, I Nyoman Kusuma, Gardner, Julian W., Fahmy, Suhaib A.
• Format: Journal Article
• Language: English
• Published: New York IEEE 01.01.2024; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Air pollution; Air quality; Atmospheric modeling; Collaborative learning; Data exchange; Data models; Deep learning; edge devices; Federated learning; Model accuracy; Monitoring; Outdoor air quality; Sensors; spatiotemporal; Spatiotemporal data; Spatiotemporal phenomena; Training
• ISSN: 0018-9456; 1557-9662
• DOI: 10.1109/TIM.2023.3341116

The rapid growth of connected sensing devices has resulted in enormous amounts of data being collected and processed. Air quality data collected from different monitoring stations is spatially and temporally correlated, and hence, collaborative learning can improve deep learning model performance. Research on collaborative learning at the edge has not specifically focused so far on air quality prediction, which is the subject of this work.We compare three collaborative learning strategies and implement them on edge devices, such as the Raspberry Pi and Jetson Nano, with communication facilitated through the MQTT protocol. Federated learning is shown to enhance model accuracy in comparison to local training alone. An approach called clustered model exchange reduces communication costs during training. Finally, our proposed spatiotemporal data exchange approach exploits information from neighboring sensing stations to enhance model performance. It achieves the highest accuracy in air quality predictions, outperforming other methods in minimizing loss during training. It results in RMSE improvements ranging from 0.525% to 8.934% when compared to models that are only trained locally. We compare the real training costs of the three methods on real hardware to validate them.