Please use this identifier to cite or link to this item:
Full metadata record
DC FieldValueLanguage
dc.contributor.authorDong, Renzhien_US
dc.identifier.citationDong, R. (2022). Contrastive learning for heterogeneous graph neural networks. Master's thesis, Nanyang Technological University, Singapore.
dc.description.abstractThe challenge of node classification in a heterogeneous graph has generated a lot of research interests in recent years. HeCo, as a novel and popular contrastive learning-based model, performs as a leading method in this field. Therefore, the technical details of HeCo model is reviewed and re-implemented in this dissertation, and some of the components were modified in order to better understand HeCo model and to explore possible new ideas based on it. This report summarizes the existing methods of data augmentation for contrastive learning, and a data augmentation method is applied. Substituting HeCo’s cross-view mechanism with the summarized method we got a revised new model called HeCo*. Then a control experiment is designed and carried out to show the performance of cross-view mechanism in HeCo, it demonstrated that HeCo does outperform the traditional contrastive learning methods in node classification and node clustering significantly. Also, a re-implementation of node clustering is done, a line chart of the variation of NMI and ARI is obtained to better illustrate the training process and the performance of HeCo and HeCo* in downstream applications.en_US
dc.publisherNanyang Technological Universityen_US
dc.subjectEngineering::Electrical and electronic engineeringen_US
dc.titleContrastive learning for heterogeneous graph neural networksen_US
dc.typeThesis-Master by Courseworken_US
dc.contributor.supervisorLihui Chenen_US
dc.contributor.schoolSchool of Electrical and Electronic Engineeringen_US
dc.description.degreeMaster of Science (Signal Processing)en_US
item.fulltextWith Fulltext-
Appears in Collections:EEE Theses
Files in This Item:
File Description SizeFormat 
dissertation-Dong Renzhi.pdf
  Restricted Access
1.8 MBAdobe PDFView/Open

Page view(s)

Updated on Apr 10, 2024


Updated on Apr 10, 2024

Google ScholarTM


Items in DR-NTU are protected by copyright, with all rights reserved, unless otherwise indicated.