Multi-arm bandit-led clustering in federated learning
Federated Learning (FL) is a machine learning technique that enables the training of models across decentralized devices or nodes, without requiring the raw data to be centrally collected in one location. Instead, the model is trained in a distributed manner across multiple nodes, with each node onl...
Saved in:
| 主要作者: | |
|---|---|
| 其他作者: | |
| 格式: | Final Year Project |
| 語言: | English |
| 出版: |
Nanyang Technological University
2024
|
| 主題: | |
| 在線閱讀: | https://hdl.handle.net/10356/175424 |
| 標簽: |
添加標簽
沒有標簽, 成為第一個標記此記錄!
|
| 總結: | Federated Learning (FL) is a machine learning technique that enables the training of models across decentralized devices or nodes, without requiring the raw data to be centrally collected in one location. Instead, the model is trained in a distributed manner across multiple nodes, with each node only sending the model updates (and not the raw data) to a central server. The project’s direction was to explore and train an agent capable of recognizing which node can contribute best to maximize an existing cluster’s federated learning accuracy. The factor that was studied in this project was noise introduced to the data of a certain node that alters the data quality. The outcomes of the project showed that using reinforcement learning to train an agent that is capable of selecting the best node significantly improves the federated accuracy. As well as some noise alterations do make the model more robust in some cases. |
|---|