Post

Deep Dive into Self-Attention by Hand

Original Article : ๐Ÿ‘‰ https://towardsdatascience.com/deep-dive-into-self-attention-by-hand-%EF%B8%8E-f02876e49857


ChatGPT4o

์…€ํ”„ ์–ดํ…์…˜ ๋ฉ”์ปค๋‹ˆ์ฆ˜ ์š”์•ฝ

์…€ํ”„ ์–ดํ…์…˜(Self-Attention)์€ ์ž…๋ ฅ ์‹œํ€€์Šค์˜ ๊ฐ ๋‹จ์–ด๊ฐ€ ๋‹ค๋ฅธ ๋ชจ๋“  ๋‹จ์–ด๋“ค๊ณผ์˜ ๊ด€๊ณ„๋ฅผ ํ‰๊ฐ€ํ•˜์—ฌ ๊ฐ€์ค‘์น˜๋ฅผ ๋ถ€์—ฌํ•˜๋Š” ๋ฐฉ๋ฒ•์ž…๋‹ˆ๋‹ค. ์ด๋ฅผ ํ†ตํ•ด ๋ฌธ๋งฅ์„ ๋” ์ž˜ ์ดํ•ดํ•˜๊ณ  ๊ธด ์ข…์†์„ฑ์„ ํฌ์ฐฉํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

์ฃผ์š” ๋‹จ๊ณ„

์ฟผ๋ฆฌ(Q), ํ‚ค(K), ๊ฐ’(V) ์ƒ์„ฑ: ์ž…๋ ฅ ํ–‰๋ ฌ์„ ์„ ํ˜• ๋ณ€ํ™˜ํ•˜์—ฌ Q, K, V ํ–‰๋ ฌ์„ ๋งŒ๋“ญ๋‹ˆ๋‹ค.

  • ์–ดํ…์…˜ ์Šค์ฝ”์–ด ๊ณ„์‚ฐ: Q์™€ K์˜ ํ–‰๋ ฌ ๊ณฑ์„ ํ†ตํ•ด ๊ฐ ๋‹จ์–ด์˜ ์œ ์‚ฌ๋„๋ฅผ ๊ตฌํ•˜๊ณ , ์†Œํ”„ํŠธ๋งฅ์Šค ํ•จ์ˆ˜๋กœ ์ •๊ทœํ™”ํ•ฉ๋‹ˆ๋‹ค.
  • ์–ดํ…์…˜ ์ถœ๋ ฅ ๊ณ„์‚ฐ: ์ •๊ทœํ™”๋œ ์Šค์ฝ”์–ด์— V๋ฅผ ๊ณฑํ•˜์—ฌ ์ตœ์ข… ์ถœ๋ ฅ ๊ฐ’์„ ์–ป์Šต๋‹ˆ๋‹ค.

์žฅ์ 

  • ๋ณ‘๋ ฌ ์ฒ˜๋ฆฌ ๊ฐ€๋Šฅ
  • ๊ธด ์ข…์†์„ฑ ํ•™์Šต์— ํšจ๊ณผ์ 

Lilys AI : ๐Ÿ‘‰ https://lilys.ai/digest/660003

Below Summary Note

Deep Dive into Self-Attention by Handโœ๏ธŽby Srijanie Dey, PhDApr, 2024Towards Data Science

