T3-Vis: visual analytic for Training and fine-Tuning Transformers in NLP

Raymond Li, Wen Xiao, Lanjun Wang, Hyeju Jang, Giuseppe Carenini


Abstract
Transformers are the dominant architecture in NLP, but their training and fine-tuning is still very challenging. In this paper, we present the design and implementation of a visual analytic framework for assisting researchers in such process, by providing them with valuable insights about the model’s intrinsic properties and behaviours. Our framework offers an intuitive overview that allows the user to explore different facets of the model (e.g., hidden states, attention) through interactive visualization, and allows a suite of built-in algorithms that compute the importance of model components and different parts of the input sequence. Case studies and feedback from a user focus group indicate that the framework is useful, and suggest several improvements. Our framework is available at: https://github.com/raymondzmc/T3-Vis.
Anthology ID:
2021.emnlp-demo.26
Volume:
Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing: System Demonstrations
Month:
November
Year:
2021
Address:
Online and Punta Cana, Dominican Republic
Editors:
Heike Adel, Shuming Shi
Venue:
EMNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
220–230
Language:
URL:
https://aclanthology.org/2021.emnlp-demo.26
DOI:
10.18653/v1/2021.emnlp-demo.26
Bibkey:
Cite (ACL):
Raymond Li, Wen Xiao, Lanjun Wang, Hyeju Jang, and Giuseppe Carenini. 2021. T3-Vis: visual analytic for Training and fine-Tuning Transformers in NLP. In Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing: System Demonstrations, pages 220–230, Online and Punta Cana, Dominican Republic. Association for Computational Linguistics.
Cite (Informal):
T3-Vis: visual analytic for Training and fine-Tuning Transformers in NLP (Li et al., EMNLP 2021)
Copy Citation:
PDF:
https://aclanthology.org/2021.emnlp-demo.26.pdf
Video:
 https://aclanthology.org/2021.emnlp-demo.26.mp4
Code
 raymondzmc/t3-vis