arrow
Return

Personalized Federated Learning Algorithm Based on Composite Feature Extractor

delete2026-02-23
delete0
PRE
AI
N
Nanrun Zhou
W
Wenjun Yu
DOI:10.1109/JIOT.2026.3667304delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
In the Internet of Things (IoT) environment, to maintain the global consistency and generalization capability of federated learning (FL) models with good data privacy, a personalized FL algorithm is proposed based on a composite feature extractor (CFE), i.e., FedCFE. The FedCFE framework consists of three core modules: a CFE, a conditional generator (CG), and a knowledge distillation mechanism. By modularly combining global with local feature extractors, the proposed CFE enables device-specific personalization without disrupting global knowledge sharing. Meanwhile, the representative pseudo-data is produced by a CG, which facilitates effective knowledge distillation from client models to the server model without requiring direct access to private data. Extensive experimental results demonstrate that the FedCFE consistently achieves competitive or superior performance across various heterogeneous data scenarios. On the highly non-IID CIFAR-100 dataset, the FedCFE improves classification accuracy by 2.19% compared with the state-of-the-art algorithms. These results indicate that the FedCFE provides an effective, lightweight, and privacy-preserving solution to realizing collaborative intelligence in resource-constrained and communication-unstable IoT environments.
Keywords:
Conditional generator (CG)
heterogeneous Internet of Things (IoT) data
information security
knowledge distillation
personalized federated learning (PFL)

Journal

IEEE Internet of Things Journal cover
IEEE Internet of Things Journal
IF:
8.9
Papers:
1.4W
Citations:
7.8W

Organization

S
Shanghai University of Engineering Science
Scholars:
7.7K
Papers: 4.8K
Citations: 6.0K