Skip to main content

Research Repository

See what's under the surface

Advanced Search

Visualising state space representations of LSTM networks

Smith, Emmanuel M.; Smith, Jim; Legg, Phil; Francis, Simon

Authors

Emmanuel M. Smith sinclair.smith@uwe.ac.uk

Jim Smith James.Smith@uwe.ac.uk
Professor in Interactive Artificial Intelligence

Profile Image

Dr Phil Legg Phil.Legg@uwe.ac.uk
Associate Professor in Cyber Security

Simon Francis



Abstract

Long Short-Term Memory (LSTM) networks have proven to be one of the most effective models for making predictions on sequence-based tasks. These models work by capturing, remembering, and forgetting information relevant to their future predictions. The non-linear complexity of the mechanisms involved in this process means we currently lack tools for achieving interpretability. Ideally, we want these models to provide an explanation of why they make a particular prediction, given a specific input. Researchers have explored the idea of interpreting LSTMs in specific contexts such as natural language processing or classification, but they put minimal focus on approaches which are generalisable across different applications. To alleviate this, in this work, we demonstrate a method which enables the interpretation and comparison of LSTM states during time series predictions. We show that by reducing the dimensionality of network states one can scalably visualise patterns and explain model behaviours.

Presentation Conference Type Other
Peer Reviewed Not Peer Reviewed
APA6 Citation Smith, E. M., Smith, J., Legg, P., & Francis, S. Visualising state space representations of LSTM networks. Presented at Workshop on Visualization for AI Explainability
Additional Information Title of Conference or Conference Proceedings : Workshop on Visualization for AI Explainability
;