Saved in:
Bibliographic Details
Main Authors: Liu, Qingxiang, Sun, Sheng, Liang, Yuxuan, Xu, Xiaolong, Liu, Min, Bilal, Muhammad, Wang, Yuwei, Li, Xujing, Zheng, Yu
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2411.14046
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915029403566080
author Liu, Qingxiang
Sun, Sheng
Liang, Yuxuan
Xu, Xiaolong
Liu, Min
Bilal, Muhammad
Wang, Yuwei
Li, Xujing
Zheng, Yu
author_facet Liu, Qingxiang
Sun, Sheng
Liang, Yuxuan
Xu, Xiaolong
Liu, Min
Bilal, Muhammad
Wang, Yuwei
Li, Xujing
Zheng, Yu
contents Multiple federated learning (FL) methods are proposed for traffic flow forecasting (TFF) to avoid heavy-transmission and privacy-leaking concerns resulting from the disclosure of raw data in centralized methods. However, these FL methods adopt offline learning which may yield subpar performance, when concept drift occurs, i.e., distributions of historical and future data vary. Online learning can detect concept drift during model training, thus more applicable to TFF. Nevertheless, the existing federated online learning method for TFF fails to efficiently solve the concept drift problem and causes tremendous computing and communication overhead. Therefore, we propose a novel method named Resource-Efficient Federated Online Learning (REFOL) for TFF, which guarantees prediction performance in a communication-lightweight and computation-efficient way. Specifically, we design a data-driven client participation mechanism to detect the occurrence of concept drift and determine clients' participation necessity. Subsequently, we propose an adaptive online optimization strategy, which guarantees prediction performance and meanwhile avoids meaningless model updates. Then, a graph convolution-based model aggregation mechanism is designed, aiming to assess participants' contribution based on spatial correlation without importing extra communication and computing consumption on clients. Finally, we conduct extensive experiments on real-world datasets to demonstrate the superiority of REFOL in terms of prediction improvement and resource economization.
format Preprint
id arxiv_https___arxiv_org_abs_2411_14046
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle REFOL: Resource-Efficient Federated Online Learning for Traffic Flow Forecasting
Liu, Qingxiang
Sun, Sheng
Liang, Yuxuan
Xu, Xiaolong
Liu, Min
Bilal, Muhammad
Wang, Yuwei
Li, Xujing
Zheng, Yu
Machine Learning
Multiple federated learning (FL) methods are proposed for traffic flow forecasting (TFF) to avoid heavy-transmission and privacy-leaking concerns resulting from the disclosure of raw data in centralized methods. However, these FL methods adopt offline learning which may yield subpar performance, when concept drift occurs, i.e., distributions of historical and future data vary. Online learning can detect concept drift during model training, thus more applicable to TFF. Nevertheless, the existing federated online learning method for TFF fails to efficiently solve the concept drift problem and causes tremendous computing and communication overhead. Therefore, we propose a novel method named Resource-Efficient Federated Online Learning (REFOL) for TFF, which guarantees prediction performance in a communication-lightweight and computation-efficient way. Specifically, we design a data-driven client participation mechanism to detect the occurrence of concept drift and determine clients' participation necessity. Subsequently, we propose an adaptive online optimization strategy, which guarantees prediction performance and meanwhile avoids meaningless model updates. Then, a graph convolution-based model aggregation mechanism is designed, aiming to assess participants' contribution based on spatial correlation without importing extra communication and computing consumption on clients. Finally, we conduct extensive experiments on real-world datasets to demonstrate the superiority of REFOL in terms of prediction improvement and resource economization.
title REFOL: Resource-Efficient Federated Online Learning for Traffic Flow Forecasting
topic Machine Learning
url https://arxiv.org/abs/2411.14046