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...
Main Authors: | , , |
---|---|
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 |