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 |
Publicly Available Date | Sep 19, 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
A novel mirror neuron inspired decision-making architecture for human–robot interaction
(2023)
Journal Article
Inter-annotator agreement using the Conversation Analysis Modelling Schema, for dialogue
(2022)
Journal Article
Sentence encoding for dialogue act classification
(2021)
Journal Article