Towards Robust Federated Image Classification: An Empirical Study of Weight Selection Strategies in Manufacturing

TL;DR

This study investigates the comparative effectiveness of two weight selection strategies: Final Epoch Weight Selection (FEWS) and Optimal Epoch Weight Selection (OEWS) to determine whether FEWS or OEWS enhances the global FL model's performance across communication rounds (CRs).

Abstract

In the realm of Federated Learning (FL), particularly within the manufacturing sector, the strategy for selecting client weights for server aggregation is pivotal for model performance. This study investigates the comparative effectiveness of two weight selection strategies: Final Epoch Weight Selection (FEWS) and Optimal Epoch Weight Selection (OEWS). Designed for manufacturing contexts where collaboration typically involves a limited number of partners (two to four clients), our research focuses on federated image classification tasks. We employ various neural network architectures, including EfficientNet, ResNet, and VGG, to assess the impact of these weight selection strategies on model convergence and robustness. Our research aims to determine whether FEWS or OEWS enhances the global FL model's performance across communication rounds (CRs). Through empirical analysis and rigorous experimentation, we seek to provide valuable insights for optimizing FL implementations in manufacturing, ensuring that collaborative efforts yield the most effective and reliable models with a limited number of participating clients. The findings from this study are expected to refine FL practices significantly in manufacturing, thereby enhancing the efficiency and performance of collaborative machine learning endeavors in this vital sector.

Figures (5)

• Figure 1: Visualization of class distribution and sample data for each client in the federated learning process.
• Figure 2: Test Datasets for Evaluating Federated Global Models, Clients' Local Centralized Models, and Entire Dataset Centralized Models.
• Figure 3: Proposed flow diagram for the training and evaluation process in a research-focused federated learning deployment within academic settings.
• Figure 4: Proposed flow diagram for the training and evaluation process in real-world federated learning deployment within industrial settings.
• Figure 5: Comparison of confidence scores: centralized model (left) vs. federated OEWS model (right) on the external test dataset using the DenseNet architecture