DGRM: Diffusion-GAN recommendation model to alleviate the mode collapse problem in sparse environments

D Jiangzhou, W Songli, Y Jianmei, J Lianghao… - Pattern Recognition, 2024 - Elsevier
D Jiangzhou, W Songli, Y Jianmei, J Lianghao, W Yong
Pattern Recognition, 2024Elsevier
Generative adversarial network (GAN) has been widely adopted in recommender systems
(RSs) to improve the recommendation accuracy. However, existing GAN-based models
often suffer from the mode collapse problem in sparse environments and fail to adequately
capture the complexity of user preferences and behaviors, which affects recommendation
performance. To address these issues, we introduce a diffusion model (DM) into the GAN
framework, proposing an efficient Diffusion-GAN recommendation model (DGRM) to achieve …
Abstract
Generative adversarial network (GAN) has been widely adopted in recommender systems (RSs) to improve the recommendation accuracy. However, existing GAN-based models often suffer from the mode collapse problem in sparse environments and fail to adequately capture the complexity of user preferences and behaviors, which affects recommendation performance. To address these issues, we introduce a diffusion model (DM) into the GAN framework, proposing an efficient Diffusion-GAN recommendation model (DGRM) to achieve mutual enhancement between the two generative models. This model first utilizes the forward process of DM to generate conditional vectors that guide the training of the GAN generator. Subsequently, the backward process of DM assists the GAN discriminator using Wasserstein distance during adversarial training. The Wasserstein distance is adopted to solve the asymmetry of Kullback-Leibler (KL) divergence as a loss function in traditional GANs. Experiments on multiple datasets demonstrate that the proposed model effectively alleviates mode collapse and surpasses other state-of-the-art (SOTA) methods in various evaluation metrics.
Elsevier
以上显示的是最相近的搜索结果。 查看全部搜索结果

Google学术搜索按钮

example.edu/paper.pdf
搜索
获取 PDF 文件
引用
References