Please use this identifier to cite or link to this item: https://hdl.handle.net/10356/164475
Title: Contrastive learning for heterogeneous graph neural networks
Authors: Dong, Renzhi
Keywords: Engineering::Electrical and electronic engineering
Issue Date: 2022
Publisher: Nanyang Technological University
Source: Dong, R. (2022). Contrastive learning for heterogeneous graph neural networks. Master's thesis, Nanyang Technological University, Singapore. https://hdl.handle.net/10356/164475
Abstract: The 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.
URI: https://hdl.handle.net/10356/164475
Schools: School of Electrical and Electronic Engineering 
Fulltext Permission: restricted
Fulltext Availability: With 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)

131
Updated on Feb 25, 2024

Download(s)

2
Updated on Feb 25, 2024

Google ScholarTM

Check

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