English

Seq2Seq-Vis: A Visual Debugging Tool for Sequence-to-Sequence Models

Computation and Language 2018-10-17 v2 Artificial Intelligence Neural and Evolutionary Computing

Abstract

Neural Sequence-to-Sequence models have proven to be accurate and robust for many sequence prediction tasks, and have become the standard approach for automatic translation of text. The models work in a five stage blackbox process that involves encoding a source sequence to a vector space and then decoding out to a new target sequence. This process is now standard, but like many deep learning methods remains quite difficult to understand or debug. In this work, we present a visual analysis tool that allows interaction with a trained sequence-to-sequence model through each stage of the translation process. The aim is to identify which patterns have been learned and to detect model errors. We demonstrate the utility of our tool through several real-world large-scale sequence-to-sequence use cases.

Keywords

Cite

@article{arxiv.1804.09299,
  title  = {Seq2Seq-Vis: A Visual Debugging Tool for Sequence-to-Sequence Models},
  author = {Hendrik Strobelt and Sebastian Gehrmann and Michael Behrisch and Adam Perer and Hanspeter Pfister and Alexander M. Rush},
  journal= {arXiv preprint arXiv:1804.09299},
  year   = {2018}
}

Comments

VAST - IEEE VIS 2018