ViNNPruner: Visual Interactive Pruning for Deep Learning
Loading...
Date
2022
Journal Title
Journal ISSN
Volume Title
Publisher
The Eurographics Association
Abstract
Neural networks grow vastly in size to tackle more sophisticated tasks. In many cases, such large networks are not deployable on particular hardware and need to be reduced in size. Pruning techniques help to shrink deep neural networks to smaller sizes by only decreasing their performance as little as possible. However, such pruning algorithms are often hard to understand by applying them and do not include domain knowledge which can potentially be bad for user goals. We propose ViNNPruner, a visual interactive pruning application that implements state-of-the-art pruning algorithms and the option for users to do manual pruning based on their knowledge. We show how the application facilitates gaining insights into automatic pruning algorithms and semi-automatically pruning oversized networks to make them more efficient using interactive visualizations.
Description
CCS Concepts: Human-centered computing --> Visual analytics; Computing methodologies --> Neural networks
@inproceedings{10.2312:mlvis.20221070,
booktitle = {Machine Learning Methods in Visualisation for Big Data},
editor = {Archambault, Daniel and Nabney, Ian and Peltonen, Jaakko},
title = {{ViNNPruner: Visual Interactive Pruning for Deep Learning}},
author = {Schlegel, Udo and Schiegg, Samuel and Keim, Daniel A.},
year = {2022},
publisher = {The Eurographics Association},
ISBN = {978-3-03868-182-3},
DOI = {10.2312/mlvis.20221070}
}