TY - JOUR
T1 - Adversarial Auto-encoder Domain Adaptation for Cold-start Recommendation with Positive and Negative Hypergraphs
AU - Wu, Hanrui
AU - Long, Jinyi
AU - Li, Nuosi
AU - Yu, Dahai
AU - Ng, Michael K.
N1 - H. Wu’s work was supported by the Fundamental Research Funds for the Central Universities (Grant No. 21622326). J. Long’s work was supported by funding from the National Natural Science Foundation of China (Grant No. 61773179), the Outstanding Youth Project of Guangdong Natural Science Foundation of China (Grant No. 2021B1515020076), the Guangdong Provincial Natural Science Foundation of China (Grant No. 2019A1515012175), the Guangdong Provincial Key Laboratory of Traditional Chinese Medicine Informatization (Grant No. 2021B1212040007), and the Fundamental Research Funds for Central Universities. M. Ng’s work was supported by Hong Kong Research Grant Council GRF Grants No. 12300218, No. 12300519, No. 17201020, No. 17300021, No. C1013-21GF, No. C7004-21GF, and Joint NSFC-RGC N-HKU76921. The paper is supported by HKU-TCL Joint Research Centre for Artificial Intelligence, Hong Kong.
Publisher Copyright:
© 2022 Association for Computing Machinery.
PY - 2023/4
Y1 - 2023/4
N2 - This article presents a novel model named Adversarial Auto-encoder Domain Adaptation to handle the recommendation problem under cold-start settings. Specifically, we divide the hypergraph into two hypergraphs, i.e., a positive hypergraph and a negative one. Below, we adopt the cold-start user recommendation for illustration. After achieving positive and negative hypergraphs, we apply hypergraph auto-encoders to them to obtain positive and negative embeddings of warm users and items. Additionally, we employ a multi-layer perceptron to get warm and cold-start user embeddings called regular embeddings. Subsequently, for warm users, we assign positive and negative pseudo-labels to their positive and negative embeddings, respectively, and treat their positive and regular embeddings as the source and target domain data, respectively. Then, we develop a matching discriminator to jointly minimize the classification loss of the positive and negative warm user embeddings and the distribution gap between the positive and regular warm user embeddings. In this way, warm users' positive and regular embeddings are connected. Since the positive hypergraph maintains the relations between positive warm user and item embeddings, and the regular warm and cold-start user embeddings follow a similar distribution, the regular cold-start user embedding and positive item embedding are bridged to discover their relationship. The proposed model can be easily extended to handle the cold-start item recommendation by changing inputs. We perform extensive experiments on real-world datasets for both cold-start user and cold-start item recommendations. Promising results in terms of precision, recall, normalized discounted cumulative gain, and hit rate verify the effectiveness of the proposed method.
AB - This article presents a novel model named Adversarial Auto-encoder Domain Adaptation to handle the recommendation problem under cold-start settings. Specifically, we divide the hypergraph into two hypergraphs, i.e., a positive hypergraph and a negative one. Below, we adopt the cold-start user recommendation for illustration. After achieving positive and negative hypergraphs, we apply hypergraph auto-encoders to them to obtain positive and negative embeddings of warm users and items. Additionally, we employ a multi-layer perceptron to get warm and cold-start user embeddings called regular embeddings. Subsequently, for warm users, we assign positive and negative pseudo-labels to their positive and negative embeddings, respectively, and treat their positive and regular embeddings as the source and target domain data, respectively. Then, we develop a matching discriminator to jointly minimize the classification loss of the positive and negative warm user embeddings and the distribution gap between the positive and regular warm user embeddings. In this way, warm users' positive and regular embeddings are connected. Since the positive hypergraph maintains the relations between positive warm user and item embeddings, and the regular warm and cold-start user embeddings follow a similar distribution, the regular cold-start user embedding and positive item embedding are bridged to discover their relationship. The proposed model can be easily extended to handle the cold-start item recommendation by changing inputs. We perform extensive experiments on real-world datasets for both cold-start user and cold-start item recommendations. Promising results in terms of precision, recall, normalized discounted cumulative gain, and hit rate verify the effectiveness of the proposed method.
KW - adversarial learning
KW - auto-encoder
KW - cold-start recommendation
KW - domain adaptation
KW - hypergraph
KW - Recommendation systems
UR - http://www.scopus.com/inward/record.url?scp=85159284804&partnerID=8YFLogxK
U2 - 10.1145/3544105
DO - 10.1145/3544105
M3 - Journal article
AN - SCOPUS:85159284804
SN - 1046-8188
VL - 41
SP - 1
EP - 25
JO - ACM Transactions on Information Systems
JF - ACM Transactions on Information Systems
IS - 2
M1 - 33
ER -