Enhancing Edge-Based Federated Learning with Privacy-Preserving Gradient Transmission for Tool Wear Detection

This paper focuses on the development of a personalized federated learning edge training platform that ensures privacy during gradient transmission (both uploading and downloading). We set up a client-side training environment for federated learning on a Raspberry Pi 4 and use the file transfer protocol for uploading training gradients to a server-side PC for aggregation. The personalized gradients are then sent back to the local clients via file transfer protocol for model updates. To validate the feasibility of the personalized federated learning algorithm, we employ a one-dimensional convolutional neural network for detecting tool wear. We present comparison results between our proposed personalized federated learning and the federated averaging algorithms to demonstrate performance enhancements. The personalized federated learning model is more closely aligned with local client needs, leading to better predictive performance, especially in scenarios of data scarcity and heterogeneous data. Experimental findings reveal that personalized federated learning can effectively manage tool wear across various notches or machines. Through the sharing of local model gradients via federated learning, we facilitate training and achieve personalization using a personalized detector. Both the personalized and global models show a 10-15% increase in accuracy, supporting preventive maintenance efforts.