Distributed multi-task classification : a decentralized online learning approach

• Main Authors: Zhang, Chi, Zhao, Peilin, Hao, Shuji, Soh, Yeng Chai, Lee, Bu Sung, Miao, Chunyan, Hoi, Steven C. H.
• Other Authors: School of Computer Science and Engineering
• Format: Article
• Language: English
• Published: 2019
• Subjects: Decentralized Distributed Learning; Multi-task Learning; Engineering::Electrical and electronic engineering
• Online Access: https://hdl.handle.net/10356/82962; http://hdl.handle.net/10220/49099
• Institution: Nanyang Technological University

Although dispersing one single task to distributed learning nodes has been intensively studied by the previous research, multi-task learning on distributed networks is still an area that has not been fully exploited, especially under decentralized settings. The challenge lies in the fact that different tasks may have different optimal learning weights while communication through the distributed network forces all tasks to converge to an unique classifier. In this paper, we present a novel algorithm to overcome this challenge and enable learning multiple tasks simultaneously on a decentralized distributed network. Specifically, the learning framework can be separated into two phases: (i) multi-task information is shared within each node on the first phase; (ii) communication between nodes then leads the whole network to converge to a common minimizer. Theoretical analysis indicates that our algorithm achieves a O(T−−√) regret bound when compared with the best classifier in hindsight, which is further validated by experiments on both synthetic and real-world datasets.