English
Related papers

Related papers: SimCE: Simplifying Cross-Entropy Loss for Collabor…

200 papers

In Multi-Label Learning (MLL), it is extremely challenging to accurately annotate every appearing object due to expensive costs and limited knowledge. When facing such a challenge, a more practical and cheaper alternative should be Single…

Machine Learning · Computer Science 2024-06-11 Xiang Li , Xinrui Wang , Songcan Chen

As language models grow ever larger, so do their vocabularies. This has shifted the memory footprint of LLMs during training disproportionately to one single layer: the cross-entropy in the loss computation. Cross-entropy builds up a logit…

Machine Learning · Computer Science 2025-03-12 Erik Wijmans , Brody Huval , Alexander Hertzberg , Vladlen Koltun , Philipp Krähenbühl

The cross entropy loss is widely used due to its effectiveness and solid theoretical grounding. However, as training progresses, the loss tends to focus on hard to classify samples, which may prevent the network from obtaining gains in…

Machine Learning · Computer Science 2021-09-14 Barak Battash , Lior Wolf , Tamir Hazan

Inspired by the success of contrastive learning, we systematically examine recommendation losses, including listwise (softmax), pairwise (BPR), and pointwise (MSE and CCL) losses. In this endeavor, we introduce InfoNCE+, an optimized…

Artificial Intelligence · Computer Science 2024-11-05 Dong Li , Ruoming Jin , Bin Ren

Pairwise learning strategies are prevalent for optimizing recommendation models on implicit feedback data, which usually learns user preference by discriminating between positive (i.e., clicked by a user) and negative items (i.e., obtained…

Information Retrieval · Computer Science 2023-02-17 Xiao Chen , Wenqi Fan , Jingfan Chen , Haochen Liu , Zitao Liu , Zhaoxiang Zhang , Qing Li

A large catalogue size is one of the central challenges in training recommendation models: a large number of items makes them memory and computationally inefficient to compute scores for all items during training, forcing these models to…

Information Retrieval · Computer Science 2023-08-15 Aleksandr Petrov , Craig Macdonald

We introduce a new loss function TripleEntropy, to improve classification performance for fine-tuning general knowledge pre-trained language models based on cross-entropy and SoftTriple loss. This loss function can improve the robust…

Computation and Language · Computer Science 2022-11-28 Witold Sosnowski , Anna Wroblewska , Piotr Gawrysiak

This study presents a comparative analysis of two objective functions, Mean Squared Error (MSE) and Softmax Cross-Entropy (SCE) for neural network classification tasks. While SCE combined with softmax activation is the conventional choice…

Machine Learning · Computer Science 2024-11-19 Kanishka Tyagi , Chinmay Rane , Ketaki Vaidya , Jeshwanth Challgundla , Soumitro Swapan Auddy , Michael Manry

Recent advancements in learning algorithms have demonstrated that the sharpness of the loss surface is an effective measure for improving the generalization gap. Building upon this concept, Sharpness-Aware Minimization (SAM) was proposed to…

Machine Learning · Computer Science 2024-06-21 Tanapat Ratchatorn , Masayuki Tanaka

The goal of a recommendation system is to model the relevance between each user and each item through the user-item interaction history, so that maximize the positive samples score and minimize negative samples. Currently, two popular loss…

Information Retrieval · Computer Science 2022-07-08 Chun Yang , Shicai Fan

One common loss function in neural network classification tasks is Categorical Cross Entropy (CCE), which punishes all misclassifications equally. However, classes often have an inherent structure. For instance, classifying an image of a…

Machine Learning · Computer Science 2020-03-09 Konstantin Kobs , Michael Steininger , Albin Zehe , Florian Lautenschlager , Andreas Hotho

Traditional empirical risk minimization (ERM) for semantic segmentation can disproportionately advantage or disadvantage certain target classes in favor of an (unfair but) improved overall performance. Inspired by the recently introduced…

Computer Vision and Pattern Recognition · Computer Science 2021-03-29 Attila Szabo , Hadi Jamali-Rad , Siva-Datta Mannava

When training classification models, it expects that the learned features are compact within classes, and can well separate different classes. As the dominant loss function for training classification models, minimizing cross-entropy (CE)…

Machine Learning · Computer Science 2025-05-12 Qiufu Li , Huibin Xiao , Linlin Shen

Contrastive learning-based recommendation algorithms have significantly advanced the field of self-supervised recommendation, particularly with BPR as a representative ranking prediction task that dominates implicit collaborative filtering.…

Information Retrieval · Computer Science 2024-03-13 Shipeng Song , Bin Liu , Fei Teng , Tianrui Li

This paper investigates the deep learning optimization problem with softmax cross-entropy loss. We propose a layer separation strategy to alleviate the strong nonconvexity encountered during training deep networks. For cross-entropy models…

Machine Learning · Computer Science 2026-04-28 Yaru Liu , Michael K. Ng , Yiqi Gu

Graph collaborative filtering (GCF) is a dominant paradigm in recommender systems, where contrastive learning (CL) objectives such as the Sampled Softmax (SSM) loss are widely used for optimization. However, it remains unclear how CL…

Information Retrieval · Computer Science 2026-05-26 Geon Lee , Sunwoo Kim , Kyungho Kim , Kijung Shin

Learning discriminative representations is a central goal of supervised deep learning. While cross-entropy (CE) remains the dominant objective for classification, it does not explicitly enforce desirable geometric properties in the…

Machine Learning · Computer Science 2026-04-13 Matheus Vinícius Todescato , Joel Luís Carbonera

Bayesian Personalized Ranking (BPR), a collaborative filtering approach based on matrix factorization, frequently serves as a benchmark for recommender systems research. However, numerous studies often overlook the nuances of BPR…

Information Retrieval · Computer Science 2024-10-21 Aleksandr Milogradskii , Oleg Lashinin , Alexander P , Marina Ananyeva , Sergey Kolesnikov

Scalability is a major challenge in modern recommender systems. In sequential recommendations, full Cross-Entropy (CE) loss achieves state-of-the-art recommendation quality but consumes excessive GPU memory with large item catalogs,…

Information Retrieval · Computer Science 2024-08-15 Danil Gusak , Gleb Mezentsev , Ivan Oseledets , Evgeny Frolov

Contrastive learning has been gradually applied to learn high-quality unsupervised sentence embedding. Among the previous un-supervised methods, the latest state-of-the-art method, as far as we know, is unsupervised SimCSE (unsup-SimCSE).…

Computation and Language · Computer Science 2022-09-13 Xing Wu , Chaochen Gao , Yipeng Su , Jizhong Han , Zhongyuan Wang , Songlin Hu