Abstract:Deep neural networks (DNNs) have been established as the state-of-the-art method for advanced machine learning applications. Recently proposed by the Google Brain’s team, the capsule networks (CapsNets) have improved the generalization ability, as compared to DNNs, due to their multidimensional capsules and preserving the spatial relationship between different objects. However, they pose significantly high computational and memory requirements, making their energy-efficient inference a challenging task. This article provides, for the first time, an in-depth analysis to highlight the design and runtime challenges for the (on-chip scratchpad) memories deployed in hardware accelerators executing fast CapsNets inference. To enable an efficient design, we propose an application-specific memory architecture, called DESCNet, which minimizes the off-chip memory accesses, while efficiently feeding the data to the hardware accelerator executing CapsNets inference. We analyze the corresponding on-chip memory requirement and leverage it to propose a methodology for exploring different scratchpad memory (SPM) designs and their energy/area tradeoffs. Afterward, an application-specific power-gating technique for the on-chip SPM is employed to further reduce its energy consumption, depending upon the mapped dataflow of the CapsNet and the utilization across different operations of its processing. We integrated our DESCNet memory design, as well as another state-of-the-art memory design Marchisio et al. [2018] for comparison studies, with an opensource DNN accelerator executing Google’s CapsNet model Sabour et al. [2017] for the MNIST dataset. We also enhanced the design to execute the recent deep CapsNet model Rajasegaran et al. [2019] for the CIFAR10 dataset. Note: we use the same benchmarks and test conditions for which these CapsNets have been proposed and evaluated by their respective teams. The complete hardware is synthesized for a 32-nm CMOS technology using the ASIC-design flow with Synopsys tools and CACTI-P, and detailed area, performance, and power/energy estimation is performed using different configurations. Our results for a selected Pareto-optimal solution demonstrate no performance loss and an energy reduction of 79% for the complete accelerator, including computational units and memories, when compared to the state-of-the-art design.

Lightweight Multi-Dimensional Memristive CapsNet

A lightweight capsule network via channel-space decoupling and self-attention routing

CapStore: Energy-Efficient Design and Management of the On-Chip Memory for CapsuleNet Inference Accelerators

DE-CapsNet: A Diverse Enhanced Capsule Network with Disperse Dynamic Routing

Ddrm-Capsnet: Capsule Network Based On Deep Dynamic Routing Mechanism For Complex Data

DeepCaps: Going Deeper With Capsule Networks

DESCNet: Developing Efficient Scratchpad Memories for Capsule Network Hardware

Spiking CapsNet: A spiking neural network with a biologically plausible routing rule between capsules

RS-CapsNet: an Advanced Capsule Network.

Deep multi-prototype capsule networks

Efficient-CapsNet: capsule network with self-attention routing

Enabling Highly Efficient Capsule Networks Processing Through A PIM-Based Architecture Design

Path Capsule Networks

FSC-CapsNet: Fractionally-Strided Convolutional Capsule Network for Complex Data.

A novel capsule network based on deep routing and residual learning

BETTER: Bayesian-Based Training and Lightweight Transfer Architecture for Reliable and High-Speed Memristor Neural Network Deployment

Dynamic neural network with memristive CIM and CAM for 2D and 3D vision

Memristor Neural Network Design

A Novel Capsule Network Based on Multi-Order Descartes Extension Transformation

Adaptive Capsule Network

A Fast Routing Capsule Network With Improved Dense Blocks