A Machine Learning Dataset of Artificial Inner Ring Damage on Cylindrical Roller Bearings Measured Under Varying Cross-Influences

Abstract

In practical machine learning (ML) applications, covariate shifts and dependencies can significantly impact model robustness and prediction quality, leading to performance degradation under distribution shifts. In industrial settings, it is crucial to account for covariates during the design of experiments to ensure reliable generalization. The presented dataset of undamaged and artificially damaged cylindrical roller bearings is designed to address the lack of data resources for targeting domain and distribution shifts in this field. The dataset considers multiple key covariates, including mounting position, load, and rotational speed. Each covariate consists of multiple levels optimized for groupbased cross-validation. This allows the user to exclude specific groups in the training to validate and test the algorithm. Using this approach, algorithms can be evaluated for their robustness and the effect on the model caused by distribution shifts, allowing their generalization capabilities to be studied under realistic conditions.