The Recommendation Algorithm Based on Improved Conditional Variational Autoencoder and Constrained Probabilistic Matrix Factorization

An improved recommendation algorithm based on Conditional Variational Autoencoder (CVAE) and Constrained Probabilistic Matrix Factorization (CPMF) is proposed to address the issues of poor recommendation performance in traditional user-based collaborative filtering algorithms caused by data sparsity...

Full description

Bibliographic Details
Main Authors: Yunfei Zhang, Hongzhen Xu, Xiaojun Yu
Format: Article
Language:English
Published: MDPI AG 2023-11-01
Series:Applied Sciences
Subjects:
Online Access:https://www.mdpi.com/2076-3417/13/21/12027
_version_ 1797632135274692608
author Yunfei Zhang
Hongzhen Xu
Xiaojun Yu
author_facet Yunfei Zhang
Hongzhen Xu
Xiaojun Yu
author_sort Yunfei Zhang
collection DOAJ
description An improved recommendation algorithm based on Conditional Variational Autoencoder (CVAE) and Constrained Probabilistic Matrix Factorization (CPMF) is proposed to address the issues of poor recommendation performance in traditional user-based collaborative filtering algorithms caused by data sparsity and suboptimal feature extraction. Firstly, in the data preprocessing stage, a hidden layer is added to CVAE, and random noise is introduced into the hidden layer to constrain the data features, thereby obtaining more accurate latent features and improving the model’s robustness and generative capability. Secondly, the category of items is incorporated as auxiliary information in CVAE to supervise the encoding and decoding of item data. By learning the distribution characteristics of the data, missing values in the rating data can be effectively reconstructed, thereby reducing the sparsity of the rating matrix. Subsequently, the reconstructed data is processed using CPMF, which optimizes the feature extraction performance by imposing constraints on user features. Finally, the prediction rating of a user for an item can be obtained through the matrix product of user and item feature matrices. Experimental results on the MovieLens-100K and MovieLens-1M datasets demonstrate the effectiveness and superiority of the proposed algorithm over four comparative algorithms, as it exhibits significant advantages in terms of root mean square error and mean absolute error metrics.
first_indexed 2024-03-11T11:33:35Z
format Article
id doaj.art-626d905e644d47f09bec7f9431b82667
institution Directory Open Access Journal
issn 2076-3417
language English
last_indexed 2024-03-11T11:33:35Z
publishDate 2023-11-01
publisher MDPI AG
record_format Article
series Applied Sciences
spelling doaj.art-626d905e644d47f09bec7f9431b826672023-11-10T14:59:34ZengMDPI AGApplied Sciences2076-34172023-11-0113211202710.3390/app132112027The Recommendation Algorithm Based on Improved Conditional Variational Autoencoder and Constrained Probabilistic Matrix FactorizationYunfei Zhang0Hongzhen Xu1Xiaojun Yu2School of Information Engineering, East China University of Technology, Nanchang 330013, ChinaSchool of Information Engineering, East China University of Technology, Nanchang 330013, ChinaSchool of Software, East China University of Technology, Nanchang 330013, ChinaAn improved recommendation algorithm based on Conditional Variational Autoencoder (CVAE) and Constrained Probabilistic Matrix Factorization (CPMF) is proposed to address the issues of poor recommendation performance in traditional user-based collaborative filtering algorithms caused by data sparsity and suboptimal feature extraction. Firstly, in the data preprocessing stage, a hidden layer is added to CVAE, and random noise is introduced into the hidden layer to constrain the data features, thereby obtaining more accurate latent features and improving the model’s robustness and generative capability. Secondly, the category of items is incorporated as auxiliary information in CVAE to supervise the encoding and decoding of item data. By learning the distribution characteristics of the data, missing values in the rating data can be effectively reconstructed, thereby reducing the sparsity of the rating matrix. Subsequently, the reconstructed data is processed using CPMF, which optimizes the feature extraction performance by imposing constraints on user features. Finally, the prediction rating of a user for an item can be obtained through the matrix product of user and item feature matrices. Experimental results on the MovieLens-100K and MovieLens-1M datasets demonstrate the effectiveness and superiority of the proposed algorithm over four comparative algorithms, as it exhibits significant advantages in terms of root mean square error and mean absolute error metrics.https://www.mdpi.com/2076-3417/13/21/12027collaborative filteringconditional variational autoencoderconstrained probabilistic matrix factorizationauxiliary informationfeature matric
spellingShingle Yunfei Zhang
Hongzhen Xu
Xiaojun Yu
The Recommendation Algorithm Based on Improved Conditional Variational Autoencoder and Constrained Probabilistic Matrix Factorization
Applied Sciences
collaborative filtering
conditional variational autoencoder
constrained probabilistic matrix factorization
auxiliary information
feature matric
title The Recommendation Algorithm Based on Improved Conditional Variational Autoencoder and Constrained Probabilistic Matrix Factorization
title_full The Recommendation Algorithm Based on Improved Conditional Variational Autoencoder and Constrained Probabilistic Matrix Factorization
title_fullStr The Recommendation Algorithm Based on Improved Conditional Variational Autoencoder and Constrained Probabilistic Matrix Factorization
title_full_unstemmed The Recommendation Algorithm Based on Improved Conditional Variational Autoencoder and Constrained Probabilistic Matrix Factorization
title_short The Recommendation Algorithm Based on Improved Conditional Variational Autoencoder and Constrained Probabilistic Matrix Factorization
title_sort recommendation algorithm based on improved conditional variational autoencoder and constrained probabilistic matrix factorization
topic collaborative filtering
conditional variational autoencoder
constrained probabilistic matrix factorization
auxiliary information
feature matric
url https://www.mdpi.com/2076-3417/13/21/12027
work_keys_str_mv AT yunfeizhang therecommendationalgorithmbasedonimprovedconditionalvariationalautoencoderandconstrainedprobabilisticmatrixfactorization
AT hongzhenxu therecommendationalgorithmbasedonimprovedconditionalvariationalautoencoderandconstrainedprobabilisticmatrixfactorization
AT xiaojunyu therecommendationalgorithmbasedonimprovedconditionalvariationalautoencoderandconstrainedprobabilisticmatrixfactorization
AT yunfeizhang recommendationalgorithmbasedonimprovedconditionalvariationalautoencoderandconstrainedprobabilisticmatrixfactorization
AT hongzhenxu recommendationalgorithmbasedonimprovedconditionalvariationalautoencoderandconstrainedprobabilisticmatrixfactorization
AT xiaojunyu recommendationalgorithmbasedonimprovedconditionalvariationalautoencoderandconstrainedprobabilisticmatrixfactorization