A deep attention-assisted and memory-augmented temporal convolutional network based model for rapid lithium-ion battery remaining useful life predictions with limited data

Research output: Journal Publications and ReviewsRGC 21 - Publication in refereed journalpeer-review

18 Scopus Citations
View graph of relations


Related Research Unit(s)


Original languageEnglish
Article number106903
Journal / PublicationJournal of Energy Storage
Online published24 Feb 2023
Publication statusPublished - Jun 2023


Most of existing data-driven studies on lithium-ion battery remaining useful life (RUL) prediction consider a large scope of cyclic data over the entire battery life. Yet, applications of these models can be hindered due to restricted availability of such data in reality. This paper thus aims to study the battery RUL prediction from a new angle, predicting RUL via data collected from a limited number of incomplete cycles, i.e., 10 cycles, at any ageing stage. An advanced deep learning framework, the attention-assisted temporal convolutional memory-augmented network (ATCMN), is developed to realize an accurate and rapid battery RUL prediction under such challenging problem setting. To build an informative input based on limited data, a three-dimensional tensor input structure is first designed to integrate 10-cycle raw battery data including the time, capacity, and temperature dimensions obtained from the partial discharge process. To process such high dimensional input, the ATCMN first develops an attention module to automate weighting different battery parameters, time steps, and ageing cycles in the input. A temporal convolution module coordinating the dilated causal convolution and point-wise convolution is next developed to learn latent spatial-temporal feature representation from the weighted input. A memory-augmented module is further developed to enhance the latent feature representation through a reconstruction based on the historical information. Finally, the ATCMN employs a prediction module to derive nonlinear mappings from learned latent features to the battery RULs. A comprehensive computational study is conducted to verify the effectiveness of the ATCMN. Results report the higher accuracy and faster prediction of the ATCMN via benchmarking against a set of the state-of-the-art methods in the considered RUL predictions. Experimental results also show that the proposed ATCMN possesses better generalizability to different battery chemistries and operational conditions. © 2023 Elsevier Ltd

Research Area(s)

  • Attention mechanism, Lithium-ion batteries, Memory augmented module, Remaining useful life, Temporal convolutional network

Citation Format(s)