Neural network structure analysis based on hierarchical force-directed graph drawing for multi-task learning

Atsushi Shibata, Fangyan Dong, Kaoru Hirota

Research output: Contribution to journalArticlepeer-review

2 Citations (Scopus)

Abstract

A hierarchical force-directed graph drawing is proposed for the analysis of a neural network structure that expresses the relationship between multitask and processes in neural networks represented as neuron clusters. The process revealed by our proposal indicates the neurons that are related to each task and the number of neurons or learning epochs that are sufficient. Our proposal is evaluated by visualizing neural networks learned on the Mixed National Institute of Standards and Technology (MNIST) database of handwritten digits, and the results show that inactive neurons, namely those that do not have a close relationship with any tasks, are located on the periphery part of the visualized network, and that cutting half of the training data on one specific task (out of ten) causes a 15% increase in the variance of neurons in clusters that react to the specific task compared to the reaction to all tasks. The proposal aims to be developed in order to support the design process of neural networks that consider multitasking of different categories, for example, one neural network for both the vision and motion system of a robot.

Original languageEnglish
Pages (from-to)225-231
Number of pages7
JournalJournal of Advanced Computational Intelligence and Intelligent Informatics
Volume19
Issue number2
DOIs
Publication statusPublished - 1 Mar 2015
Externally publishedYes

Keywords

  • Clustering
  • Multi-task learning
  • Network structure
  • Neural network
  • Visualization

Fingerprint

Dive into the research topics of 'Neural network structure analysis based on hierarchical force-directed graph drawing for multi-task learning'. Together they form a unique fingerprint.

Cite this