Attention Mechanism-Based Neural Network for Prediction of Battery Cycle Life in the Presence of Missing Data

Abstract

As batteries become widespread applications across various domains, the prediction of battery cycle life has attracted increasing attention. However, the intricate internal mechanisms of batteries pose challenges to achieving accurate battery lifetime prediction, and the inherent patterns within temporal data from battery experiments are often elusive. Meanwhile, the commonality of missing data in real-world battery usage further complicates accurate lifetime prediction. To address these issues, this article develops a self-attention-based neural network (NN) to precisely forecast battery cycle life, leveraging an attention mechanism that proficiently manages time-series data without the need for recurrent frameworks and adeptly handles the data-missing scenarios. Furthermore, a two-stage training approach is adopted, where certain network hyperparameters are fine-tuned in a sequential manner to enhance training efficacy. The results show that the proposed self-attention-based NN approach not only achieves superior predictive precision compared with the benchmarks including Elastic Net and CNN-LSTM but also maintains resilience against missing-data scenarios, ensuring reliable battery lifetime predictions. This work highlights the superior performance of utilizing attention mechanism for battery cycle life prognostics.