Green DetNet: Computation and Memory efficient DetNet using Smart Compression and Training

  • 2021-04-16 16:59:03
  • Nancy Nayak, Thulasi Tholeti, Muralikrishnan Srinivasan, Sheetal Kalyani
  • 0

Abstract

This paper introduces an incremental training framework for compressingpopular Deep Neural Network (DNN) based unfolded multiple-input-multiple-output(MIMO) detection algorithms like DetNet. The idea of incremental training isexplored to select the optimal depth while training. To reduce the computationrequirements or the number of FLoating point OPerations (FLOPs) and enforcesparsity in weights, the concept of structured regularization is explored usinggroup LASSO and sparse group LASSO. Our methods lead to an astounding $98.9\%$reduction in memory requirement and $81.63\%$ reduction in FLOPs when comparedwith DetNet without compromising on BER performance.

 

Quick Read (beta)

loading the full paper ...