#48ïŒã¯ãã¹ã¢ãã³ã·ã§ã³ã®æ·±ã€ã€è©±ââãããªãã¯æšæ¥æ ç»ãèŠããïŒããè±èªã«ãªããŸã§
Transformer ã®è§£èª¬å³ãçºããŠãããšãEncoder ãš Decoder ã®ããã ã«åŒãããç¢å°ããã©ãããŠããåèªãšåèªã®å¯Ÿå¿è¡šãã«èŠããŠããŸããæ¥æ¬èªã®ãèŠãããè±èªã® "watch" ã«ç·ã§çµã°ããŠããââãããªã€ã¡ãŒãžã§ãã
ãšãããå®éã«æšè«ã®æé ãäžæ©ãã€è¿œã£ãŠã¿ããšãããã§èµ·ããŠããããšã¯å¯Ÿå¿è¡šãšã¯ãŸã£ããéããŸãããã®èšäºã¯ããã®ç¢ºèªäœæ¥ã®èšé²ã§ãã
顿ã¯äžæã ãã
ããªãã¯ æšæ¥ æ ç»ã èŠã ãïŒãâ¡ãDid you watch a movie yesterday?
ããã§ã¯è©±ã EncoderâDecoder å Transformer ã®æšè«æã«éå®ããŸããåŠç¿æã§ã¯ãããŸããããã®åºå¥ããããšã§å¹ããŠããŸãã
1. Encoder ã¯æ¥æ¬èªããäžåºŠã«ãèªã
ãŸãå
¥åæãããŒã¯ã³åããŸããç°¡åã®ãã 5 ããŒã¯ã³ãšããŸãããã
```
ããªãã¯ æšæ¥ æ ç»ã èŠã ãïŒ
ããâããã âãã â ãâã â
ããX1 ããX2 ã X3 X4 X5
```
ããã Encoder ã«ãŸãšããŠå
¥åããŸããEncoder ã® Self-Attention ã§ã¯ãåããŒã¯ã³ãã Q, K, V ãäœããäºããåç
§ãããŸããããã« Causal MaskïŒæªæ¥ãé ããã¹ã¯ïŒã¯ãããŸãããåæã¯ãã§ã«å
šéšããã«ããã®ã§ããããé ãçç±ããªãããã§ãã
ãããã£ãŠãèŠããã® Q4 ã¯ããããªãã¯ããæšæ¥ããæ ç»ããããïŒãã®ãã¹ãŠã® K ãèŠãããšãã§ããŸãã
ãããæåã®ãã€ã³ãã§ããEncoder ããåºãŠãããèŠããã®è¡šçŸã¯ãèŸæžçãªæå³ã§ã®ãèŠããã§ã¯ãããŸããããããªãã¯æšæ¥æ ç»ãèŠããïŒããšããæèã®ãªãã§ã®ãèŠããã«ãªã£ãŠããŸãã
Encoder ã®åŠçãçµãããšããã®åºåãã Cross-Attention çšã® K, V ãäœããŸãã
```
Encoder åºå
â
Kâ,Vâ ãKâ,Vâã Kâ,Vâã Kâ,Vâ ãKâ
,Vâ
```
ãã®5çµã®K,Vã¯ã翻蚳ãçµãããŸã§Decoderã®Cross-Attentionããç¹°ãè¿ãåç
§ãããŸãã
2. Decoder ã¯è±èªã1ããŒã¯ã³ãã€çæãã
Decoder ã«ã¯æåã`<BOS>`ïŒç¿»èš³éå§ã瀺ãç¹æ®ããŒã¯ã³ïŒã ããå
¥ããŸãã
Decoder Self-Attention ã§ã¯ããã® 1 ããŒã¯ã³ãã Q, K, V ãäœããŸãã1åãããªãã®ã§ãåç
§ã§ããã®ã¯èªåèªèº«ã ãã§ãã
ãããŠãEncoderâDecoder åã®æ žå¿ã§ãã Cross-Attention ã«é²ã¿ãŸãã
3. Cross-Attention ââã圢ããèŠããšæ§è³ªãåãã
Cross-Attention ã§ã¯ãQ 㯠Decoder ãããK ãš V 㯠Encoder ããæ¥ãŸãã
```
Decoder EncoderïŒæ¥æ¬èªïŒ
Q<BOS> â Kâ Vâ
ããããã Kâ Vâ
ããããã Kâ Vâ
ããããã Kâ Vâ
ããããã Kâ
Vâ
```
èšç®ããã®ã¯ QKáµ ã§ããDecoder åŽã® Q 㯠1 æ¬ãEncoder åŽã® K 㯠5 æ¬ã§ãããã
(1Ãd)(dÃ5)=1Ã5
çµæã¯ 1Ã5 ã®ãã¯ãã«ãã€ãŸããæ¥æ¬èª5ããŒã¯ã³ãããããšã®é¢é£åºŠã衚ãAttention Scoreãã«ãªããŸãã
ãã®åœ¢ãèŠãã°ãCross-Attention ãäœãããŠãããã¯æããã§ãã
Cross-Attentionã§ã¯ãQãšKã®æ¬æ°ãããããäžèŽããŠããŸãããæ¥æ¬èª 5 ããŒã¯ã³ãšè±èªã®çææžã¿ããŒã¯ã³ãäžå¯Ÿäžã§çµãã§ããã®ã§ã¯ãªããè±èªåŽã® 1 æ¬ã® Q ããæ¥æ¬èªåŽã® 5 æ¬ã® K ãšã®é¡äŒŒåºŠãèšç®ãããããŸæ¬¡ã®è±åèªãåºãããã«ãæ¥æ¬èªã®ã©ããèŠãã¹ããããæ±ºããŠããã®ã§ãã
å¿
èŠãªã®ã¯ãQã®1æ¬ãããã®ãã¯ãã«æ¬¡å
ãšãKã®1æ¬ãããã®ãã¯ãã«æ¬¡å
ãåãã§ããããšã§ãã
ããã Cross-Attention ã§ãã翻蚳ã«ãããŠæ¥è±ã§ããŒã¯ã³æ°ãèªé ãéãã®ã«ããªãæ©æ¢°ç¿»èš³ãæç«ããã®ãââãã®çããããã«ãããŸããããããæ°ãåãããå¿
èŠããªãã®ã§ãã
4. æšè«äžãEncoderã¯äžåºŠã ãèšç®ãã
`<BOS>` ã®æ¬¡ã« "Did" ãçæããããšããŸããDecoder ã®çæå±¥æŽã¯ `<BOS> Did` ã«ãªããŸããã
ãã®ãšããEncoder ã¯åèšç®ãããŸãããKâãKâ
ãVâãVâ
ã¯ãã®ãŸãŸä¿åãããŠãããæ¬¡ã®ã¹ãããã§ããã®ãŸãŸäœ¿ãããŸããæ¥æ¬èªã®åæã¯å€ãããªãã®ã§ããããåœããåãšããã°åœããåã§ãããããã Cross-Attention åŽã® KV Cache ã§ãã
äžæ¹ãDecoder Self-Attention ã§ã¯çŸåš `<BOS> Did` ã® 2 ããŒã¯ã³ããããŸãã
åŠç¿æã Prefill 段éãªããããã«ã¯ Causal Mask ãå¿
èŠã§ãã
```
ãããBOS Did
BOSã âãã Ã
Did ã âãã â
```
ãšããã Decode 段éïŒ1 ããŒã¯ã³ãã€çæããŠããæäžïŒã§ KV Cache ã䜿ã£ãŠããå Žåãæ°ããçæããã "Did" ã«å¯Ÿå¿ãã Q ã ããèšç®ããK_BOS ãš K_Did ãèŠãã ãã§ãã
æªæ¥ã®è±èªã¯ããŸã ååšããŸããã
`you watch a movie yesterday` ã¯ãã®æç¹ã§ã¯ãã®äžã«ãªãã®ã§ããããé ãã¹ãæªæ¥ããªããã€ãŸã Decode 段éã§ã¯ Causal Mask ã¯åççã«äžèŠãšããããšã«ãªããŸãããã¹ã¯ã¯ãæªæ¥ãèŠããªããããããã®ä»çµã¿ã§ããã鿬¡çæã§ã¯æªæ¥ãããããçæãããŠããªãããã§ãã
ïŒããã§ãããåççã«äžèŠãã¯ãKV Cacheãå©çšããŠæ°ãã1ããŒã¯ã³ã鿬¡åŠçããDecode段éãæããŸããå®è£
äžã¯ãåãCausal Attentionã®ä»çµã¿ãæ®ããŠããå ŽåããããŸãã
ããããŠã
```
<BOS>
â Did
â Did you
â Did you watch
â Did you watch a
â Did you watch a movie
â Did you watch a movie yesterday
â ?
```
ãšäŒžã³ãŠãããŸããåã¹ãããã§ Decoder ã¯ãSelf-Attention ã§ããããŸã§æžããè±èªããèŠãŠãCross-Attention ã§ãæ¥æ¬èªã®å
šæããèŠãŠã次㮠1ããŒã¯ã³ã決ããŠããŸãã
5. æ·±ã€ã€è©± ãã® 1 ââããïŒãã ãã§ã¯éå»åœ¢ã¯æ±ºãŸããªã
ããŠããããããæ¬é¡ã§ãã
æåã® "Did" ãçæãããšãã®ã¢ãã³ã·ã§ã³éã¿ã¯ãããšãã°ãããªã£ãŠããââãšãã説æãããèŠãããŸãã
```
ããªãã¯ ãæšæ¥ã æ ç»ããèŠãããïŒ
ã0.05ãã 0.10ã 0.05ã 0.05 ã0.75
```
çåæã ããããïŒãã«åŒ·ã泚ç®ãããããã«ããããããªèª¬æã§ãã
ããããã®èª¬æã§ã¯ã次ã®ãããªéãçªã£èŸŒã¿ãè¿ã£ãŠãããã§ãã
> ããïŒãã«æ³šç®ããã ãã§ã¯ãéå»åœ¢ãã©ããåãããªãã®ã§ã¯ïŒ
ãŸã£ãããã®éãã§ããã®èª¬æã¯åçŽåããããŠããŸããããïŒããæããŠãããã®ã¯ãçåæã§ããããšã ãã§ããããã ãã§ã¯ `Do` ãªã®ã `Did` ãªã®ã `Will` ãªã®ãã¯æ±ºãŸããªãã
å®éã«ã¯ã"Did" ãæ±ºããæ
å ±ã¯ 1 åã® K ããåºãŠãããã®ã§ã¯ãããŸããã

