ํ”ผ๋“œ๋กœ ๋Œ์•„๊ฐ€๊ธฐ
Probabilistic Time Series Forecasting with ๐Ÿค— Transformers
Hugging Face BlogHugging Face Blog
AI/ML

Hugging Face Transformers ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๊ฐ€ Vanilla Transformer ๊ธฐ๋ฐ˜ ํ™•๋ฅ ์  ์‹œ๊ณ„์—ด ์˜ˆ์ธก ๋ชจ๋ธ์„ ์ œ๊ณตํ•ด ๊ตญ์†Œ ํฌ์ธํŠธ ์˜ˆ์ธก ๋ฐฉ์‹์—์„œ ๊ธ€๋กœ๋ฒŒ ํ™•๋ฅ  ๋ถ„ํฌ ๋ชจ๋ธ๋ง์œผ๋กœ์˜ ์ „ํ™˜ ์‹คํ˜„

Probabilistic Time Series Forecasting with ๐Ÿค— Transformers

2022๋…„ 12์›” 1์ผ10๋ถ„intermediate

Context

๊ณ ์ „์  ์‹œ๊ณ„์—ด ์˜ˆ์ธก ๋ฐฉ์‹(ARIMA)์€ ๊ฐ ์‹œ๊ณ„์—ด์„ ๊ฐœ๋ณ„์ ์œผ๋กœ ํ•™์Šตํ•˜๋Š” ๊ตญ์†Œ ๋ชจ๋ธ์ด๋ฉฐ, ํฌ์ธํŠธ ๊ฐ’๋งŒ ์ถœ๋ ฅํ•˜๋ฏ€๋กœ ์˜ˆ์ธก์˜ ๋ถˆํ™•์‹ค์„ฑ์„ ์ œ๊ณตํ•  ์ˆ˜ ์—†๋‹ค๋Š” ํ•œ๊ณ„๊ฐ€ ์žˆ๋‹ค. ๋Œ€๊ทœ๋ชจ ์‹œ๊ณ„์—ด ๋ฐ์ดํ„ฐ๋ฅผ ์ฒ˜๋ฆฌํ•  ๋•Œ ์—ฌ๋Ÿฌ ๋ฐ์ดํ„ฐ ์†Œ์Šค์˜ ์ž ์žฌ ํ‘œํ˜„์„ ํ•™์Šตํ•  ์ˆ˜ ์žˆ๋Š” ๊ธ€๋กœ๋ฒŒ ๋ชจ๋ธ์˜ ํ•„์š”์„ฑ์ด ์ฆ๋Œ€๋˜์—ˆ๋‹ค.

Technical Solution

  • Encoder-Decoder Transformer ์•„ํ‚คํ…์ฒ˜ ๋„์ž…: ์ปจํ…์ŠคํŠธ ์œˆ๋„์šฐ๋ฅผ ์ธ์ฝ”๋”์— ์ž…๋ ฅํ•˜๊ณ  ์˜ˆ์ธก ๊ธธ์ด ํฌ๊ธฐ์˜ ์œˆ๋„์šฐ๋ฅผ ์ธ๊ณผ ๋งˆ์Šคํฌ ์ ์šฉ ๋””์ฝ”๋”์— ์ „๋‹ฌํ•˜๋Š” ๊ตฌ์กฐ ๊ตฌ์„ฑ
  • ํ™•๋ฅ ์  ๋ถ„ํฌ ๋ชจ๋ธ๋ง: Gaussian, Student-T ๋“ฑ ๋งค๊ฐœ๋ณ€์ˆ˜ ๋ถ„ํฌ์˜ ๋ฏธ๋ž˜ ํŒŒ๋ผ๋ฏธํ„ฐ๋ฅผ ํ•™์Šตํ•˜๋Š” ๋ฐฉ์‹์œผ๋กœ ๊ตฌํ˜„ํ•˜๊ณ , ์ƒ˜ํ”Œ๋ง์„ ํ†ตํ•ด ์˜ˆ์ธก ๋ถˆํ™•์‹ค์„ฑ ์ œ๊ณต
  • ์žํšŒ๊ท€ ์„ธ๋Œ€(Ancestral Sampling) ์ ์šฉ: ํ•™์Šต๋œ ๋ถ„ํฌ์—์„œ ์ƒ˜ํ”Œ๋งํ•˜์—ฌ ๋ชฉํ‘œ ์˜ˆ์ธก ์ง€ํ‰๊นŒ์ง€ ์ˆœ์ฐจ์ ์œผ๋กœ ์˜ˆ์ธก๊ฐ’ ์ƒ์„ฑ
  • ๊ฒฐ์ธก๊ฐ’ ์ฒ˜๋ฆฌ: Attention ๋งˆ์Šคํฌ๋ฅผ ํ™œ์šฉํ•ด ๊ฒฐ์ธก๊ฐ’์„ ๋ณด์กฐ ๋งˆ์Šคํฌ๋กœ ์ธ์ฝ”๋”ฉํ•˜์—ฌ ๋ฐ์ดํ„ฐ ๋Œ€์ฒด/๋Œ€์ž… ์—†์ด ์ง์ ‘ ํ•™์Šต
  • GluonTS ๋ฐ์ดํ„ฐ ํฌ๋งท ์ฑ„ํƒ: ์•„ํ‹ฐํด ์ œ๊ณต ๋…ธํŠธ๋ถ์—์„œ ๋ฐ์ดํ„ฐ๋ฅผ Hugging Face Datasets ํ˜•์‹์œผ๋กœ ๋ณ€ํ™˜ํ•˜๋Š” ํ”„๋กœ์„ธ์Šค ๋ฌธ์„œํ™”

