Ergun Biçici. Power Loss Function in Neural Networks for Predicting Click-Through Rate. In Proc. of the 17th ACM Conference on Recommender Systems (RecSys), Singapore, September 2023. URL: https://dl.acm.org/doi/10.1145/3604915.3610658, https://www.growkudos.com/publications/10.1145%252F3604915.3610658/reader
Loss functions guide machine learning models towards concentrating on the error most important to improve upon. We introduce power loss functions for neural networks and apply them on imbalanced click-through rate datasets. Power loss functions decrease the loss for confident predictions and increase the loss for error-prone predictions. They improve both AUC and F1 and produce better calibrated results. We obtain improvements in the results on four different classifiers and on two different datasets. We obtain significant improvements in AUC that reach 0.44% for DeepFM on the Avazu dataset.
No comments:
Post a Comment