DrDavis's picture
Upload folder using huggingface_hub
17c6d62 verified

Graphormer[[graphormer]]

์ด ๋ชจ๋ธ์€ ์œ ์ง€ ๋ณด์ˆ˜ ๋ชจ๋“œ๋กœ๋งŒ ์šด์˜๋˜๋ฉฐ, ์ฝ”๋“œ๋ฅผ ๋ณ€๊ฒฝํ•˜๋Š” ์ƒˆ๋กœ์šด PR(Pull Request)์€ ๋ฐ›์ง€ ์•Š์Šต๋‹ˆ๋‹ค. ์ด ๋ชจ๋ธ์„ ์‹คํ–‰ํ•˜๋Š” ๋ฐ ๋ฌธ์ œ๊ฐ€ ๋ฐœ์ƒํ•œ๋‹ค๋ฉด, ์ด ๋ชจ๋ธ์„ ์ง€์›ํ•˜๋Š” ๋งˆ์ง€๋ง‰ ๋ฒ„์ „์ธ v4.40.2๋ฅผ ๋‹ค์‹œ ์„ค์น˜ํ•ด ์ฃผ์„ธ์š”. ๋‹ค์Œ ๋ช…๋ น์–ด๋ฅผ ์‹คํ–‰ํ•˜์—ฌ ์žฌ์„ค์น˜ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค: pip install -U transformers==4.40.2.

๊ฐœ์š”[[overview]]

Graphormer ๋ชจ๋ธ์€ Chengxuan Ying, Tianle Cai, Shengjie Luo, Shuxin Zheng, Guolin Ke, Di He, Yanming Shen, Tie-Yan Liu๊ฐ€ ์ œ์•ˆํ•œ ํŠธ๋žœ์Šคํฌ๋จธ๊ฐ€ ๊ทธ๋ž˜ํ”„ ํ‘œํ˜„์— ์žˆ์–ด์„œ ์ •๋ง ์•ฝํ• ๊นŒ? ๋ผ๋Š” ๋…ผ๋ฌธ์—์„œ ์†Œ๊ฐœ๋˜์—ˆ์Šต๋‹ˆ๋‹ค. Graphormer๋Š” ๊ทธ๋ž˜ํ”„ ํŠธ๋žœ์Šคํฌ๋จธ ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค. ํ…์ŠคํŠธ ์‹œํ€€์Šค ๋Œ€์‹  ๊ทธ๋ž˜ํ”„์—์„œ ๊ณ„์‚ฐ์„ ์ˆ˜ํ–‰ํ•  ์ˆ˜ ์žˆ๋„๋ก ์ˆ˜์ •๋˜์—ˆ์œผ๋ฉฐ, ์ „์ฒ˜๋ฆฌ์™€ ๋ณ‘ํ•ฉ ๊ณผ์ •์—์„œ ์ž„๋ฒ ๋”ฉ๊ณผ ๊ด€์‹ฌ ํŠน์„ฑ์„ ์ƒ์„ฑํ•œ ํ›„ ์ˆ˜์ •๋œ ์–ดํ…์…˜์„ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค.

ํ•ด๋‹น ๋…ผ๋ฌธ์˜ ์ดˆ๋ก์ž…๋‹ˆ๋‹ค:

