Temporally and Distributionally Robust Optimization for Cold-Start Recommendation

Authors

  • Xinyu Lin National University of Singapore
  • Wenjie Wang National University of Singapore
  • Jujia Zhao National University of Singapore
  • Yongqi Li The Hong Kong Polytechnic University
  • Fuli Feng MoE Key Laboratory of Brain-inspired Intelligent Perception and Cognition, University of Science and Technology of China
  • Tat-Seng Chua National University of Singapore

DOI:

https://doi.org/10.1609/aaai.v38i8.28721

Keywords:

DMKM: Recommender Systems, DMKM: Applications

Abstract

Collaborative Filtering (CF) recommender models highly depend on user-item interactions to learn CF representations, thus falling short of recommending cold-start items. To address this issue, prior studies mainly introduce item features (e.g., thumbnails) for cold-start item recommendation. They learn a feature extractor on warm-start items to align feature representations with interactions, and then leverage the feature extractor to extract the feature representations of cold-start items for interaction prediction. Unfortunately, the features of cold-start items, especially the popular ones, tend to diverge from those of warm-start ones due to temporal feature shifts, preventing the feature extractor from accurately learning feature representations of cold-start items. To alleviate the impact of temporal feature shifts, we consider using Distributionally Robust Optimization (DRO) to enhance the generation ability of the feature extractor. Nonetheless, existing DRO methods face an inconsistency issue: the worse-case warm-start items emphasized during DRO training might not align well with the cold-start item distribution. To capture the temporal feature shifts and combat this inconsistency issue, we propose a novel temporal DRO with new optimization objectives, namely, 1) to integrate a worst-case factor to improve the worst-case performance, and 2) to devise a shifting factor to capture the shifting trend of item features and enhance the optimization of the potentially popular groups in cold-start items. Substantial experiments on three real-world datasets validate the superiority of our temporal DRO in enhancing the generalization ability of cold-start recommender models.

Published

2024-03-24

How to Cite

Lin, X., Wang, W., Zhao, J., Li, Y., Feng, F., & Chua, T.-S. (2024). Temporally and Distributionally Robust Optimization for Cold-Start Recommendation. Proceedings of the AAAI Conference on Artificial Intelligence, 38(8), 8750-8758. https://doi.org/10.1609/aaai.v38i8.28721

Issue

Section

AAAI Technical Track on Data Mining & Knowledge Management