Emmanuel M. Smith
Visualising state space representations of LSTM networks
Smith, Emmanuel M.; Smith, Jim; Legg, Phil; Francis, Simon
Authors
Jim Smith James.Smith@uwe.ac.uk
Professor in Interactive Artificial Intelligence
Dr Phil Legg Phil.Legg@uwe.ac.uk
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.
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
Presentation Conference Type | Other |
---|---|
Conference Name | Workshop on Visualization for AI Explainability |
Acceptance Date | Aug 2, 2018 |
Peer Reviewed | Not Peer Reviewed |
Additional Information | Title of Conference or Conference Proceedings : Workshop on Visualization for AI Explainability |
Files
visualising_networks.html
(1.5 Mb)
Other
You might also like
Exploiting antipheromone in ant colony optimisation for interactive search-based software design and refactoring
(2016)
Presentation / Conference
Evolving atomic aesthetics and dynamics
(2016)
Journal Article
Sphere: A novel platform for increasing safety & security on unmanned systems
(2015)
Presentation / Conference
From evolutionary computation to the evolution of things
(2015)
Journal Article