English

deepstruct -- linking deep learning and graph theory

Machine Learning 2021-12-15 v2 Artificial Intelligence Neural and Evolutionary Computing

Abstract

deepstruct connects deep learning models and graph theory such that different graph structures can be imposed on neural networks or graph structures can be extracted from trained neural network models. For this, deepstruct provides deep neural network models with different restrictions which can be created based on an initial graph. Further, tools to extract graph structures from trained models are available. This step of extracting graphs can be computationally expensive even for models of just a few dozen thousand parameters and poses a challenging problem. deepstruct supports research in pruning, neural architecture search, automated network design and structure analysis of neural networks.

Keywords

Cite

@article{arxiv.2111.06679,
  title  = {deepstruct -- linking deep learning and graph theory},
  author = {Julian Stier and Michael Granitzer},
  journal= {arXiv preprint arXiv:2111.06679},
  year   = {2021}
}