이 논문은 그래프 변환기의 확장성을 높이기 위한 AnchorGT 모델을 제안한다.
그래프 변환기는 메시지 전달 그래프 신경망의 한계를 극복하고 우수한 성능과 표현력을 보여주지만, 자기 주목 메커니즘의 2차 복잡도로 인해 확장성이 제한되어 왔다. 이를 해결하기 위해 이전 연구들은 수용 범위 제한 또는 선형 주목 방법을 사용했지만, 이는 각각 전역 수용 범위 감소와 구조 표현력 저하의 문제가 있었다.
AnchorGT는 k-지배 집합 앵커를 사용하여 전역 수용 범위를 유지하면서도 거의 선형 복잡도를 달성한다. 앵커 기반 주목 메커니즘을 통해 각 노드가 이웃과 앵커 노드에 모두 주목할 수 있도록 하여, 계산 복잡도를 낮추면서도 구조 정보를 효과적으로 학습할 수 있다. 또한 이론적으로 AnchorGT가 Weisfeiler-Lehman 테스트보다 강한 표현력을 가질 수 있음을 증명한다.
실험 결과, AnchorGT 기반 그래프 변환기 모델들이 기존 모델과 유사한 성능을 보이면서도 메모리 사용량과 훈련 시간이 크게 감소하는 것을 확인했다.
翻譯成其他語言
從原文內容
arxiv.org
深入探究