mlsanityData to 14 Oct 2026

NDCG-Consistent Softmax Approximation with Accelerated Convergence

arXiv 2506.09454 ↗First posted 11 Jun 2025cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Ranking tasks constitute fundamental components of extreme similarity learning frameworks, where extremely large corpora of objects are modeled through relative similarity relationships adhering to predefined ordinal structures. Among various ranking surrogates, Softmax (SM) Loss has been widely adopted due to its natural capability to handle listwise ranking via global negative comparisons, along with its flexibility across diverse application scenarios. However, despite its effectiveness, SM Loss often suffers from significant computational overhead and scalability limitations when applied to large-scale object spaces. To address this challenge, we propose novel loss formulations that align directly with ranking metrics: the Ranking-Generalizable squared (RG2) Loss and the Ranking-Generalizable interactive (RG^×) Loss, both derived through Taylor expansions of the SM Loss. Notably, RG2 reveals the intrinsic mechanisms underlying weighted squared losses (WSL) in ranking methods and uncovers fundamental connections between sampling-based and non-sampling-based loss paradigms. Furthermore, we integrate the proposed RG losses with the highly efficient Alternating Least Squares (ALS) optimization method, providing both generalization guarantees and convergence rate analyses. Empirical evaluations on real-world datasets demonstrate that our approach achieves comparable or superior ranking performance relative to SM Loss, while significantly accelerating convergence. This framework offers the similarity learning community both theoretical insights and practically efficient tools, with methodologies applicable to a broad range of tasks where balancing ranking quality and computational efficiency is essential.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 25 of its 65 references that are in the map. First posted on arXiv 11 Jun 2025.

Builds on

25 of 65 references.

References · 25

  1. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
  2. 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
  3. 2013Distributed Representations of Words and Phrases and their Compositionalitycited by 3,474
  4. 2015FaceNet: A Unified Embedding for Face Recognition and Clusteringcited by 2,849
  5. 2021SimCSE: Simple Contrastive Learning of Sentence Embeddingscited by 2,173
  6. 2018Self-Attentive Sequential Recommendationcited by 1,184
  7. 2016Optimization Methods for Large-Scale Machine Learningcited by 1,150
  8. 2017Neural Collaborative Filteringcited by 1,111
  9. 2019BERT4Rec: Sequential Recommendation with Bidirectional Encoder Representations from Transformercited by 964
  10. 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 724
  11. 2014SAGA: A Fast Incremental Gradient Method With Support for Non-Strongly Convex Composite Objectivescited by 402
  12. 2016From Softmax to Sparsemax: A Sparse Model of Attention and Multi-Label Classificationcited by 341
13 more
  1. 2018Variational Autoencoders for Collaborative Filteringcited by 310
  2. 2014Learning Fine-grained Image Similarity with Deep Rankingcited by 190
  3. 2017Non-convex Optimization for Machine Learningcited by 133
  4. 2018A no-regret generalization of hierarchical softmax to extreme multi-label classificationcited by 52
  5. 2022On the Effectiveness of Sampled Softmax Loss for Item Recommendationcited by 39
  6. 2019On the Consistency of Top-k Surrogate Lossescited by 35
  7. 2021Revisiting the Performance of iALS on Item Recommendation Benchmarkscited by 21
  8. 2021Item Recommendation from Implicit Feedbackcited by 18
  9. 2019An Alternative Cross Entropy Loss for Learning-to-Rankcited by 17
  10. 2016A Generic Coordinate Descent Framework for Learning from Implicit Feedbackcited by 16
  11. 2018Efficient Training on Very Large Corpora via Gramian Estimationcited by 9
  12. 2021Learning Explicit User Interest Boundary for Recommendationcited by 6
  13. 2022Cooperative Retriever and Ranker in Deep Recommenderscited by 6

Built on by

Not cited yet.