AI & Analytics

Grafische neurale netwerken: uitleg over GCN, MPNN en GAT

Towards Data Science (Medium)
Grafische neurale netwerken: uitleg over GCN, MPNN en GAT

Samenvatting

Grafische neurale netwerken verwerken relaties tussen knopen en leggen GCN, MPNN en GAT uit voor analyses van moleculen, verkeer en netwerken.

Grafische neurale netwerken: wat er gebeurt

Gewone neurale netwerken leren een wiskundige functie van invoer naar doelwaarde, maar kennen standaard geen relaties tussen onderdelen van de invoer. Bij afbeeldingen lossen convoluties dat op door pixels met hun buren te combineren. Grafische neurale netwerken passen hetzelfde principe toe op grafen, waarin objecten als knopen en relaties als verbindingen worden weergegeven.

De besproken architecturen zijn GCN, MPNN en GAT. Het artikel werkt vooral het idee achter een Graph Convolutional Network uit: een knoop verzamelt informatie van aangrenzende knopen en krijgt daarna nieuwe kenmerken. De graafstructuur blijft behouden. De knooprepresentaties veranderen van X naar H, met de adjacency matrix A als beschrijving van de verbindingen.

Grafische neurale netwerken: waarom dit belangrijk is

Een GNN kijkt niet alleen naar eigenschappen van afzonderlijke objecten, maar ook naar hun positie binnen een netwerk. Dat maakt de aanpak geschikt voor moleculen, sociale netwerken, voetballers tijdens een wedstrijd, verkeer en metrolijnen. Daarmee verschilt een GNN van een model dat ieder record los behandelt.

De uitvoer ondersteunt verschillende analysetaken. Je kunt afzonderlijke knopen classificeren, verbindingen beoordelen of de volledige graaf indelen. Een getraind model kan bovendien worden toegepast op een nieuwe graaf met een andere structuur. Het artikel noemt antibioticumontdekking als toepassing: een model leert op moleculen en verwerkt daarna een molecuul met een niet eerder geziene graafstructuur.

Grafische neurale netwerken: concrete takeaway

Breng eerst het analysetype terug tot de structuur van je probleem. Zijn afzonderlijke objecten het doel, kijk dan naar node classification; gaat het om relaties, kies edge classification; moet het geheel een label krijgen, gebruik graph classification. Leg daarnaast expliciet vast welke kenmerken in X staan en hoe A de verbindingen beschrijft. Die keuze bepaalt welke context de GNN tijdens het leren kan meenemen.

Lees het volledige artikel
Meer over AI & Analytics →