Data-driven machine learning is playing a crucial role in the advancements of Industry 4.0, specifically in enhancing predictive maintenance and quality inspection. Federated learning (FL) enables multiple participants to develop a machine learning model without compromising the privacy and confidentiality of their data. In this paper, we evaluate the performance of different FL aggregation methods and compare them to central and local training approaches. Our study is based on four datasets with varying data distributions. The results indicate that the performance of FL is highly dependent on the data and its distribution among clients. In some scenarios, FL can be an effective alternative to traditional central or local training methods. Additionally, we introduce a new federated learning dataset from a real-world quality inspection setting.
翻译:数据驱动的机器学习在工业4.0的发展中发挥着至关重要的作用,特别是在增强预测性维护和质量检查方面。联合学习使多个参与者能够在不牺牲其数据的隐私和机密性的情况下开发机器学习模型。在本文中,我们评估了不同的联合学习聚合方法的性能,并将其与中心化和本地训练方法进行比较。我们的研究基于四个具有不同数据分布的数据集。结果表明,联合学习的性能高度依赖于数据及其在客户端之间的分布情况。在某些情况下,联合学习可以成为传统的中央或本地训练方法的有效替代方法。此外,我们介绍了一个来自真实质量检查环境的新的联合学习数据集。