FedEE: A Federated Graph Learning Solution for Extended Enterprise Collaboration

Abstract

Today's business environment is characterized by uncertainty and competition, so the capability to adapt to the evolving era and unforeseen challenges is essential in business strategies. Recent studies on extended enterprise indicate that collaboration among different stakeholders is beneficial for surviving these unexpected changes. However, the barriers such as market uncertainty, privacy and trust concerns, and individual contribution evaluation limit the implementation and application of the extended enterprise concept. Federated learning (FL), in which multiple enterprise entities can use a shared model while retaining all training data locally, has emerged as a promising AI solution for accumulating insights from multiple stakeholders and providing collaborative decision-making. Furthermore, the enhanced privacy-protection benefits of FL remove the barriers to implementing extended enterprise collaboration. In particular, an FL central server manages the local updates of multiple enterprise entities (FL clients) and aggregates their contributions to improve the global model training. Meanwhile, to address the time-series graph learning problem in most business environments, we incorporate TCN (Temporal Convolutional Network), GCN (Graph Convolutional Neural Network) and GRU (Gated Recurrent Unit) architecture into FL to capture the temporal-spatial dependencies in individual data sources. Furthermore, we use traffic flow forecasting as the use case of our proposed framework to verify its effectiveness. Finally, the experimental results on a real traffic flow dataset and the comparison results with the state-of-the-art baseline methods show that our proposed solution achieves superior performance.