Deep-Unrolling Multidimensional Harmonic Retrieval Algorithms on Neuromorphic Hardware

TL;DR

By casting the MHR problem as a sparse recovery problem, the currently proposed, deep-unrolling-based Structured Learned Iterative Shrinkage and Thresholding (S-LISTA) algorithm is devised to solve it efficiently using complex-valued convolutional neural networks with complex-valued activations, which are trained using a supervised regression objective.

Abstract

This paper explores the potential of conversion-based neuromorphic algorithms for highly accurate and energy-efficient single-snapshot multidimensional harmonic retrieval (MHR). By casting the MHR problem as a sparse recovery problem, we devise the currently proposed, deep-unrolling-based Structured Learned Iterative Shrinkage and Thresholding (S-LISTA) algorithm to solve it efficiently using complex-valued convolutional neural networks with complex-valued activations, which are trained using a supervised regression objective. Afterward, a novel method for converting the complex-valued convolutional layers and activations into spiking neural networks (SNNs) is developed. At the heart of this method lies the recently proposed Few Spikes (FS) conversion, which is extended by modifying the neuron model's parameters and internal dynamics to account for the inherent coupling between real and imaginary parts in complex-valued computations. Finally, the converted SNNs are mapped onto the SpiNNaker2 neuromorphic board, and a comparison in terms of estimation accuracy and power efficiency between the original CNNs deployed on an NVIDIA Jetson Xavier and the SNNs is being conducted. The measurement results show that the converted SNNs achieve almost five-fold power efficiency at moderate performance loss compared to the original CNNs.

Figures (4)

• Figure 1: Schematic representation of S-LISTA architecture.
• Figure 2: FS neuron approximating Eq. \ref{['eq:st']} ($\alpha = 1$) and $f(s)=s^{2}$ for different $K$. Blue denotes the true function, and orange is the approximation.
• Figure 3: SRE averaged over SNRs from $10\text{dB}$ to $30\text{dB}$.
• Figure 4: Power consumption statistical distribution.