1.ํŠธ๋žœ์Šคํฌ๋จธ๋ฅผ ๊ฐ€์†ํ•˜๋Š” ์ฃผ์˜ ์ง‘์ค‘์˜ ๋ณธ๊ฒฉ์  ํƒ๊ตฌ

  • ํŠธ๋žœ์Šคํฌ๋จธ๋ฅผ ์ด‰์ง„ํ•˜๋Š” ์ฃผ์˜ ๋ฉ”์ปค๋‹ˆ์ฆ˜์„ ์ƒ์„ธํžˆ ์‚ดํŽด๋ณธ๋‹ค.
  • ๋จธ๋ฆฌ๊ธ€: โ€˜์ฃผ์˜โ€™ ๊ฐœ๋…์— ๋Œ€ํ•œ ๋ณต์žก์„ฑ์„ ์•Œ์•„๋ณด์ž.
  • โ€˜๋กœ๋น—๋จธ์Šค ํ”„๋ผ์ž„โ€™ ๊ฐ™์€ ๊ฒƒ์ด ์•„๋‹ˆ๋ผ ์‹ ๊ฒฝ๋ง์„ ๊ตฌ์„ฑํ•˜๋Š” ํŠธ๋žœ์Šคํฌ๋จธ์˜ ํž˜์€ โ€˜์ฃผ์˜โ€™ ๊ฐœ๋… ๋•Œ๋ฌธ์ด๋‹ค.
  • ๊ทธ๋ ‡๋‹ค๋ฉด, ํŠธ๋žœ์Šคํฌ๋จธ์˜ ๋งฅ๋ฝ์—์„œ ์ฃผ์˜๋ž€ ์ •ํ™•ํžˆ ๋ฌด์—‡์„ ์˜๋ฏธํ•˜๋Š” ๊ฒƒ์ผ๊นŒ? ์—ฌ๊ธฐ์„œ ๋ช‡ ๊ฐ€์ง€ ๋‹ต์„ ์ฐพ์•„๋ณด๋ ค๊ณ  ํ•œ๋‹ค.

2.ํŠธ๋žœ์Šคํฌ๋จธ์™€ ์„ธ๋ถ€ ๋ฉ”์ปค๋‹ˆ์ฆ˜ ์ถ”๋ก 

  • ํŠธ๋žœ์Šคํฌ๋จธ๋Š” ๋ฐ์ดํ„ฐ๋กœ๋ถ€ํ„ฐ ๋งฅ๋ฝ์„ ํ•™์Šตํ•˜๋Š” ์‹ ๊ฒฝ๋ง์œผ๋กœ, โ€˜์ฃผ์˜์™€ ๋งฅ๋ฝโ€™์˜ ์˜๋ฏธ๋ฅผ ์ฐพ๋Š” ์šฐ๋ฆฌ์™€ ์œ ์‚ฌํ•˜๋‹ค.
  • ํŠธ๋žœ์Šคํฌ๋จธ๊ฐ€ ๋ฐ์ดํ„ฐ๋กœ๋ถ€ํ„ฐ ๋งฅ๋ฝ์„ ์–ด๋–ป๊ฒŒ ํ•™์Šตํ•˜๋Š”๊ฐ€? : ์ฃผ์˜ ๋ฉ”์ปค๋‹ˆ์ฆ˜์„ ์‚ฌ์šฉํ•จ.
  • ์ฃผ์˜ ๋ฉ”์ปค๋‹ˆ์ฆ˜์€ ๋ชจ๋ธ์ด ๊ฐ ๋‹จ๊ณ„์—์„œ ์‹œํ€€์Šค์˜ ๋ชจ๋“  ๋ถ€๋ถ„์„ ์กฐ์‚ฌํ•˜๊ณ  ์ค‘์ ์„ ๋‘์–ด์•ผ ํ•  ์š”์†Œ๋ฅผ ๊ฒฐ์ •ํ•˜๋Š” ๋ฐ ๋„์›€์ด ๋œ๋‹ค.
  • ์…€ํ”„-์ฃผ์˜๋Š” RNN์˜ ์„ฑ๋Šฅ์„ ํ–ฅ์ƒ์‹œํ‚ค๋Š” ๋ฐ ๊ธฐ์—ฌํ•˜๋ฉฐ, 2017๋…„์— ์†Œ๊ฐœ๋œ ํŠธ๋žœ์Šคํฌ๋จธ ์•„ํ‚คํ…์ฒ˜๋Š” RNN ๋ฐ CNN์˜ ํ•„์š”์„ฑ์„ ์†Œ๋ฉธ์‹œ์ผฐ๋‹ค.
  • ํŠธ๋žœ์Šคํฌ๋จธ ์•„ํ‚คํ…์ฒ˜์—๋Š” ์Šค์ผ€์ผ๋“œ ๋‹ท-ํ”„๋กœ๋•ํŠธ ๋ฉ”์ปค๋‹ˆ์ฆ˜์ด ์žˆ์œผ๋ฉฐ, ์…€ํ”„-์ฃผ์˜๋Š” ์™ธ๋ถ€๊ฐ™์€ ์…ธ๊ณผ ๋‚ด๋ถ€๊ฐ™์€ ์…ธ์˜ ๊ฒฐํ•ฉ์œผ๋กœ ๋ณผ ์ˆ˜ ์žˆ๋‹ค.

