Federated Learning for Surface Roughness
This study proposes a federated learning-based real-time surface roughness prediction framework for WEDM to address issues of empirical parameter tuning and data privacy. By sharing only the model parameters, cross-machine training was enabled without exposing raw data. A custom data acquisition sys...
Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2025-06-01
|
| Series: | Applied Sciences |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2076-3417/15/13/7046 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| Summary: | This study proposes a federated learning-based real-time surface roughness prediction framework for WEDM to address issues of empirical parameter tuning and data privacy. By sharing only the model parameters, cross-machine training was enabled without exposing raw data. A custom data acquisition system collected discharge current and spindle current signals, which were solely used as input features to train the deep learning model. Data balancing techniques improved prediction accuracy, achieving performance comparable to centralized models. After optimizing the training dataset through balancing and augmentation, the federated model achieved a Root Mean Square Error (RMSE) of 0.076, which closely approaches the 0.074 RMSE obtained by the centralized model. The results show that federated learning enhances both data security and model generalization, offering an effective solution for smart manufacturing. |
|---|---|
| ISSN: | 2076-3417 |