A New Loss Function for Enhancing Peak Prediction in Time Series Data with High Variability

Time series models are considered among the most intricate models in machine learning. Due to sharp temporal variations, time series models normally fall short in predicting the peaks or local minima accurately. To overcome this challenge, we proposed a novel custom loss function, Enhanced Peak (EP) loss, specifically designed to pinpoint peaks and troughs in time series models, to address underestimations and overestimations in the forecasting process. EP loss applies an adaptive penalty when prediction errors exceed a specified threshold, encouraging the model to focus more effectively on these regions. To evaluate the effectiveness and versatility of EP loss, the loss function was tested on three highly variable datasets: NO x emissions, streamflow measurements, and gold price, implementing Gated Recurrent Unit and Transformer-based models. The results consistently demonstrated that EP loss significantly mitigates peak prediction errors compared to conventional loss functions, highlighting its potential for highly variable time series applications.