GNNs Prof. Anderson Rocha Tavares Slide 19 / 24

Cabeça de Predição: do Embedding à Tarefa

Bloco 5 — Conclusão do Pipeline

Selecione o nível da tarefa:
🔵
Nível de Nó
ŷv = MLP(hv(L))
Classifica cada nó individualmente usando seu embedding final.
🔗
Nível de Aresta
ŷuv = MLP(hu(L) ‖ hv(L))
Concatena embeddings dos dois nós e classifica o par (u, v).
🗺️
Nível de Grafo
hG = READOUT({hv}) ou hglobal  →  ŷ = MLP(hG)
Agrega todos os embeddings em um vetor do grafo; MLP classifica o grafo inteiro.
READOUT:
MODO:
Pipeline da Tarefa
hv(L)
→
?
→
MLP
→
ŷ
hᵥ [...] MLP W, b ŷv = Classe A softmax([...]) hᵤ [...] hᵥ [...] CONCAT [hᵤ‖hᵥ] MLP W, b ŷuv = 0.87 P(aresta existe) h₁ h₂ h₃ h₄ READOUT Σ sum h_G [...] MLP W, b ŷG Tóxico? ← selecione uma tarefa para ver o diagrama
Clique em uma das tarefas à esquerda para ver como o MLP usa os embeddings finais da GNN.
💡 Os pesos da GNN (Message Passing) e os pesos do MLP (cabeça de predição) são treinados conjuntamente com backpropagation — a GNN aprende embeddings úteis para a tarefa específica.