ãããã Encoder ã®åããŒã¯ã³ã®è¡šçŸã®ãªãã«åæ£ããŠååšããŠãããQ_BOS ã¯ãã®ãã¹ãŠãéã¿ä»ãã§åãåããŸãã
ãããŠãããã§ã¹ããã 1 ã®è©±ãå¹ããŠããŸãã
Kâã¯åãªããèŠãããšããåèªã®æ
å ±ã§ã¯ãªãããã®åšå²ã®æèãåæ ãã衚çŸã«ãªã£ãŠããŸããåãããã«KâïŒãæšæ¥ãïŒã«ãããèŠãããšçµã³ã€ããæéæ
å ±ãåæ ãããŠããŸããã€ãŸããCross-Attention ãåç
§ããŠããã®ã¯çŽ ã®åèªã§ã¯ãªããEncoder ãæèããå³ä»ããæœãã衚çŸã§ãã
ããïŒãã«æ³šç®ããããšããã£ããšããŠããéå»åœ¢ã«é¢ããæ
å ±ã¯KâãKâãªã©ãä»ã®æèåããã衚çŸã«ãå«ãŸããŠããŸãã
ãããïŒãã ãã§ã¯éå»åœ¢ã¯åãããªããââãã®åŒã£ãããã¯ãCross-Attention ãéèªçãªå¯Ÿå¿è¡šã§ã¯ãªãããšãçè§£ããããã§ããã以äžãªãã»ã©è¯ãå
¥å£ã ãšæããŸãã
6. æ·±ã€ã€è©± ãã® 2 ââ ççŸããåæãã©ãèš³ãã
åã調åã§ãããäžæ®µãããããªåããç«ãŠãŠã¿ãŸãã
> 人éãªããããªãã¯ææ¥æ ç»ãèŠããïŒããšèšãããã°ããææ³çã«ããããã®ã§ã¯ããšæããªããã "Did you watch a movie tomorrow?" ãšèš³ããšæããããã AI ã«ã¯ãããã¯é£ããã®ã§ã¯ãªããïŒ
æªæ¥ã®æé衚çŸãšéå»åœ¢ã®åè©ãåå±
ãããå£ããåæã§ãã
çµè«ããèšããšãAI ã«ããã®èš³ã¯ååã«çæã§ããŸãããã ããã®çŽæã«ã¯éèŠãªéšåãå«ãŸããŠããŸãã
Encoder ã¯ççŸããä¿æã§ããã
```
ããªã㯠ââ
ææ¥ ââ€
æ ç»ã ââ€â æèåããã衚çŸ
èŠã ââ€
ãïŒ ââ
```
ãææ¥ãã¯æªæ¥ããèŠããã¯éå»ãEncoder ã® Self-Attention ã¯å
šæãåæ¹åã«èŠãŸãããããã®äžèªç¶ãªçµã¿åãããã®ãã®ã衚çŸã®ãªãã«åæ ããåŸãŸããEncoder ã¯ãææ¥ããšãèŠãããå¥ã
ã«åŠçããŠçµããã§ã¯ãªããäž¡è
ã®é¢ä¿ãå«ãã 衚çŸãäœã£ãŠãã Decoder ã«æž¡ããŠããããšããããšã§ãã
人éã¯ããã§ãããäžæ®µã®å€æãããã
人éã®ç¿»èš³è
ã«ã¯ããã®æç¹ã§éžæè¢ããããŸãã
· åæå¿ å®åïŒDid you watch a movie tomorrow?ïŒççŸããšèš³ãïŒ
· æå³ä¿®æ£åïŒWill you watch a movie tomorrow?ïŒèª€èšãšå€æããŠçŽãïŒ
· 確èªåïŒã"Will you watch a movie tomorrow?" ã®æå³ã§ããããïŒããšèãè¿ã
ãã®åå²ã«ã¯ãææ³ãæå³ã®æŽåæ§ã倿ãã人éã®ç¿»èš³è
ãšããŠã®å€æãé¢ãã£ãŠããŸãã
ã¢ãã«ã®åŽã§èµ·ããŠããããš
Transformerã«ã¯ã人éãæžããææ³èŠåããã®ãŸãŸæ ŒçŽããææ³ãšã³ãžã³ãããããã§ã¯ãããŸããã倧éã®èšèªããŒã¿ãããææ³ãæå³ã®é¢ä¿ãå«ãè€éãªãã¿ãŒã³ãåŠç¿ãããããæ¬¡ã®ããŒã¯ã³ã®ç¢ºçååžãšããŠåºåããŸãã
ãããã£ãŠãåŠç¿ãã倧éã®èšèªãã¿ãŒã³ã翻蚳ãã¿ãŒã³ã®äžããããã®æèã«ç¶ãå¯èœæ§ã®é«ã衚çŸãçæãããããåæãå¿ å®ã«èš³ãå Žåãããã°ãæèããäžèªç¶ããè£æ£ããæ¹åã«é²ãå ŽåããããŸããããã«ãDecoder èªèº«ã `Did ... tomorrow` ãšããçµã¿åãããçæããŠããéçšã§ãããã®è±èªã¯äžèªç¶ã ããšããæ¹åã«ç¢ºçãäžããããšãèµ·ããåŸãŸãã
ãã ããããã§ãAI ã«ã¯ææ³çè§£ããªãããšèšãåãã®ãæ£ç¢ºã§ã¯ãããŸãããå€§èŠæš¡ã¢ãã«ã¯ã`Did you ... tomorrow?` ãã `Will you ... tomorrow?` ã®ã»ããèªç¶ã ããšããæ¥µããŠè€éãªçµ±èšçæ§é ãç²åŸããŠããŸããççŸãããçšåºŠæ€åºããããšã¯ã§ããããããããã¯ãæå¶ãšããèŠåã«ç
§ãããŠæç€ºçã«æšè«ããçµæã§ã¯ãªãââããã人éãšã®éãã§ãã
7. äžã€ã®ã¢ãã³ã·ã§ã³ãäžæã«æŽçãã
ãããŸã§ã®è©±ã衚ã«ãŸãšããŸãã