3.๏ธSelf-Attention ๋ฉ”์ปค๋‹ˆ์ฆ˜๊ณผ ์–ดํ…์…˜ ์›จ์ดํŒ… ์ดํ•ด

  • ์–ดํ…์…˜ ์›จ์ดํŠธ ๋งคํŠธ๋ฆญ์Šค A๋Š” ์ž…๋ ฅ ํŠน์ง•์„ Query-Key(QK) ๋ชจ๋“ˆ์— ํ”ผ๋“œํ•˜์—ฌ ์–ป๊ฒŒ ๋œ๋‹ค.
  • ์…€ํ”„ ์–ดํ…์…˜์€ QK-๋ชจ๋“ˆ์„ ์‚ฌ์šฉํ•˜์—ฌ ์–ดํ…์…˜ ์›จ์ดํŠธ ๋งคํŠธ๋ฆญ์Šค A๋ฅผ ๋งŒ๋“ค ๋•Œ ์ˆ˜ํ–‰๋œ๋‹ค.
  • ์ฟผ๋ฆฌ(Q), ํ‚ค(K), ๋ฐธ๋ฅ˜(V) ๋“ฑ ์—ฌ๋Ÿฌ ๊ตฌ์„ฑ์š”์†Œ๋ฅผ ์‚ดํŽด๋ณด๋ฉฐ ์„ค๋ช…์ ์œผ๋กœ ํ•œ๊ตญ์˜ ์ƒ์ƒ์„ ์ด์šฉํ•˜์—ฌ Self-Attention์˜ ์›๋ฆฌ๋ฅผ ์ดํ•ดํ•œ๋‹ค.
  • ํ–‰๋ ฌ ๊ณฑ์…ˆ์˜ ๊ณผ์ •๊ณผ ์Šค์ผ€์ผ๋ง ๊ณผ์ •์— ๋Œ€ํ•œ ์‹ค์ œ ์—ฐ์‚ฐ๊ณผ ๋ฒกํ„ฐ ๊ฐ„ ์ฝ”์‚ฌ์ธ ์œ ์‚ฌ๋„ ๋ฐ ์ฐจ์›๊ณผ์˜ ๊ด€๊ณ„๋ฅผ ์ดํ•ดํ•œ๋‹ค.

