Reduced Softmax Unit for Deep Neural Network Accelerators
Abstract
The Softmax activation layer is a very popular Deep Neural Network (DNN) component when dealing with multi-class prediction problems. However, in DNN accelerator implementations it creates additional complexities due to the need for computation of the exponential for each of its inputs. In this brief we propose a simplified version of the activation unit for accelerators, where only a comparator unit produces the classification result, by choosing the maximum among its inputs. Due to the nature of the activation function, we show that this result is always identical to the classification produced by the Softmax layer.
- Publication:
-
arXiv e-prints
- Pub Date:
- December 2021
- DOI:
- 10.48550/arXiv.2201.04562
- arXiv:
- arXiv:2201.04562
- Bibcode:
- 2022arXiv220104562S
- Keywords:
-
- Computer Science - Hardware Architecture