Neural network training fingerprint: visual analytics of the training process in classification neural networks

Martha Dais Ferreira (Corresponding author), Gabriel Dias Cantareira, Rodrigo Fernandes de Mello, Fernando V. Paulovich

Research output: Contribution to journalArticleAcademicpeer-review

4 Citations (Scopus)

Abstract

The striking results of deep neural networks (DNN) have motivated its wide acceptance to tackle large datasets and complex tasks such as natural language processing, facial recognition, and artificial image generation. However, DNN parameters are often empirically selected on a trial-and-error approach without detailed information on convergence behavior. While some visualization techniques have been proposed to aid the comprehension of general-purpose neural networks, only a few explore the training process, lacking the ability to adequately display how abstract representations are formed and represent the influence of training parameters during this process. This paper describes neural network training fingerprint (NNTF), a visual analytics approach to investigate the training process of any neural network performing classification. NNTF allows understanding how classification decisions change along the training process, displaying information about convergence, oscillations, and training rates. We show its usefulness through case studies and demonstrate how it can support the analysis of training parameters.
Original languageEnglish
Article number3
Pages (from-to)593-612
Number of pages20
JournalJournal of Visualization
Volume25
Issue number3
DOIs
Publication statusPublished - Jun 2022
Externally publishedYes

Keywords

  • Deep learning
  • Neural network training
  • Neural network visualization
  • Visual analytics
  • Visualization

Fingerprint

Dive into the research topics of 'Neural network training fingerprint: visual analytics of the training process in classification neural networks'. Together they form a unique fingerprint.

Cite this