Cross-Attention ã®éšåã ããQ ã®åºæãš Kã»V ã®åºæãéããŸãããã® 1 è¡ããCross-Attentionã®æ žå¿ã§ãã
ïŒããã§ã¯ãçŸåšã®LLMæšè«ã§ãã䜿ãããçšèªãåããŠãå
¥åããŸãšããŠåŠçããæ®µéãPrefillãçæã1ããŒã¯ã³ãã€é²ããæ®µéãDecodeãšåŒã³ãŸãã
ãããŠçæã®æµããå³ã«ãããšããããªããŸãã
```
ãEncoderãæ¥æ¬èª 5 ããŒã¯ã³
ããããããâ Self-AttentionïŒãã¹ã¯ãªãã»åæ¹åïŒ
ãæèåããã衚çŸ
ããããããâ K, V ãšããŠä¿æïŒä»¥åŸãåèšç®ãªãïŒ
ããããããâ
ããããããâ
ããããCross-Attention âââ
ããããããâãããããã â
ããããããâ Q ãããããâ
ãDecoderãããããããã â
ã<BOS> â Did â you â .. â
ããããããâãããããã â
Self-Attention ââââââââââ
ãïŒéå»ã®è±èªãèŠãïŒ
```
æ°ããããŒã¯ã³ããåºã Q ãäºæã«åãããçæ¹ã¯èªåã®éå»ïŒè±èªïŒããããçæ¹ã¯åæïŒæ¥æ¬èªïŒãèŠã«è¡ãããã®äºéã®åç
§ããEncoderâDecoder å Transformer ã®èšèšãã®ãã®ã§ãã
è£è¶³ïŒçŸåšäž»æµã®Decoder-only LLMã«ã¯ãCross-Attentionããªã
æåŸã«ãæ··åããããç¹ãã²ãšã€ã
ãããŸã§èª¬æããã®ã¯ EncoderâDecoder åïŒãªãªãžãã«ã® TransformerãT5ã翻蚳ã¢ãã«ãªã©ïŒã®æ§é ã§ããäžæ¹ãããŸåºã䜿ãããŠãã GPT ç³»ã»Claude ç³»ãšãã£ãå€§èŠæš¡èšèªã¢ãã«ã®å€ã㯠Decoder-only åã§ãEncoder ã Cross-Attention ãæã¡ãŸããã
ã§ã¯ Decoder-only ã®ã¢ãã«ã¯ã©ããã£ãŠç¿»èš³ããŠããã®ããçãã¯åçŽã§ãåæããçææžã¿ããŒã¯ã³ããšããŠåãç³»åã«å
¥ããŠããŸãã®ã§ãã
```
Translate to English: ããªãã¯æšæ¥æ ç»ãèŠããïŒ â Did you watch ...
```
åæãèš³æãäžæ¬ã®ç³»åã«äžŠã³ãSelf-Attention ã ãã§åŠçãããŸããCross-Attention ããã£ãŠãããåæãåç
§ãããä»äºã¯ãSelf-Attention ãå
Œåããããšã«ãªããŸãã
æ§é ãšããŠã¯å¥ç©ã§ããããæèåããã衚çŸãéããŠåæå
šäœãåç
§ããããšããæ¬è³ªã¯å€ãããŸãããCross-Attention ãçè§£ããŠãããšãDecoder-only ã¢ãã«ãäœãçµ±åããã®ããããèŠããããã«ãªããŸãã
ãŸãšã
Cross-Attention ã® Q 㯠Decoder ãããKã»V 㯠Encoder ããæ¥ãããã®é察称æ§ããã¹ãŠ
Cross-Attentionã§ã¯QãšKã®æ¬æ°ã¯äžèŽããªããŠãããã ããæ¥è±ã§èªé ãããŒã¯ã³æ°ãéã£ãŠãã
åç §ããã K ã¯çŽ ã®åèªã§ã¯ãªããEncoder ãå šæãèŠãŠäœã£ãæèåæžã¿ã®è¡šçŸ
ããã«ããïŒãã«æ³šç®ããŠãéå»åœ¢ã®æ å ±ã¯å€±ãããªããæ å ±ã¯æå šäœã«åæ£ããŠãã
Decode 段éã§ã¯æªæ¥ã®ããŒã¯ã³ããŸã ååšããªããããCausal Mask ã¯åççã«äžèŠ
Cross-Attention ã¯å¯Ÿå¿è¡šã§ã¯ãªãããããŸäœãæžãããã«ãåæã®ã©ããèŠããããæ¯ã¹ãããæ±ºãçŽãä»çµã¿
å³ã®ç¢å°ãã察å¿è¡šããšèªããšãTransformer ã®ç¿»èš³ã¯ãã€ãŸã§ãäžæè°ãªãŸãŸã§ãããæ¯åãåæå šäœãåç §ããŠããããšèªã¿æ¿ããç¬éã«ãè ã«èœã¡ãŸãã
ãããªãšæã£ããå¿æŽãããïŒ
ãããããã°å¿æŽãé¡ãããŸãïŒ ããã ãããããã¯ã¯ãªãšã€ã¿ãŒãšããŠã®æŽ»åè²»ã«äœ¿ãããŠããã ããŸãïŒ 