학술논문

Federated Multitask Learning for Complaint Identification Using Graph Attention Network
Document Type
Periodical
Source
IEEE Transactions on Artificial Intelligence IEEE Trans. Artif. Intell. Artificial Intelligence, IEEE Transactions on. 5(3):1277-1286 Mar, 2024
Subject
Computing and Processing
Task analysis
Data models
Multitasking
Training
Federated learning
Adaptation models
Standards organizations
Complaint identification
deep learning
federated learning (FL)
graph-attention network
multitask learning (MTL)
Language
ISSN
2691-4581
Abstract
Prior study on automatically identifying complaints on social media has relied on extensive feature engineering in centralized settings, with no consideration for the decentralized, nonidentically independently distributed (Non-IID), and privacy-conscious aspect of complaints, which can hinder data collection, distribution, and learning. In this work, we propose a graph attention network (GAT)-based multitask framework that intends to learn two closely related tasks, complaint detection (primary task) and sentiment classification (auxiliary task), simultaneously in federated learning scenarios. We propose the Federated Combination (FedComb) algorithm, a two-sided adaptive optimization technique that simultaneously optimizes global and local models. The proposed methodology outperforms several baselines for the intended task of recognizing complaints in decentralized settings, according to quantitative and qualitative studies on two benchmark datasets.