Abstract:We propose a co-design approach for compute-in-memory inference for deep neural networks (DNN). We use multiplication-free function approximators based on $ell _{1}$ norm along with a co-adapted processing array and compute flow. Using the approach, we overcame many deficiencies in the current art of in-SRAM DNN processing such as the need for digital-to-analog converters (DACs) at each operating SRAM row/column, the need for high precision analog-to-digital converters (ADCs), limited support for multi-bit precision weights, and limited vector-scale parallelism. Our co-adapted implementation seamlessly extends to multi-bit precision weights, it doesn't require DACs, and it easily extends to higher vector-scale parallelism. We also propose an SRAM-immersed successive approximation ADC (SA-ADC), where we exploit the parasitic capacitance of bit lines of SRAM array as a capacitive DAC. Since the dominant area overhead in SA-ADC comes due to its capacitive DAC, by exploiting the intrinsic parasitic of SRAM array, our approach allows low area implementation of within-SRAM SA-ADC. Our $8times 62$ SRAM macro, which requires a 5-bit ADC, achieves ~105 tera operations per second per Watt (TOPS/W) with 8-bit input/weight processing at 45 nm CMOS. Our $8times 30$ SRAM macro, which requires a 4-bit ADC, achieves ~84 TOPS/W. SRAM macros that require lower ADC precision are more tolerant of process variability, however, have lower TOPS/W as well. We evaluated the accuracy and performance of our proposed network for MNIST, CIFAR10, and CIFAR100 datasets. We chose a network configuration which adaptively mixes multiplication-free and regular operators. The network configura-ions utilize the multiplication-free operator for more than 85% operations from the total. The selected configurations are 98.6% accurate for MNIST, 90.2% for CIFAR10, and 66.9% for CIFAR100. Since most of the operations in the considered configurations are based on proposed SRAM macros, our compute-in-memory's efficiency benefits broadly translate to the system-level.

Memorization Capacity of Neural Networks with Conditional Computation

Exponentially Increasing the Capacity-to-Computation Ratio for Conditional Computation in Deep Learning

On the Complexity of Neural Computation in Superposition

Conditional computation in neural networks: principles and research trends

Hardware Efficient Reconfigurable Logic-in-Memory Circuit Based Neural Network Computing.

Assessment of inference accuracy and memory capacity of computation-in-memory enabled neural network due to quantized weights, gradients, input and output signals, and memory non-idealities

A Task-Adaptive In-Situ ReRAM Computing for Graph Convolutional Networks

Parameter Compression of Recurrent Neural Networks and Degradation of Short-term Memory

Training Deep Nets with Sublinear Memory Cost

Short-term Sequence Memory: Compressive Effects of Recurrent Network Dynamics

Learning Narrow One-Hidden-Layer ReLU Networks

Neural networks with ReLU powers need less depth

Neurocoder: Learning General-Purpose Computation Using Stored Neural Programs

MF-Net: Compute-In-Memory SRAM for Multibit Precision Inference Using Memory-Immersed Data Conversion and Multiplication-Free Operators

Generalizability of Memorization Neural Networks

The Expressive Leaky Memory Neuron: an Efficient and Expressive Phenomenological Neuron Model Can Solve Long-Horizon Tasks

ReMeCo: Reliable Memristor-Based In-Memory Neuromorphic Computation

Memorization Capacity for Additive Fine-Tuning with Small ReLU Networks

Reversible Recurrent Neural Networks

Memorization with neural nets: going beyond the worst case

Stably unactivated neurons in ReLU neural networks