4.๏ธ ํŠธ๋žœ์Šคํฌ๋จธ ์„น์…˜์—์„œ QK-๋ชจ๋“ˆ์„ ํ†ต๊ณผํ•œ ํ›„ ์–ป์€ ์–ดํ…์…˜ ๊ฐ€์ค‘์น˜ ํ–‰๋ ฌ.

  • โ€˜์Šค์ผ€์ผ๋งโ€™๋œ ๋‹ท-ํ”„๋กœ๋•ํŠธ ์–ดํ…์…˜์˜ โ€˜์Šค์ผ€์ผ๋งโ€™ ๋ถ€๋ถ„ ์„ค๋ช….
  • ์†Œํ”„ํŠธ๋งฅ์Šค ๋‹จ๊ณ„์˜ ์„ธ ๊ฐ€์ง€ ๋ถ€๋ถ„: 1. ๊ฐ ์…€์˜ ์ˆซ์ž์— e์˜ ๊ฑฐ๋“ญ์ œ๊ณฑ์„ ์ทจํ•จ, 2. ๊ฐ ์—ด์„ ๋”ฐ๋ผ ์ด ์ƒˆ๋กœ์šด ๊ฐ’๋“ค์„ ํ•ฉํ•จ, 3. ๊ฐ ์—ด๋งˆ๋‹ค, ํ•ด๋‹น ํ•ฉ์œผ๋กœ ๋‚˜๋ˆ ์คŒ.
  • ๊ฐ ์—ด์„ ์ •๊ทœํ™”ํ•˜์—ฌ ์ˆซ์ž๋“ค์ด 1๋กœ ํ•ฉ์ณ์ง€๋„๋ก ํ•จ. ๊ฒฐ๊ณผ์ ์œผ๋กœ ๊ฐ ์—ด์€ ์–ดํ…์…˜์˜ ํ™•๋ฅ  ๋ถ„ํฌ๋กœ ๋˜๋ฉฐ, ์–ดํ…์…˜ ๊ฐ€์ค‘์น˜ ํ–‰๋ ฌ (A)์„ ์ œ๊ณต.
  • ์–ดํ…์…˜ ๊ฐ€์ค‘์น˜ ํ–‰๋ ฌ์€ ํŠธ๋žœ์Šคํฌ๋จธ ์„น์…˜์—์„œ์˜ ๋‹จ๊ณ„ 2(QK-๋ชจ๋“ˆ์„ ํ†ต๊ณผํ•œ ํ›„)์—์„œ ์–ป์€ ๊ฒƒ.
  • ์†Œํ”„ํŠธ๋งฅ์Šค ๋‹จ๊ณ„๋Š” ์ด์ „ ๋‹จ๊ณ„์—์„œ ์–ป์€ ์ ์ˆ˜์— ํ™•๋ฅ ์„ ํ• ๋‹นํ•˜์—ฌ ๋ชจ๋ธ์ด ํ˜„์žฌ ์ฟผ๋ฆฌ์— ๋Œ€ํ•ด ๊ฐ ๋‹จ์–ด์— ์–ผ๋งˆ๋‚˜ ์ค‘์š”ํ•˜๋‹ค๊ณ  ๊ฒฐ์ •ํ•˜๋Š” ๋ฐ ๋„์›€์„ ์คŒ.

5.์†Œํ”„ํŠธ๋งฅ์Šค ์Šคํ…์˜ ์ค‘์š”์„ฑ๊ณผ ๊ฐ ๋‹จ์–ด์˜ ์ค‘์š”๋„ ํŒ๋‹จ

  • ์†Œํ”„ํŠธ๋งฅ์Šค ์Šคํ…์€ ์ด์ „ ๋‹จ๊ณ„์—์„œ ์–ป์€ ์ ์ˆ˜์— ํ™•๋ฅ ์„ ํ• ๋‹นํ•˜์—ฌ, ๋ชจ๋ธ์ด ํ˜„์žฌ ์ฟผ๋ฆฌ์— ๋Œ€ํ•ด ๊ฐ ๋‹จ์–ด์— ์–ผ๋งˆ๋‚˜ ์ค‘์š”๋„๋ฅผ ๋ถ€์—ฌํ•ด์•ผ ํ•˜๋Š”์ง€ ๊ฒฐ์ •ํ•˜๋Š” ๋ฐ ์ค‘์š”ํ•˜๋‹ค.
  • ๋†’์€ ์ฃผ์˜ ๊ฐ€์ค‘์น˜๊ฐ€ ๋” ๋†’์€ ๊ด€๋ จ์„ฑ์„ ๋‚˜ํƒ€๋‚ด๋ฉฐ, ๋ชจ๋ธ์ด ์˜์กด์„ฑ์„ ๋” ์ •ํ™•ํ•˜๊ฒŒ ํŒŒ์•…ํ•˜๋„๋ก ๋•๋Š”๋‹ค.
  • ์ด์ „ ๋‹จ๊ณ„์—์„œ์˜ ์Šค์ผ€์ผ๋ง์ด ์—ฌ๊ธฐ์„œ ์ค‘์š”ํ•ด์ง€๋Š”๋ฐ, ์Šค์ผ€์ผ๋ง ์—†์ด ๊ฒฐ๊ณผ ํ–‰๋ ฌ์˜ ๊ฐ’์ด ์†Œํ”„ํŠธ๋งฅ์Šค ํ•จ์ˆ˜์—์„œ ์ž˜ ์ฒ˜๋ฆฌ๋˜์ง€ ๋ชปํ•˜๋Š” ์˜์—ญ์œผ๋กœ ๋ฐ€๋ ค ๋‚˜์„œ, ์‚ฌ๋ผ์ง€๋Š” ๊ธฐ์šธ๊ธฐ๋กœ ์ด์–ด์งˆ ์ˆ˜ ์žˆ๋‹ค.
  • ๋งˆ์ง€๋ง‰์œผ๋กœ ํ–‰๋ ฌ ๊ณฑ์…ˆ์„ ํ†ตํ•ด ๊ฐ’ ๋ฒกํ„ฐ(Vs)๋ฅผ *์ฃผ์˜ ๊ฐ€์ค‘์น˜ ํ–‰๋ ฌ(A)๊ณผ ๊ณฑํ•œ๋‹ค. ์ด ๊ฐ’ ๋ฒกํ„ฐ๋Š” ๊ฐ ๋‹จ์–ด์— ์—ฐ๊ด€๋œ ์ •๋ณด๋ฅผ ํฌํ•จํ•˜๋ฏ€๋กœ ์ค‘์š”ํ•˜๋‹ค.

