Remaining lifespan prediction on multiple types of hard disks under conditions of data imbalance

Abstract

Most of the previous regression models for predicting the remaining useful life (RUL) of hard disks primarily focused on failure data obtained through SMART monitoring. However, models that only considered failure data were not accurate enough. In order to address the issue of imbalanced sample quantities in different categories within the dataset, this paper proposes a method for predicting the RUL of hard disks under imbalanced conditions. It takes into account the imbalance issues in both SMART failure data and healthy data, as well as the imbalance in monitoring data attributes between hard disk drives and solid-state drives. The paper improves upon the traditional Transformer time series model and constructs a comprehensive and unified model for the RUL of hard disks across multiple categories. The proposed model achieves an RMSE score of 3.325, and has 66 % in MAE improved to the conventional models. The proposed model has promising prospects for future applications in predicting the remaining life of hard disks.