Connectivity-Guided Sparsification of 2-FWL GNNs: Preserving Full Expressivity with Improved Efficiency
Created by
Haebom
Category
Empty
저자
Rongqin Chen, Fan Mo, Pak Lon Ip, Shenghui Zhang, Dan Wu, Ye Li, Leong Hou U
개요
본 논문은 2-FWL 테스트를 기반으로 하는 고차 그래프 신경망(HOGNN)의 계산 비용 문제를 해결하기 위해, 연결성을 고려한 희소화 프레임워크인 Co-Sparsify를 제안한다. Co-Sparsify는 2-FWL 표현력을 유지하면서 불필요한 계산을 제거하여 효율성을 높인다. 핵심 아이디어는 3-노드 상호작용이 이중 연결 성분(biconnected components) 내에서만 표현적으로 필요하다는 점을 활용한다. Co-Sparsify는 2-노드 메시지 전달을 연결 성분으로 제한하고 3-노드 상호작용을 이중 연결 성분으로 제한하여 계산을 제거한다. Co-Sparsified GNNs는 2-FWL 테스트와 동일한 표현력을 가지며, 합성 하위 구조 계산 작업에서 정확도를 유지하거나 향상시키고, ZINC 및 QM9과 같은 실제 벤치마크에서 최첨단 성능을 달성한다.
시사점, 한계점
•
시사점:
◦
Co-Sparsify는 고차 그래프 신경망의 계산 효율성을 향상시키는 새로운 프레임워크를 제시한다.