Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

PSL: Rethinking and Improving Softmax Loss from Pairwise Perspective for Recommendation

About

Softmax Loss (SL) is widely applied in recommender systems (RS) and has demonstrated effectiveness. This work analyzes SL from a pairwise perspective, revealing two significant limitations: 1) the relationship between SL and conventional ranking metrics like DCG is not sufficiently tight; 2) SL is highly sensitive to false negative instances. Our analysis indicates that these limitations are primarily due to the use of the exponential function. To address these issues, this work extends SL to a new family of loss functions, termed Pairwise Softmax Loss (PSL), which replaces the exponential function in SL with other appropriate activation functions. While the revision is minimal, we highlight three merits of PSL: 1) it serves as a tighter surrogate for DCG with suitable activation functions; 2) it better balances data contributions; and 3) it acts as a specific BPR loss enhanced by Distributionally Robust Optimization (DRO). We further validate the effectiveness and robustness of PSL through empirical experiments. The code is available at https://github.com/Tiny-Snow/IR-Benchmark.

Weiqin Yang, Jiawei Chen, Xin Xin, Sheng Zhou, Binbin Hu, Yan Feng, Chun Chen, Can Wang• 2024

Related benchmarks

TaskDatasetResultRank
RecommendationGowalla
Recall@200.2089
100
RecommendationAmazon-Movie IID (test)
Recall@2013.13
42
RecommendationAmazon-Book IID (test)
Recall@200.1575
33
RecommendationBeauty
P@201.74
30
RecommendationElectronics
Precision@200.66
30
RecommendationElectronics
MRR@201.83
30
RecommendationGames
Precision@202.1
30
Top-K RecommendationBeauty
MRR@200.0469
30
Top-K RecommendationGames
MRR@205.14
30
RecommendationAmazon-Electronic IID (test)
Recall@200.0838
24
Showing 10 of 24 rows

Other info

Code

Follow for update