Federated Multi-task Graph Learning

Yijing Liu,Dongming Han,Jianwei Zhang,Haiyang Zhu,Mingliang Xu,Wei Chen
DOI: https://doi.org/10.1145/3527622
IF: 5
2022-04-22
ACM Transactions on Intelligent Systems and Technology
Abstract:Distributed processing and analysis of large-scale graph data remain challenging because of the high-level discrepancy among graphs. This study investigates a novel subproblem: the distributed multi-task learning on the graph, which jointly learns multiple analysis tasks from decentralized graphs. We propose a federated multi-task graph learning (FMTGL) framework to solve the problem within a privacy-preserving and scalable scheme. Its core is an innovative data-fusion mechanism and a low-latency distributed optimization method. The former captures multi-source data relatedness and generates universal task representation for local task analysis. The latter enables the quick update of our framework with gradients sparsification and tree-based aggregation. As a theoretical result, the proposed optimization method has a convergence rate interpolates between \(\mathcal {O}(1/T) \) and \(\mathcal {O}(1/\sqrt {T}) \) , up to logarithmic terms. Unlike previous studies, our work analyzes the convergence behavior with adaptive stepsize selection and non-convex assumption. Experimental results on three graph datasets verify the effectiveness and scalability of FMTGL.
computer science, information systems, artificial intelligence
What problem does this paper attempt to address?