English
Related papers

Related papers: Exploring Alternatives to Softmax Function

200 papers

The softmax (also called softargmax) function is widely used in machine learning models to normalize real-valued scores into a probability distribution. To avoid floating-point overflow, the softmax function is conventionally implemented in…

Performance · Computer Science 2020-01-14 Marat Dukhan , Artsiom Ablavatski

Recent neural network and language models rely on softmax distributions with an extremely large number of categories. Since calculating the softmax normalizing constant in this context is prohibitively expensive, there is a growing…

Machine Learning · Statistics 2018-03-26 Francois Fagan , Garud Iyengar

State-of-the-art deep face recognition methods are mostly trained with a softmax-based multi-class classification framework. Despite being popular and effective, these methods still have a few shortcomings that limit empirical performance.…

Computer Vision and Pattern Recognition · Computer Science 2022-04-12 Yandong Wen , Weiyang Liu , Adrian Weller , Bhiksha Raj , Rita Singh

Many loss functions have been derived from cross-entropy loss functions such as large-margin softmax loss and focal loss. The large-margin softmax loss makes the classification more rigorous and prevents overfitting. The focal loss…

Computer Vision and Pattern Recognition · Computer Science 2023-02-10 Jiajie Chen

Within the framework of natural orbital functional theory, having a convenient representation of the occupation numbers and orbitals becomes critical for the computational performance of the calculations. Recognizing this, we propose an…

This paper questions whether the strong performance of softmax attention in transformers stems from producing a probability distribution over inputs. Instead, we argue that softmax's effectiveness lies in its implicit regularization of the…

Machine Learning · Computer Science 2026-03-16 Hemanth Saratchandran , Jianqiao Zheng , Yiping Ji , Wenbo Zhang , Simon Lucey

In state-of-the-art deep learning for object recognition, SoftMax and Sigmoid functions are most commonly employed as the predictor outputs. Such layers often produce overconfident predictions rather than proper probabilistic scores, which…

Computer Vision and Pattern Recognition · Computer Science 2022-05-13 Gledson Melotti , Cristiano Premebida , Jordan J. Bird , Diego R. Faria , Nuno Gonçalves

Face recognition is one of the most widely publicized feature in the devices today and hence represents an important problem that should be studied with the utmost priority. As per the recent trends, the Convolutional Neural Network (CNN)…

Computer Vision and Pattern Recognition · Computer Science 2019-11-07 Yash Srivastava , Vaishnav Murali , Shiv Ram Dubey

Many evaluation metrics can be used to assess the performance of models in binary classification tasks. However, most of them are derived from a confusion matrix in a non-differentiable form, making it very difficult to generate a…

Machine Learning · Computer Science 2024-05-24 Doheon Han , Nuno Moniz , Nitesh V Chawla

Deep embedding learning is expected to learn a metric space in which features have smaller maximal intra-class distance than minimal inter-class distance. In recent years, one research focus is to solve the open-set problem by…

Computer Vision and Pattern Recognition · Computer Science 2020-11-24 Hao Zhu , Yang Yuan , Guosheng Hu , Xiang Wu , Neil Robertson

A common practice in most of deep convolutional neural architectures is to employ fully-connected layers followed by Softmax activation to minimize cross-entropy loss for the sake of classification. Recent studies show that substitution or…

Machine Learning · Computer Science 2017-10-23 Arash Shahriari

Two-tower neural networks are a popular architecture for the retrieval stage in recommender systems. These models are typically trained with a softmax loss over the item catalog. However, in web-scale settings, the item catalog is often…

Information Retrieval · Computer Science 2025-07-29 Kirill Khrylchenko , Vladimir Baikalov , Sergei Makeev , Artem Matveev , Sergei Liamaev

Loss functions play a pivotal role in optimizing recommendation models. Among various loss functions, Softmax Loss (SL) and Cosine Contrastive Loss (CCL) are particularly effective. Their theoretical connections and differences warrant…

Information Retrieval · Computer Science 2025-06-19 Shengjia Zhang , Jiawei Chen , Changdong Li , Sheng Zhou , Qihao Shi , Yan Feng , Chun Chen , Can Wang

Dual-encoder (DE) models are widely used in retrieval tasks, most commonly studied on open QA benchmarks that are often characterized by multi-class and limited training data. In contrast, their performance in multi-label and data-rich…

Machine Learning · Computer Science 2024-03-19 Nilesh Gupta , Devvrit Khatri , Ankit S Rawat , Srinadh Bhojanapalli , Prateek Jain , Inderjit Dhillon

As hashing becomes an increasingly appealing technique for large-scale image retrieval, multi-label hashing is also attracting more attention for the ability to exploit multi-level semantic contents. In this paper, we propose a novel deep…

Computer Vision and Pattern Recognition · Computer Science 2021-02-03 Cheng Ma , Jiwen Lu , Jie Zhou

The quadratic computational complexity of softmax transformers has become a bottleneck in long-context scenarios. In contrast, linear attention model families provide a promising direction towards a more efficient sequential model. These…

Computation and Language · Computer Science 2026-02-04 Difan Deng , Andreas Bentzen Winje , Lukas Fehring , Marius Lindauer

This paper addresses deep face recognition (FR) problem under open-set protocol, where ideal face features are expected to have smaller maximal intra-class distance than minimal inter-class distance under a suitably chosen metric space.…

Computer Vision and Pattern Recognition · Computer Science 2018-01-31 Weiyang Liu , Yandong Wen , Zhiding Yu , Ming Li , Bhiksha Raj , Le Song

The softmax function is a fundamental component in deep learning. This study delves into the often-overlooked parameter within the softmax function, known as "temperature," providing novel insights into the practical and theoretical aspects…

Machine Learning · Computer Science 2025-03-03 Hao Xuan , Bokai Yang , Xingyu Li

In this paper, we focus on the separability of classes with the cross-entropy loss function for classification problems by theoretically analyzing the intra-class distance and inter-class distance (i.e. the distance between any two points…

Machine Learning · Computer Science 2019-09-17 Rudrajit Das , Subhasis Chaudhuri

Sequence-to-sequence models are a powerful workhorse of NLP. Most variants employ a softmax transformation in both their attention mechanism and output layer, leading to dense alignments and strictly positive output probabilities. This…

Computation and Language · Computer Science 2019-06-14 Ben Peters , Vlad Niculae , André F. T. Martins