English
Related papers

Related papers: Scalable Cross-Entropy Loss for Sequential Recomme…

200 papers

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

Supervised Fine-Tuning (SFT) followed by Reinforcement Learning (RL) has emerged as the standard post-training paradigm for large language models (LLMs). However, the conventional SFT process, driven by Cross-Entropy (CE) loss, often…

Computation and Language · Computer Science 2026-02-10 Yijie Chen , Yijin Liu , Fandong Meng

Click-through rate (CTR) prediction is a crucial area of research in online advertising. While binary cross entropy (BCE) has been widely used as the optimization objective for treating CTR prediction as a binary classification problem,…

Information Retrieval · Computer Science 2024-07-09 Zhutian Lin , Junwei Pan , Shangyu Zhang , Ximei Wang , Xi Xiao , Shudong Huang , Lei Xiao , Jie Jiang

The Improved Cross-Entropy (ICE) method is a powerful tool for estimating failure probabilities in reliability analysis. Its core idea is to approximate the optimal importance-sampling density by minimizing the forward Kullback-Leibler…

Numerical Analysis · Mathematics 2025-09-10 Zhiwei Gao , George Karniadakis

Model calibration aims to align confidence with prediction correctness. The Cross-Entropy (CE) loss is widely used for calibrator training, which enforces the model to increase confidence on the ground truth class. However, we find the CE…

Computer Vision and Pattern Recognition · Computer Science 2025-02-13 Yuchi Liu , Lei Wang , Yuli Zou , James Zou , Liang Zheng

Despite their success, large pretrained vision models remain vulnerable to catastrophic forgetting when adapted to new tasks in class-incremental settings. Parameter-efficient fine-tuning (PEFT) alleviates this by restricting trainable…

Machine Learning · Computer Science 2026-02-17 Yaqian Zhang , Bernhard Pfahringer , Eibe Frank , Albert Bifet

In this work we present a new method of black-box optimization and constraint satisfaction. Existing algorithms that have attempted to solve this problem are unable to consider multiple modes, and are not able to adapt to changes in…

Machine Learning · Computer Science 2020-02-19 Kourosh Hakhamaneshi , Keertana Settaluri , Pieter Abbeel , Vladimir Stojanovic

Cross-entropy (CE) loss is the de-facto standard for training deep neural networks to perform classification. However, CE-trained deep neural networks struggle with robustness and generalisation issues. To alleviate these issues, we propose…

Machine Learning · Computer Science 2025-01-22 Michael W. Spratling , Heiko H. Schütt

Time series forecasting is an important task that involves analyzing temporal dependencies and underlying patterns (such as trends, cyclicality, and seasonality) in historical data to predict future values or trends. Current deep…

Machine Learning · Computer Science 2025-12-01 Jieting Wang , Huimei Shi , Feijiang Li , Xiaolei Shang

Sequential recommendation (SR), which encodes user activity to predict the next action, has emerged as a widely adopted strategy in developing commercial personalized recommendation systems. A critical component of modern SR models is the…

Information Retrieval · Computer Science 2025-02-25 Jun Yuan , Guohao Cai , Zhenhua Dong

The inability to guarantee robustness is one of the major obstacles to the application of deep learning models in security-demanding domains. We identify that the most commonly used cross-entropy (CE) loss does not guarantee robust boundary…

Machine Learning · Computer Science 2023-02-21 Weiwei Gao , Dazhi Zhang , Yao Li , Zhichang Guo , Ovanes Petrosian

A novel method for tackling the problem of imbalanced data in medical image segmentation is proposed in this work. In balanced cross entropy (CE) loss, which is a type of weighted CE loss, the weight assigned to each class is the in-verse…

Image and Video Processing · Electrical Eng. & Systems 2024-12-10 Seyed Mohsen Hosseini , Mahdieh Soleymani Baghshah

State-of-the-art natural language understanding classification models follow two-stages: pre-training a large language model on an auxiliary task, and then fine-tuning the model on a task-specific labeled dataset using cross-entropy loss.…

Computation and Language · Computer Science 2021-04-06 Beliz Gunel , Jingfei Du , Alexis Conneau , Ves Stoyanov

The cross-entropy (CE) method is simple and versatile technique for optimization, based on Kullback-Leibler (or cross-entropy) minimization. The method can be applied to a wide range of optimization tasks, including continuous, discrete,…

Computation · Statistics 2015-03-09 Tim Benham , Qibin Duan , Dirk P. Kroese , Benoit Liquet

Nearly all practical neural models for classification are trained using cross-entropy loss. Yet this ubiquitous choice is supported by little theoretical or empirical evidence. Recent work (Hui & Belkin, 2020) suggests that training using…

Machine Learning · Computer Science 2023-02-09 Like Hui , Mikhail Belkin , Stephen Wright

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

Achieving backward compatibility when rolling out new models can highly reduce costs or even bypass feature re-encoding of existing gallery images for in-production visual retrieval systems. Previous related works usually leverage losses…

Computer Vision and Pattern Recognition · Computer Science 2021-08-05 Qiang Meng , Chixiang Zhang , Xiaoqiang Xu , Feng Zhou

Superconductor electronics (SCE) is a promising complementary and beyond CMOS technology. However, despite its practical benefits, the realization of SCE logic faces a significant challenge due to the absence of dense and scalable…

Superconductivity · Physics 2024-12-05 Mustafa Altay Karamuftuoglu , Beyza Zeynep Ucpinar , Sasan Razmkhah , Massoud Pedram

Deep Learning Recommendation Models (DLRMs) play a crucial role in delivering personalized content across web applications such as social networking and video streaming. However, with improvements in performance, the parameter size of DLRMs…

Hardware Architecture · Computer Science 2025-04-02 Jinho Yang , Ji-Hoon Kim , Joo-Young Kim

We study the impact of different loss functions on lesion segmentation from medical images. Although the Cross-Entropy (CE) loss is the most popular option when dealing with natural images, for biomedical image segmentation the soft Dice…

Computer Vision and Pattern Recognition · Computer Science 2022-09-15 Adrian Galdran , Gustavo Carneiro , Miguel Ángel González Ballester