File size: 4,221 Bytes
17c6d62
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
<!--Copyright 2022 The HuggingFace Team and Microsoft. All rights reserved.

Licensed under the MIT License; you may not use this file except in compliance with
the License.

Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
specific language governing permissions and limitations under the License.

โš ๏ธ Note that this file is in Markdown but contain specific syntax for our doc-builder (similar to MDX) that may not be
rendered properly in your Markdown viewer.

-->

# Graphormer[[graphormer]]

<Tip warning={true}>

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

</Tip>

## ๊ฐœ์š”[[overview]]

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

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

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

์ด ๋ชจ๋ธ์€ [clefourrier](https://huggingface.co/clefourrier)๊ฐ€ ๊ธฐ์—ฌํ–ˆ์Šต๋‹ˆ๋‹ค. ์›๋ณธ ์ฝ”๋“œ๋Š” [์ด๊ณณ](https://github.com/microsoft/Graphormer)์—์„œ ํ™•์ธํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

## ์‚ฌ์šฉ ํŒ[[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