vix.ing · top · new · best · stats

ItNet: iterative neural networks with small graphs for accurate, efficient and anytime semantic segmentation

2021/01/21 by Thomas Pfeil, Pfeil, Thomas
Computer Science · Mathematics · #Advanced Neural Network Applications #Adversarial Robustness in Machine Learning #Computer Vision and Pattern Recognition (cs.CV) #Domain Adaptation and Few-Shot Learning #FOS: Computer and information sciences #Hardware Architecture (cs.AR) #Machine Learning (cs.LG) #Machine Learning (stat.ML) #cs.AR #cs.CV #cs.LG #stat.ML

paper · pdf · doi:10.48550/arxiv.2101.08685

11 pages, 8 figures

openalex publication_date 2021/01/21 · arxiv created 2021/08/03 · arxiv updated 2021/08/04 · openalex created_date 2021/08/16 · openalex updated_date 2026/07/28

Abstract

Deep neural networks have usually to be compressed and accelerated for their usage in low-power, e.g. mobile, devices. Recently, massively-parallel hardware accelerators were developed that offer high throughput and low latency at low power by utilizing in-memory computation. However, to exploit these benefits the computational graph of a neural network has to fit into the in-computation memory of these hardware systems that is usually rather limited in size. In this study, we introduce a class of network models that have a small memory footprint in terms of their computational graphs. To this end, the graph is designed to contain loops by iteratively executing a single network building block. Furthermore, the trade-off between accuracy and latency of these so-called iterative neural networks is improved by adding multiple intermediate outputs during both training and inference. We show state-of-the-art results for semantic segmentation on the CamVid and Cityscapes datasets that are especially demanding in terms of computational resources. In ablation studies, the improvement of network training by intermediate network outputs as well as the trade-off between weight sharing over iterations and the network size are investigated.

Citations

Related