6.์ž๊ฐ€์ฃผ์˜ ๋ฉ”์ปค๋‹ˆ์ฆ˜์˜ ์ตœ์ข… ํ•ด๊ฒฐ์ฑ…: ์ฃผ์˜ ์ง‘์ค‘ ํŠน์ง• Zs

  • ์ด ๋‹จ๊ณ„์˜ ์ตœ์ข… ๊ณฑ์…ˆ ๊ฒฐ๊ณผ๋Š” ์ฃผ์˜ ์ง‘์ค‘ ๊ฐ€์ค‘ ํŠน์ง• Z๋กœ, ์ž๊ฐ€ ์ฃผ์˜ ๋ฉ”์ปค๋‹ˆ์ฆ˜์˜ ๊ถ๊ทน์ ์ธ ํ•ด๊ฒฐ์ฑ…์ด๋‹ค.
  • ์ด๋Ÿฌํ•œ ์ฃผ์˜ ์ง‘์ค‘ ๊ฐ€์ค‘ ํŠน์ง•์€ ๋ณธ์งˆ์ ์œผ๋กœ ํŠน์ง•์˜ ๊ฐ€์ค‘ ํ‘œํ˜„์„ ๋‹ด๊ณ  ์žˆ์œผ๋ฉฐ, ์ฃผ๋ณ€ ๋งฅ๋ฝ์— ๋”ฐ๋ผ ์ค‘์š”๋„๊ฐ€ ๋†’์€ ํŠน์ง•์— ๋Œ€ํ•ด ๋†’์€ ๊ฐ€์ค‘์น˜๋ฅผ ํ• ๋‹นํ•œ๋‹ค.

7.ํŠธ๋žœ์Šคํฌ๋จธ ์•„ํ‚คํ…์ฒ˜์˜ ํ”ผ๋“œ ํฌ์›Œ๋“œ ๋ ˆ์ด์–ด์™€ ์ž๊ฐ€ ์ฃผ์˜ ๊ธฐ๋ฒ•์˜ ๋

  • ์ด ์ •๋ณด๋ฅผ ๊ธฐ๋ฐ˜์œผ๋กœ
    1
    
    ํŠธ๋žœ์Šคํฌ๋จธ ์•„ํ‚คํ…์ฒ˜
    
    ์˜ ๋‹ค์Œ ๋‹จ๊ณ„๋กœ ๋„˜์–ด๊ฐ€๊ฒŒ ๋˜๋Š”๋ฐ, ์—ฌ๊ธฐ์„œ๋Š”
    1
    
    ํ”ผ๋“œ ํฌ์›Œ๋“œ ๋ ˆ์ด์–ด
    
    ๊ฐ€ ์ด ์ •๋ณด๋ฅผ ๋” ์ฒ˜๋ฆฌํ•œ๋‹ค.
  • ์ด๋กœ์จ
    1
    
    ์ž๊ฐ€ ์ฃผ์˜ ๊ธฐ๋ฒ•
    
    ์˜ ๋›ฐ์–ด๋‚œ ๊ธฐ๋ฒ•์˜ ๋์„ ํ™•์ธํ•˜๊ฒŒ ๋ฉ๋‹ˆ๋‹ค!
This post is licensed under CC BY 4.0 by the author.