Enhancing Distributed Machine Learning through Data Shuffling: Techniques, Challenges, and Implications

School of Electronic and Information Engineering, Beijing Jiaotong University, 100044, Beijing, China

^* Corresponding author: 22251093@bjtu.edu.cn

Abstract

In distributed machine learning, data shuffling is a crucial data preprocessing technique that significantly impacts the efficiency and performance of model training. As distributed machine learning scales across multiple computing nodes, the ability to shuffle data effectively and efficiently has become essential for achieving high-quality model performance and minimizing communication costs. This paper systematically explores various data shuffling methods, including random shuffling, stratified shuffling, K-fold shuffling, and coded shuffling, each with distinct advantages, limitations, and application scenarios. Random shuffling is simple and fast but may lead to imbalanced class distributions, while stratified shuffling maintains class proportions at the cost of increased complexity. K-fold shuffling provides robust model evaluation through multiple training-validation splits, though it is computationally demanding. Coded shuffling, on the other hand, optimizes communication costs in distributed settings but requires sophisticated encoding-decoding techniques. The study also highlights the challenges associated with current shuffling techniques, such as handling class imbalance, high computational complexity, and adapting to dynamic, real-time data. This paper proposes potential solutions to enhance the efficacy of data shuffling, including hybrid methodologies, automated stratification processes, and optimized coding strategies. This work aims to guide future research on data shuffling in distributed machine learning environments, ultimately advancing model robustness and generalization across complex real-world applications.