Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces a method to reduce the O(n) complexity of label prediction in neural networks by leveraging predefined vector systems as targets for latent space configuration. By associating label prediction with O(1) closest cluster center search within these vector systems, the method avoids direct comparison with all class prototypes. Experiments across multiple datasets demonstrate up to 11.6x speedup in overall inference time without sacrificing training accuracy, and enables prediction of novel classes.
Forget the softmax bottleneck: this method achieves order-of-magnitude speedups in multi-class classification by structuring the latent space with predefined vector systems, enabling O(1) label prediction.
Label prediction in neural networks (NNs) has O(n) complexity proportional to the number of classes. This holds true for classification using fully connected layers and cosine similarity with some set of class prototypes. In this paper we show that if NN latent space (LS) geometry is known and possesses specific properties, label prediction complexity can be significantly reduced. This is achieved by associating label prediction with the O(1) complexity closest cluster center search in a vector system used as target for latent space configuration (LSC). The proposed method only requires finding indexes of several largest and lowest values in the embedding vector making it extremely computationally efficient. We show that the proposed method does not change NN training accuracy computational results. We also measure the time required by different computational stages of NN inference and label prediction on multiple datasets. The experiments show that the proposed method allows to achieve up to 11.6 times overall acceleration over conventional methods. Furthermore, the proposed method has unique properties which allow to predict the existence of new classes.