MNIST数据库
计算机科学
独立同分布随机变量
联合学习
数据建模
最优化问题
数据挖掘
优化算法
机器学习
人工智能
深度学习
数据库
算法
数学优化
随机变量
统计
数学
作者
Xinyan Li,Huimin Zhao,Wu Deng
出处
期刊:IEEE Internet of Things Journal
[Institute of Electrical and Electronics Engineers]
日期:2024-01-16
卷期号:11 (9): 16693-16699
被引量:10
标识
DOI:10.1109/jiot.2024.3354942
摘要
Federated learning (FL) algorithm has been widely studied in recent years due to its ability for sharing data while protecting privacy. However, FL has risks such as model inversion attack, and is less effective when data is non-independent and identically distributed (non-IID). In response to these challenges, an intelligent optimization-based federated learning (IOFL) framework is developed to improve the privacy protection performance and global model performance in this paper. In the IOFL, the server searches model parameters by using intelligent optimization algorithm and distributes it to the clients. The clients use local data to validate the issued model by the server and return the validation results to the server. The server calculates the fitness function based on the weighted average of the received validation results, which guide the intelligent optimization algorithm to search for new model parameters. The experimental results on MNIST and Fashion-MNIST dataset show that the accuracy of the IOFL can reach over 0.8 and 0.68 under different non-IID settings with 200 round communications, whose performance is not affected by non-IID data distribution at clients.
科研通智能强力驱动
Strongly Powered by AbleSci AI