DeepGraph: A PyCharm Tool for Visualizing and Understanding Deep Learning Models

Qiang Hu, Lei Ma, Jianjun Zhao

Research output: Chapter in Book/Report/Conference proceedingConference contribution

3 Citations (Scopus)

Abstract

As more and more domain specific big data become available, there comes a strong need on the fast development and deployment of deep learning (DL) systems with high quality for domain specific applications, including many safety-critical scenarios. In traditional software engineering, software visualization plays an important role to enhance developers' performance with many tools available. However, there are limited visualization supports existing for DL systems, especially in integrated development environments (IDEs) that allow a developer to visualize the source code of a deep neural network (DNN) and its graph architecture. In this paper, we propose DeepGraph, a visualization tool for visualizing and understanding a deep neural network. DeepGraph analyzes the training program to construct the graph representation of a DNN, and establishes and maintains the linkage (mapping) between the source code of the training program and its corresponding neural network architecture. We implemented DeepGraph as a PyCharm plugin and performed preliminary empirical study to demonstrate its usefulness for understanding deep nueral networks.

Original languageEnglish
Title of host publicationProceedings - 25th Asia-Pacific Software Engineering Conference, APSEC 2018
PublisherIEEE Computer Society
Pages628-632
Number of pages5
ISBN (Electronic)9781728119700
DOIs
Publication statusPublished - Jul 2 2018
Event25th Asia-Pacific Software Engineering Conference, APSEC 2018 - Nara, Japan
Duration: Dec 4 2018Dec 7 2018

Publication series

NameProceedings - Asia-Pacific Software Engineering Conference, APSEC
Volume2018-December
ISSN (Print)1530-1362

Conference

Conference25th Asia-Pacific Software Engineering Conference, APSEC 2018
CountryJapan
CityNara
Period12/4/1812/7/18

All Science Journal Classification (ASJC) codes

  • Software

Fingerprint Dive into the research topics of 'DeepGraph: A PyCharm Tool for Visualizing and Understanding Deep Learning Models'. Together they form a unique fingerprint.

  • Cite this

    Hu, Q., Ma, L., & Zhao, J. (2018). DeepGraph: A PyCharm Tool for Visualizing and Understanding Deep Learning Models. In Proceedings - 25th Asia-Pacific Software Engineering Conference, APSEC 2018 (pp. 628-632). [8719435] (Proceedings - Asia-Pacific Software Engineering Conference, APSEC; Vol. 2018-December). IEEE Computer Society. https://doi.org/10.1109/APSEC.2018.00079