返回
Heterogeneous Defect Prediction Algorithm Combined With Federated Sparse Compression
DOI:10.1109/ACCESS.2023.3253765.png)
摘要
En 中文
Heterogeneous defect prediction (HDP) constructs a defect prediction model through the source project to realize the defect tendency prediction of the target project. HDP based on federated learning can combine multi-party defect data to improve the defect prediction effect under the condition of protecting privacy. However, with the increase of task complexity and model performance requirements, the number of neural network layers gradually deepens, and the model parameters also increase accordingly. In a federated learning scenario with multiple model parameters, limited communication bandwidth, and multiple clients, the amount of data received by the server is very large, causing huge communication pressure and seriously affecting the overall training efficiency. In order to reduce the communication cost, this paper proposes a federated sparse compression (FSC) algorithm. First, in order to improve the generalization performance of the model, each participant performs local training based on CapsNet, and achieves better prediction performance by calculating the relative position information of feature combinations. Then the gradient parameters are encrypted based on differential privacy to ensure data security. In order to reduce the amount of communication bits, the encrypted model parameters are subjected to sparse binary compression, the model training is converted from dense calculation to sparse calculation, and Golomb encoding is selected to be sent to the server for aggregation. Finally, the server side decodes the received data, performs sparse binary compression and encoding and sends it back to each participant. Experiments on nine projects in three public databases (Relink, NASA, and AEEEM) show that FSC can effectively reduce the amount of communication bits.
Keyword:
Software
Predictive models
Federated learning
Servers
Data models
Encoding
Computational modeling
CapsNet
sparse binary compression
Golomb encoding
期刊
IF:
3.6
论文数:
9.8W
被引数:
29.4W
机构
引用论文
WSCC: A Weight-Similarity-Based Client Clustering Approach for Non-IID Federated LearningWSCC: 一种基于权重相似度的非IID联邦学习客户端聚类方法
Toward Efficient Hierarchical Federated Learning Design Over Multi-Hop Wireless Communications Networks
IEEE ACCESS
IF3.6
Clustered Federated Learning: Model-Agnostic Distributed Multitask Optimization Under Privacy Constraints集群联邦学习: 隐私约束下的模型不可知的分布式多任务优化

