이 논문은 그래프 신경망(GNN)을 활용하여 그래프 분류 문제를 해결하는 SPGNN 모델을 제안한다. SPGNN은 두 가지 핵심 기여점을 가진다:
그래프 컨볼루션 메커니즘: SPGNN은 기존 GNN의 단순 합산 기반 집계 방식의 한계를 극복하기 위해 연결 기반 집계 방식을 제안한다. 이를 통해 노드 표현을 더욱 구분력 있게 업데이트할 수 있다.
WL-SortPool 기법: SPGNN은 새로운 그래프 풀링 모듈인 WL-SortPool을 제안한다. WL-SortPool은 각 레이어에서 노드 표현을 중요도 순으로 정렬하여 다양한 크기의 중요 하위 그래프 패턴을 효과적으로 추출한다. 이를 통해 그래프 전체 구조 정보를 잘 보존할 수 있다.
실험 결과, SPGNN은 다양한 그래프 분류 벤치마크 데이터셋에서 기존 그래프 커널 방법과 다른 GNN 모델 대비 우수한 성능을 보였다. 이는 SPGNN이 그래프 내 중요 하위 구조 패턴을 효과적으로 인식할 수 있음을 보여준다.
إلى لغة أخرى
من محتوى المصدر
arxiv.org
الرؤى الأساسية المستخلصة من
by Zehao Dong,M... في arxiv.org 04-23-2024
https://arxiv.org/pdf/2404.13655.pdfاستفسارات أعمق