RRA-FFSCIL: Inter-intra classes representation and relationship augmentation federated few-shot incremental learning

Yalan Jiang,Yang Cheng,Dan Wang,Bin Song
DOI: https://doi.org/10.1016/j.neucom.2024.127956
IF: 6
2024-09-01
Neurocomputing
Abstract:Federated learning (FL), as a distributed machine learning paradigm, enables on-device model training and inference without data updates or privacy breaches, promoting edge intelligence in Industrial Internet of Things (IIoTs) applications. However, FL applying in IIoTs faces some challenges in practice: (1) the catastrophic forgetting of old data resulting from a streaming reception for arriving data; (2) the forgetting exacerbated by the non-i.i.d. data distribution across devices; (3) the overfitting caused by insufficient sample size of new data (i.e., few samples). To address the above problems, we propose an inter-intra classes representation and relationship augmentation federated few-shot incremental learning framework called RRA-FFSCIL, effectively mitigating catastrophic forgetting from local and global perspectives. Specifically, forgetting from streaming data is alleviated by designed modules in the client side of RRA-FFSCIL: a random few-shot (FS) episodic adaptation strategy, a class-projection relation strengthening loss function, and a class-aware distribution balancing loss function. Meanwhile, the mitigation of global forgetting from non-i.i.d. of distributions across devices thanks to selecting the best old models on the server to assist local class relation enhancement. We conduct extensive experiments on three datasets (cifar100, tinyImageNet, miniImageNet) to validate the effectiveness and superiority of RRA-FFSCIL in addressing the forgetting problem associated with streaming data.
computer science, artificial intelligence
What problem does this paper attempt to address?