Skip to content

CV-Attention

Attention Mechanism

Self-Attention

Attention Score

Attention Score:

  • q1=Wqa1
  • k1=Wka1
  • k2=Wka2
  • k3=Wka3
  • k4=Wka4

a1,1=q1k1a1,2=q1k2a1,3=q1k3a1,4=q1k4

Attention Score

Attention Score

Attention

qi=Wqaiki=Wkaivi=Wvai

Attention Score

A=KTQA=softmax(A)

Attention

O=VTA

Take away:

Attention Mechanism

Q=WqIK=WkIV=WvIA=KTQA=softmax(A)O=VTA

Multi-Head Attention

Multi-Head Attention

Transformer