Hyperbolic Contrastive Learning with Model-Augmentation for Knowledge-aware Recommendation

Research output: Chapters, Conference Papers, Creative and Literary WorksRGC 32 - Refereed conference paper (with host publication)peer-review

1 Scopus Citations
View graph of relations

Related Research Unit(s)

Detail(s)

Original languageEnglish
Title of host publicationMachine Learning and Knowledge Discovery in Databases. Research Track
Subtitle of host publicationEuropean Conference, ECML PKDD 2024 - Proceedings
EditorsAlbert Bifet, Jesse Davis, Tomas Krilavičius, Meelis Kull, Eirini Ntoutsi, Indrė Žliobaitė
PublisherSpringer, Cham
Pages199–217
Number of pages19
VolumePart I
ISBN (electronic)978-3-031-70341-6
ISBN (print)978-3-031-70340-9
Publication statusPublished - 2024

Publication series

NameLecture Notes in Computer Science
Volume14941
ISSN (Print)0302-9743
ISSN (electronic)1611-3349

Conference

TitleEuropean Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases (ECML PKDD 2044)
PlaceLithuania
CityVilnius
Period9 - 13 September 2024

Abstract

Benefiting from the effectiveness of graph neural networks (GNNs) and contrastive learning, GNN-based contrastive learning has become mainstream for knowledge-aware recommendation. However, most existing contrastive learning-based methods have difficulties in effectively capturing the underlying hierarchical structure within user-item bipartite graphs and knowledge graphs. Moreover, they commonly generate positive samples for contrastive learning by perturbing the graph structure, which may lead to a shift in user preference learning. To overcome these limitations, we propose hyperbolic contrastive learning with modelaugmentation for knowledge-aware recommendation. To capture the intrinsic hierarchical graph structures, we first design a novel Lorentzian knowledge aggregation mechanism, which enables more effective representations of users and items. Then, we propose three model-level augmentation techniques to assist Hyperbolic contrastive learning. Different from the classical structure-level augmentation (e.g., edge dropping), the proposed model-augmentations can avoid preference shifts between the augmented positive pair. Finally, we conduct extensive experiments to demonstrate the superiority (maximum improvement of 11.03%) of proposed methods over existing baselines. c The Author(s), under exclusive license to Springer Nature Switzerland AG 2024.

Research Area(s)

  • Knowledge-aware recommendation, Model-augmentation, Hyperbolic space

Bibliographic Note

Information for this record is supplemented by the author(s) concerned.

Citation Format(s)

Hyperbolic Contrastive Learning with Model-Augmentation for Knowledge-aware Recommendation. / Sun, Shengyin; Ma, Chen.
Machine Learning and Knowledge Discovery in Databases. Research Track: European Conference, ECML PKDD 2024 - Proceedings. ed. / Albert Bifet; Jesse Davis; Tomas Krilavičius; Meelis Kull; Eirini Ntoutsi; Indrė Žliobaitė. Vol. Part I Springer, Cham, 2024. p. 199–217 (Lecture Notes in Computer Science; Vol. 14941).

Research output: Chapters, Conference Papers, Creative and Literary WorksRGC 32 - Refereed conference paper (with host publication)peer-review