Key Takeaway

Transformer ๊ธฐ๋ฐ˜ ์‹œ๊ณ„์—ด ๋ชจ๋ธ์€ ์ˆ˜์ฒœ ๊ฐœ์˜ ํƒ€์ž„ ์Šคํ…์„ ์ฒ˜๋ฆฌํ•  ์ˆ˜ ์žˆ์œผ๋ฉฐ, ๋‹ค์ค‘ ์‹œ๊ณ„์—ด๋กœ๋ถ€ํ„ฐ ๊ณต์œ  ํ‘œํ˜„์„ ํ•™์Šตํ•˜๋Š” ๊ธ€๋กœ๋ฒŒ ๋ชจ๋ธ๋ง์ด ๊ฐ€๋Šฅํ•˜๋‹ค๋Š” ์ ์ด ํ•ต์‹ฌ์ด๋‹ค. ์ด๋ฅผ ํ†ตํ•ด ๊ณ ์ „์  ๋ฐฉ์‹์˜ ํ•œ๊ณ„๋ฅผ ๊ทน๋ณตํ•˜๊ณ  ์‹ค์ œ ์˜์‚ฌ๊ฒฐ์ • ํŒŒ์ดํ”„๋ผ์ธ์— ํ•„์š”ํ•œ ์˜ˆ์ธก ๋ถˆํ™•์‹ค์„ฑ์„ ์ •๋Ÿ‰ํ™”ํ•  ์ˆ˜ ์žˆ๋‹ค.


์‹œ๊ณ„์—ด ์˜ˆ์ธก์ด ํ•„์š”ํ•œ ํ”„๋กœ๋•์…˜ ํ™˜๊ฒฝ์—์„œ Hugging Face Transformers์˜ TimeSeriesTransformer ๋ชจ๋ธ์„ ์‚ฌ์šฉํ•  ๋•Œ, ์ปจํ…์ŠคํŠธ ์œˆ๋„์šฐ ํฌ๊ธฐ์™€ ์˜ˆ์ธก ๊ธธ์ด๋ฅผ ๋ฐ์ดํ„ฐ์…‹์˜ ์‹œ๊ณ„์—ด ํŠน์„ฑ์— ๋งž๊ฒŒ ์„ค์ •ํ•˜๊ณ  ํ™•๋ฅ  ๋ถ„ํฌ ํŒŒ๋ผ๋ฏธํ„ฐ๋ฅผ ํ•™์Šตํ•˜๋ฉด, ๋‹จ์ˆœ ํฌ์ธํŠธ ์˜ˆ์ธก๋งŒ ์ œ๊ณตํ•˜๋Š” ๋ฐฉ์‹ ๋Œ€๋น„ ์˜ˆ์ธก๊ฐ’์˜ ์‹ ๋ขฐ๋„ ๋ฒ”์œ„๋ฅผ ํ•จ๊ป˜ ์ œ๊ณตํ•  ์ˆ˜ ์žˆ์–ด ์˜์‚ฌ๊ฒฐ์ •์˜ ์ •ํ™•์„ฑ์„ ๋†’์ผ ์ˆ˜ ์žˆ๋‹ค.

์›๋ฌธ ์ฝ๊ธฐ