DFM-GCN: A Multi-Task Learning Recommendation Based on a Deep Graph Neural Network
Among the inherent problems in recommendation systems are data sparseness and cold starts; the solutions to which lie in the introduction of knowledge graphs to improve the performance of the recommendation systems. The results in previous research, however, suffer from problems such as data compres...
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2022-02-01
|
| Series: | Mathematics |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2227-7390/10/5/721 |
| _version_ | 1797474529608466432 |
|---|---|
| author | Yan Xiao Congdong Li Vincenzo Liu |
| author_facet | Yan Xiao Congdong Li Vincenzo Liu |
| author_sort | Yan Xiao |
| collection | DOAJ |
| description | Among the inherent problems in recommendation systems are data sparseness and cold starts; the solutions to which lie in the introduction of knowledge graphs to improve the performance of the recommendation systems. The results in previous research, however, suffer from problems such as data compression, information damage, and insufficient learning. Therefore, a DeepFM Graph Convolutional Network (DFM-GCN) model was proposed to alleviate the above issues. The prediction of the click-through rate (CTR) is critical in recommendation systems where the task is to estimate the probability that a user will click on a recommended item. In many recommendation systems, the goal is to maximize the number of clicks so the items returned to a user can be ranked by an estimated CTR. The DFM-GCN model consists of three parts: the left part DeepFM is used to capture the interactive information between the users and items; the deep neural network is used in the middle to model the left and right parts; and the right one obtains a better item representation vector by the GCN. In an effort to verify the validity and precision of the model built in this research, and based on the public datasets ml1m-kg20m and ml1m-kg1m, a performance comparison experiment was designed. It used multiple comparison models and the MKR and FM_MKR algorithms as well as the DFM-GCN algorithm constructed in this paper. Having achieved a state-of-the-art performance, the experimental results of the AUC and f1 values verified by the CTR as well as the accuracy, recall, and f1 values of the top-k showed that the proposed approach was excellent and more effective when compared with different recommendation algorithms. |
| first_indexed | 2024-03-09T20:31:24Z |
| format | Article |
| id | doaj.art-b929ad3708cc42e5a9a2b23c2212a310 |
| institution | Directory Open Access Journal |
| issn | 2227-7390 |
| language | English |
| last_indexed | 2024-03-09T20:31:24Z |
| publishDate | 2022-02-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Mathematics |
| spelling | doaj.art-b929ad3708cc42e5a9a2b23c2212a3102023-11-23T23:22:39ZengMDPI AGMathematics2227-73902022-02-0110572110.3390/math10050721DFM-GCN: A Multi-Task Learning Recommendation Based on a Deep Graph Neural NetworkYan Xiao0Congdong Li1Vincenzo Liu2School of Business, Macau University of Science and Technology, Macao 999078, ChinaSchool of Business, Macau University of Science and Technology, Macao 999078, ChinaSchool of Business, Macau University of Science and Technology, Macao 999078, ChinaAmong the inherent problems in recommendation systems are data sparseness and cold starts; the solutions to which lie in the introduction of knowledge graphs to improve the performance of the recommendation systems. The results in previous research, however, suffer from problems such as data compression, information damage, and insufficient learning. Therefore, a DeepFM Graph Convolutional Network (DFM-GCN) model was proposed to alleviate the above issues. The prediction of the click-through rate (CTR) is critical in recommendation systems where the task is to estimate the probability that a user will click on a recommended item. In many recommendation systems, the goal is to maximize the number of clicks so the items returned to a user can be ranked by an estimated CTR. The DFM-GCN model consists of three parts: the left part DeepFM is used to capture the interactive information between the users and items; the deep neural network is used in the middle to model the left and right parts; and the right one obtains a better item representation vector by the GCN. In an effort to verify the validity and precision of the model built in this research, and based on the public datasets ml1m-kg20m and ml1m-kg1m, a performance comparison experiment was designed. It used multiple comparison models and the MKR and FM_MKR algorithms as well as the DFM-GCN algorithm constructed in this paper. Having achieved a state-of-the-art performance, the experimental results of the AUC and f1 values verified by the CTR as well as the accuracy, recall, and f1 values of the top-k showed that the proposed approach was excellent and more effective when compared with different recommendation algorithms.https://www.mdpi.com/2227-7390/10/5/721DeepFMGCNknowledge graphDNNrepresentation learningrecommendation systems |
| spellingShingle | Yan Xiao Congdong Li Vincenzo Liu DFM-GCN: A Multi-Task Learning Recommendation Based on a Deep Graph Neural Network Mathematics DeepFM GCN knowledge graph DNN representation learning recommendation systems |
| title | DFM-GCN: A Multi-Task Learning Recommendation Based on a Deep Graph Neural Network |
| title_full | DFM-GCN: A Multi-Task Learning Recommendation Based on a Deep Graph Neural Network |
| title_fullStr | DFM-GCN: A Multi-Task Learning Recommendation Based on a Deep Graph Neural Network |
| title_full_unstemmed | DFM-GCN: A Multi-Task Learning Recommendation Based on a Deep Graph Neural Network |
| title_short | DFM-GCN: A Multi-Task Learning Recommendation Based on a Deep Graph Neural Network |
| title_sort | dfm gcn a multi task learning recommendation based on a deep graph neural network |
| topic | DeepFM GCN knowledge graph DNN representation learning recommendation systems |
| url | https://www.mdpi.com/2227-7390/10/5/721 |
| work_keys_str_mv | AT yanxiao dfmgcnamultitasklearningrecommendationbasedonadeepgraphneuralnetwork AT congdongli dfmgcnamultitasklearningrecommendationbasedonadeepgraphneuralnetwork AT vincenzoliu dfmgcnamultitasklearningrecommendationbasedonadeepgraphneuralnetwork |