the.ai

Graphs / Attention

verified

Graph Attention

A graph convolution treats every neighbour as equally worth listening to. Graph attention learns the weights instead, so a node can lean on the two neighbours that matter and ignore the forty that do not. It is the same idea as attention in a transformer, restricted to the edges the graph actually has.

Viz primitive · attention-heatmaptokens = 8
query ↓key →

8 queries against 10 keys; a brighter cell means more of that query's attention went to that key. Nothing is masked: every target position can read every source position.

Nodes against the neighbours they attend over, nothing masked within the neighbourhood. Drag the neighbour count up to watch each node's fixed attention budget spread thinner — the softmax sums to one however many arrive.

10

Reviewed by opendroid · 2026-08-18