ํŠธ๋žœ์Šคํฌ๋จธ ์•„ํ‚คํ…์ฒ˜๋Š” ์ž์—ฐ์–ด ์ฒ˜๋ฆฌ์™€ ์ปดํ“จํ„ฐ ๋น„์ „ ๋“ฑ ๋งŽ์€ ๋ถ„์•ผ์—์„œ ์ง€๋ฐฐ์ ์ธ ์„ ํƒ์„ ๋ฐ›๊ณ  ์žˆ๋Š” ์•„ํ‚คํ…์ฒ˜ ์ž…๋‹ˆ๋‹ค. ๊ทธ๋Ÿฌ๋‚˜ ๊ทธ๋ž˜ํ”„ ์ˆ˜์ค€ ์˜ˆ์ธก ๋ฆฌ๋”๋ณด๋“œ ์ƒ์—์„œ๋Š” ์ฃผ๋ฅ˜ GNN ๋ณ€ํ˜•๋ชจ๋ธ๋“ค์— ๋น„ํ•ด ๊ฒฝ์Ÿ๋ ฅ ์žˆ๋Š” ์„ฑ๋Šฅ์„ ๋‹ฌ์„ฑํ•˜์ง€ ๋ชปํ–ˆ์Šต๋‹ˆ๋‹ค. ๋”ฐ๋ผ์„œ ํŠธ๋žœ์Šคํฌ๋จธ๊ฐ€ ๊ทธ๋ž˜ํ”„ ํ‘œํ˜„ ํ•™์Šต์—์„œ ์–ด๋–ป๊ฒŒ ์ž˜ ์ˆ˜ํ–‰๋  ์ˆ˜ ์žˆ์„์ง€๋Š” ์—ฌ์ „ํžˆ ๋ฏธ์Šคํ„ฐ๋ฆฌ์˜€์Šต๋‹ˆ๋‹ค. ๋ณธ ๋…ผ๋ฌธ์—์„œ๋Š” Graphormer๋ฅผ ์ œ์‹œํ•จ์œผ๋กœ์จ ์ด ๋ฏธ์Šคํ„ฐ๋ฆฌ๋ฅผ ํ•ด๊ฒฐํ•ฉ๋‹ˆ๋‹ค. Graphormer๋Š” ํ‘œ์ค€ ํŠธ๋žœ์Šคํฌ๋จธ ์•„ํ‚คํ…์ฒ˜๋ฅผ ๊ธฐ๋ฐ˜์œผ๋กœ ๊ตฌ์ถ•๋˜์—ˆ์œผ๋ฉฐ, ํŠนํžˆ ์ตœ๊ทผ์˜ OpenGraphBenchmark Large-Scale Challenge(OGB-LSC)์˜ ๊ด‘๋ฒ”์œ„ํ•œ ๊ทธ๋ž˜ํ”„ ํ‘œํ˜„ ํ•™์Šต ์ž‘์—…์—์„œ ํƒ์›”ํ•œ ๊ฒฐ๊ณผ๋ฅผ ์–ป์„ ์ˆ˜ ์žˆ์—ˆ์Šต๋‹ˆ๋‹ค. ๊ทธ๋ž˜ํ”„์—์„œ ํŠธ๋žœ์Šคํฌ๋จธ๋ฅผ ํ™œ์šฉํ•˜๋Š”๋ฐ ํ•ต์‹ฌ์€ ๊ทธ๋ž˜ํ”„์˜ ๊ตฌ์กฐ์  ์ •๋ณด๋ฅผ ๋ชจ๋ธ์— ํšจ๊ณผ์ ์œผ๋กœ ์ธ์ฝ”๋”ฉํ•˜๋Š” ๊ฒƒ์ž…๋‹ˆ๋‹ค. ์ด๋ฅผ ์œ„ํ•ด ์šฐ๋ฆฌ๋Š” Graphormer๊ฐ€ ๊ทธ๋ž˜ํ”„ ๊ตฌ์กฐ ๋ฐ์ดํ„ฐ๋ฅผ ๋” ์ž˜ ๋ชจ๋ธ๋งํ•  ์ˆ˜ ์žˆ๋„๋ก ๋•๋Š” ๋ช‡ ๊ฐ€์ง€ ๊ฐ„๋‹จํ•˜๋ฉด์„œ๋„ ํšจ๊ณผ์ ์ธ ๊ตฌ์กฐ์  ์ธ์ฝ”๋”ฉ ๋ฐฉ๋ฒ•์„ ์ œ์•ˆํ•ฉ๋‹ˆ๋‹ค. ๋˜ํ•œ, ์šฐ๋ฆฌ๋Š” Graphormer์˜ ํ‘œํ˜„์„ ์ˆ˜ํ•™์ ์œผ๋กœ ํŠน์„ฑํ™”ํ•˜๊ณ , ๊ทธ๋ž˜ํ”„์˜ ๊ตฌ์กฐ์  ์ •๋ณด๋ฅผ ์ธ์ฝ”๋”ฉํ•˜๋Š” ์šฐ๋ฆฌ์˜ ๋ฐฉ์‹์œผ๋กœ ๋งŽ์€ ์ธ๊ธฐ ์žˆ๋Š” GNN ๋ณ€ํ˜•๋ชจ๋ธ๋“ค์ด Graphormer์˜ ํŠน์ˆ˜ํ•œ ๊ฒฝ์šฐ๋กœ ํฌํ•จ๋  ์ˆ˜ ์žˆ์Œ์„ ๋ณด์—ฌ์ค๋‹ˆ๋‹ค.

