HyperAIHyperAI

Command Palette

Search for a command to run...

vor 4 Monaten

Multi-Mask-Aggregator für Graph Neural Networks

{Ahmet Sureyya Rifaioglu Ahmet Sarıgün}

Multi-Mask-Aggregator für Graph Neural Networks

Abstract

Eine der entscheidenden Operationen in Graph Neural Networks (GNNs) ist die Aggregationsoperation, die darauf abzielt, Informationen aus den Nachbarn eines Zielknotens zu extrahieren. Es wurden mehrere Konvolutionsmethoden vorgeschlagen, darunter die Standard-Graph-Konvolution (GCN), Graph Attention (GAT) und Message Passing (MPNN). In dieser Studie stellen wir eine Aggregationsmethode namens Multi-Mask Aggregators (MMA) vor, bei der das Modell vor der Sammlung benachbarter Nachrichten für jeden Aggregator eine gewichtete Maske lernt. MMA ähnelt sowohl GAT als auch MPNN, verfügt jedoch über theoretische und praktische Vorteile. Intuitiv ist unser Framework nicht durch die Anzahl der Heads von GAT eingeschränkt und weist eine höhere Diskriminativkraft als MPNN auf. Die Leistungsfähigkeit von MMA wurde sowohl in Aufgaben zur Knotenklassifikation als auch zur Graph-Regression anhand weit verbreiteter Benchmark-Datensätze mit etablierten Baseline-Methoden verglichen und zeigte eine verbesserte Leistung. Die Datensätze und Codes sind unter https://github.com/asarigun/mma verfügbar.

Code-Repositories

Benchmarks

BenchmarkMethodikMetriken
graph-regression-on-zincMMA
MAE: 0.156
node-classification-on-citeseerMMA
Accuracy: 76.30%
node-classification-on-coraMMA
Accuracy: 85.80%
node-classification-on-pubmedMMA
Accuracy: 86.00%

KI mit KI entwickeln

Von der Idee bis zum Start — beschleunigen Sie Ihre KI-Entwicklung mit kostenlosem KI-Co-Coding, sofort einsatzbereiter Umgebung und den besten GPU-Preisen.

KI-Co-Coding
Sofort einsatzbereit GPUs
Beste Preise
Jetzt starten

Hyper Newsletters

Abonnieren Sie unsere neuesten Updates
Wir werden die neuesten Updates der Woche in Ihren Posteingang liefern um neun Uhr jeden Montagmorgen
Unterstützt von MailChimp