Accurate graph classification via two-staged contrastive curriculum learning

Copyright: © 2024 Shim et al. This is an open access article distributed under the terms of the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited..

Given a graph dataset, how can we generate meaningful graph representations that maximize classification accuracy? Learning representative graph embeddings is important for solving various real-world graph-based tasks. Graph contrastive learning aims to learn representations of graphs by capturing the relationship between the original graph and the augmented graph. However, previous contrastive learning methods neither capture semantic information within graphs nor consider both nodes and graphs while learning graph embeddings. We propose TAG (Two-staged contrAstive curriculum learning for Graphs), a two-staged contrastive learning method for graph classification. TAG learns graph representations in two levels: node-level and graph level, by exploiting six degree-based model-agnostic augmentation algorithms. Experiments show that TAG outperforms both unsupervised and supervised methods in classification accuracy, achieving up to 4.08% points and 4.76% points higher than the second-best unsupervised and supervised methods on average, respectively.

Medienart:

E-Artikel

Erscheinungsjahr:

2024

Erschienen:

2024

Enthalten in:

Zur Gesamtaufnahme - volume:19

Enthalten in:

PloS one - 19(2024), 1 vom: 31., Seite e0296171

Sprache:

Englisch

Beteiligte Personen:

Shim, Sooyeon [VerfasserIn]
Kim, Junghun [VerfasserIn]
Park, Kahyun [VerfasserIn]
Kang, U [VerfasserIn]

Links:

Volltext

Themen:

Journal Article

Anmerkungen:

Date Completed 05.01.2024

Date Revised 06.01.2024

published: Electronic-eCollection

Citation Status MEDLINE

doi:

10.1371/journal.pone.0296171

funding:

Förderinstitution / Projekttitel:

PPN (Katalog-ID):

NLM366613391