Guided Diffusion-based Counterfactual Augmentation for Robust Session-based Recommendation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Gupta, Muskan, Gupta, Priyanka, Vig, Lovekesh
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917821443735552
author Gupta, Muskan
Gupta, Priyanka
Vig, Lovekesh
author_facet Gupta, Muskan
Gupta, Priyanka
Vig, Lovekesh
contents Session-based recommendation (SR) models aim to recommend top-K items to a user, based on the user's behaviour during the current session. Several SR models are proposed in the literature, however,concerns have been raised about their susceptibility to inherent biases in the training data (observed data) such as popularity bias. SR models when trained on the biased training data may encounter performance challenges on out-of-distribution data in real-world scenarios. One way to mitigate popularity bias is counterfactual data augmentation. Compared to prior works that rely on generating data using SR models, we focus on utilizing the capabilities of state-of-the art diffusion models for generating counterfactual data. We propose a guided diffusion-based counterfactual augmentation framework for SR. Through a combination of offline and online experiments on a real-world and simulated dataset, respectively, we show that our approach performs significantly better than the baseline SR models and other state-of-the art augmentation frameworks. More importantly, our framework shows significant improvement on less popular target items, by achieving up to 20% gain in Recall and 13% gain in CTR on real-world and simulated datasets,respectively.
format Preprint
id arxiv_https___arxiv_org_abs_2410_21892
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Guided Diffusion-based Counterfactual Augmentation for Robust Session-based Recommendation
Gupta, Muskan
Gupta, Priyanka
Vig, Lovekesh
Information Retrieval
Session-based recommendation (SR) models aim to recommend top-K items to a user, based on the user's behaviour during the current session. Several SR models are proposed in the literature, however,concerns have been raised about their susceptibility to inherent biases in the training data (observed data) such as popularity bias. SR models when trained on the biased training data may encounter performance challenges on out-of-distribution data in real-world scenarios. One way to mitigate popularity bias is counterfactual data augmentation. Compared to prior works that rely on generating data using SR models, we focus on utilizing the capabilities of state-of-the art diffusion models for generating counterfactual data. We propose a guided diffusion-based counterfactual augmentation framework for SR. Through a combination of offline and online experiments on a real-world and simulated dataset, respectively, we show that our approach performs significantly better than the baseline SR models and other state-of-the art augmentation frameworks. More importantly, our framework shows significant improvement on less popular target items, by achieving up to 20% gain in Recall and 13% gain in CTR on real-world and simulated datasets,respectively.
title Guided Diffusion-based Counterfactual Augmentation for Robust Session-based Recommendation
topic Information Retrieval
url https://arxiv.org/abs/2410.21892