Command Palette

Search for a command to run...

4달 전

다중 마스크 아그리게이터를 활용한 그래프 신경망

{Ahmet Sureyya Rifaioglu Ahmet Sarıgün}

다중 마스크 아그리게이터를 활용한 그래프 신경망

초록

그래프 신경망(GNN)에서 가장 핵심적인 연산 중 하나는 타깃 노드의 이웃 노드로부터 정보를 추출하는 데 목적이 있는 집계(aggregation) 연산이다. 이를 위해 기존에 표준 그래프 컨볼루션(GCN), 그래프 어텐션(GAT), 메시지 전달(MPNN) 등 다양한 컨볼루션 방법이 제안되어 왔다. 본 연구에서는 각 집계기(aggregator)에 대해 이웃 메시지를 수집하기 전에 가중치 마스크를 학습하는 다중 마스크 집계기(Multi-Mask Aggregators, MMA)라는 새로운 집계 방법을 제안한다. MMA는 GAT과 MPNN과 유사한 구조를 가지나, 이론적 및 실용적인 측면에서 몇 가지 장점을 지닌다. 직관적으로 본 연구의 프레임워크는 GAT의 헤드 수 제한에 얽매이지 않으며, MPNN보다 더 뛰어난 구분 능력을 갖는다. MMA의 성능은 널리 사용되는 벤치마크 데이터셋에서 노드 분류 및 그래프 회귀 작업 모두에서 잘 알려진 기준 모델들과 비교되었으며, 전반적으로 개선된 성능을 보였다. 데이터셋과 코드는 https://github.com/asarigun/mma 에서 제공된다.

벤치마크

벤치마크방법론지표
graph-regression-on-zincMMA
MAE: 0.156
node-classification-on-citeseerMMA
Accuracy: 76.30%
node-classification-on-coraMMA
Accuracy: 85.80%
node-classification-on-pubmedMMA
Accuracy: 86.00%

AI로 AI 구축

아이디어에서 출시까지 — 무료 AI 공동 코딩, 즉시 사용 가능한 환경, 최적 가격 GPU로 AI 개발을 가속화하세요.

AI 공동 코딩
즉시 사용 가능한 GPU
최적 가격
시작하기

Hyper Newsletters

최신 정보 구독하기
한국 시간 매주 월요일 오전 9시 에 이번 주의 최신 업데이트를 메일로 발송합니다
이메일 서비스 제공: MailChimp
다중 마스크 아그리게이터를 활용한 그래프 신경망 | 연구 논문 | HyperAI초신경