Node selection based on label quantity information in federated learning
Aiming at the problem that the difference of node data distribution has adverse effect on the performance of federated learning algorithm, a node selection algorithm based on label quantity information was proposed.An optimization objective based on the label quantity information of nodes was design...
Saved in:
| Main Authors: | , , , , , |
|---|---|
| Format: | Article |
| Language: | zho |
| Published: |
China InfoCom Media Group
2021-12-01
|
| Series: | 物联网学报 |
| Subjects: | |
| Online Access: | http://www.wlwxb.com.cn/thesisDetails#10.11959/j.issn.2096-3750.2021.00249 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| Summary: | Aiming at the problem that the difference of node data distribution has adverse effect on the performance of federated learning algorithm, a node selection algorithm based on label quantity information was proposed.An optimization objective based on the label quantity information of nodes was designed, considering the optimization problem of selecting the nodes with balanced label distribution under a certain time consumption limit.According to the correlation between the aggregated label distribution of selected nodes and the convergence of the global model, the upper bound of the weight divergence of the global model was reduced to improve the convergence stability of the algorithm.Simulation results shows that the new algorithm had higher convergence efficiency than the existing node selection algorithm. |
|---|---|
| ISSN: | 2096-3750 |