Traditional IoT device collaboration is usually static and cannot adjust the collaboration mode between devices according to various changes, which limits work efficiency. To this end, an IoT device collaboration optimization algorithm based on graph neural network and federated learning is studied. This method abstracts various IoT device nodes and their communication relationships into graph structured data for storage, and then uses federated learning to train the graph convolutional network with graph structured data. The obtained model can be used to optimize the collaboration mode of IoT devices. During the training process, the total average MSE (mean square error) between the output and the label of the graph convolutional network model based on federated learning is 0.968; the total standard deviation of MSE is 0.0353; the total time from training to model convergence is 435.82 s, of which data transmission time accounts for 27.1% and model training time accounts for 72.9%. In a 2-h practical experiment, the graph convolutional network model based on federated learning was used to optimize the collaboration mode of smart homes, achieving a target environment residence time of 87 min and a total power consumption reduction of 0.69 kW·h. The results show that this method can effectively optimize the collaboration efficiency of IoT devices, reduce training time and network overhead, but it fails to improve the prediction accuracy of the model and may also lead to a decrease in stability.
Read full abstract