์ด ๋ชจ๋ธ์€ clefourrier๊ฐ€ ๊ธฐ์—ฌํ–ˆ์Šต๋‹ˆ๋‹ค. ์›๋ณธ ์ฝ”๋“œ๋Š” ์ด๊ณณ์—์„œ ํ™•์ธํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

์‚ฌ์šฉ ํŒ[[usage-tips]]

์ด ๋ชจ๋ธ์€ ํฐ ๊ทธ๋ž˜ํ”„(100๊ฐœ ์ด์ƒ์˜ ๋…ธ๋“œ๊ฐœ์ˆ˜/์—ฃ์ง€๊ฐœ์ˆ˜)์—์„œ๋Š” ๋ฉ”๋ชจ๋ฆฌ ์‚ฌ์šฉ๋Ÿ‰์ด ํญ๋ฐœ์ ์œผ๋กœ ์ฆ๊ฐ€ํ•˜๋ฏ€๋กœ ์ž˜ ์ž‘๋™ํ•˜์ง€ ์•Š์Šต๋‹ˆ๋‹ค. ๋Œ€์•ˆ์œผ๋กœ ๋ฐฐ์น˜ ํฌ๊ธฐ๋ฅผ ์ค„์ด๊ฑฐ๋‚˜, RAM์„ ๋Š˜๋ฆฌ๊ฑฐ๋‚˜ ๋˜๋Š” algos_graphormer.pyx ํŒŒ์ผ์˜ UNREACHABLE_NODE_DISTANCE ๋งค๊ฐœ๋ณ€์ˆ˜๋ฅผ ์ค„์ด๋Š” ๋ฐฉ๋ฒ•๋„ ์žˆ์ง€๋งŒ, 700๊ฐœ ์ด์ƒ์˜ ๋…ธ๋“œ๊ฐœ์ˆ˜/์—ฃ์ง€๊ฐœ์ˆ˜๋ฅผ ์ฒ˜๋ฆฌํ•˜๊ธฐ์—๋Š” ์—ฌ์ „ํžˆ ์–ด๋ ค์šธ ๊ฒƒ์ž…๋‹ˆ๋‹ค.

์ด ๋ชจ๋ธ์€ ํ† ํฌ๋‚˜์ด์ €๋ฅผ ์‚ฌ์šฉํ•˜์ง€ ์•Š๊ณ , ๋Œ€์‹  ํ›ˆ๋ จ ์ค‘์— ํŠน๋ณ„ํ•œ ์ฝœ๋ ˆ์ดํ„ฐ(collator)๋ฅผ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค.

GraphormerConfig[[transformers.GraphormerConfig]]

[[autodoc]] GraphormerConfig

GraphormerModel[[transformers.GraphormerModel]]

[[autodoc]] GraphormerModel - forward

GraphormerForGraphClassification[[transformers.GraphormerForGraphClassification]]

[[autodoc]] GraphormerForGraphClassification - forward