ES-Mask: Evolutionary Strip Mask for Explaining Time Series Prediction (Student Abstract)

Authors

  • Yifei Sun School of Computer Science and Technology, Huazhong University of Science and Technology
  • Cheng Song School of Computer Science and Technology, Huazhong University of Science and Technology
  • Feng Lu School of Computer Science and Technology, Huazhong University of Science and Technology
  • Wei Li School of Computer Science, The University of Sydney
  • Hai Jin School of Computer Science and Technology, Huazhong University of Science and Technology
  • Albert Y. Zomaya School of Computer Science, The University of Sydney

DOI:

https://doi.org/10.1609/aaai.v37i13.27031

Keywords:

Machine Learning, Explainable AI, Time Series, Evolutionary Algorithm

Abstract

Machine learning models are increasingly used in time series prediction with promising results. The model explanation of time series prediction falls behind the model development and makes less sense to users in understanding model decisions. This paper proposes ES-Mask, a post-hoc and model-agnostic evolutionary strip mask-based saliency approach for time series applications. ES-Mask designs the mask consisting of strips with the same salient value in consecutive time steps to produce binary and sustained feature importance scores over time for easy understanding and interpretation of time series. ES-Mask uses an evolutionary algorithm to search for the optimal mask by manipulating strips in rounds, thus is agnostic to models by involving no internal model states in the search. The initial experiments on MIMIC-III data set show that ES-Mask outperforms state-of-the-art methods.

Downloads

Published

2024-07-15

How to Cite

Sun, Y., Song, C., Lu, F., Li, W., Jin, H., & Zomaya, A. Y. (2024). ES-Mask: Evolutionary Strip Mask for Explaining Time Series Prediction (Student Abstract). Proceedings of the AAAI Conference on Artificial Intelligence, 37(13), 16342-16343. https://doi.org/10.1609/aaai.v37i13.27031