SONG: Self-Organizing Neural Graphs

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Struski, Łukasz, Danel, Tomasz, Śmieja, Marek, Tabor, Jacek, Zieliński, Bartosz
Formato: Preprint
Publicado: 2021
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916811777245184
author Struski, Łukasz
Danel, Tomasz
Śmieja, Marek
Tabor, Jacek
Zieliński, Bartosz
author_facet Struski, Łukasz
Danel, Tomasz
Śmieja, Marek
Tabor, Jacek
Zieliński, Bartosz
contents Recent years have seen a surge in research on deep interpretable neural networks with decision trees as one of the most commonly incorporated tools. There are at least three advantages of using decision trees over logistic regression classification models: they are easy to interpret since they are based on binary decisions, they can make decisions faster, and they provide a hierarchy of classes. However, one of the well-known drawbacks of decision trees, as compared to decision graphs, is that decision trees cannot reuse the decision nodes. Nevertheless, decision graphs were not commonly used in deep learning due to the lack of efficient gradient-based training techniques. In this paper, we fill this gap and provide a general paradigm based on Markov processes, which allows for efficient training of the special type of decision graphs, which we call Self-Organizing Neural Graphs (SONG). We provide an extensive theoretical study of SONG, complemented by experiments conducted on Letter, Connect4, MNIST, CIFAR, and TinyImageNet datasets, showing that our method performs on par or better than existing decision models.
format Preprint
id arxiv_https___arxiv_org_abs_2107_13214
institution arXiv
publishDate 2021
record_format arxiv
spellingShingle SONG: Self-Organizing Neural Graphs
Struski, Łukasz
Danel, Tomasz
Śmieja, Marek
Tabor, Jacek
Zieliński, Bartosz
Machine Learning
Artificial Intelligence
Recent years have seen a surge in research on deep interpretable neural networks with decision trees as one of the most commonly incorporated tools. There are at least three advantages of using decision trees over logistic regression classification models: they are easy to interpret since they are based on binary decisions, they can make decisions faster, and they provide a hierarchy of classes. However, one of the well-known drawbacks of decision trees, as compared to decision graphs, is that decision trees cannot reuse the decision nodes. Nevertheless, decision graphs were not commonly used in deep learning due to the lack of efficient gradient-based training techniques. In this paper, we fill this gap and provide a general paradigm based on Markov processes, which allows for efficient training of the special type of decision graphs, which we call Self-Organizing Neural Graphs (SONG). We provide an extensive theoretical study of SONG, complemented by experiments conducted on Letter, Connect4, MNIST, CIFAR, and TinyImageNet datasets, showing that our method performs on par or better than existing decision models.
title SONG: Self-Organizing Neural Graphs
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2107.13214