Core Concepts
GNNBENCH는 시스템 혁신에 초점을 맞춘 플러그 앤 플레이 벤치마킹 플랫폼으로, 안정적인 시스템 API와 생산성 및 공정성을 제공하여 GNN 시스템 연구를 발전시킨다.
Abstract
GNNBENCH는 GNN 시스템 연구를 발전시키기 위해 제안된 플러그 앤 플레이 벤치마킹 플랫폼이다.
안정적인 시스템 API:
DL 프레임워크의 제한을 극복하기 위해 제안된 제작자 전용 DLPack 프로토콜을 사용하여 시스템 API의 안정성을 보장한다.
이를 통해 사용자 정의 클래스와 같은 강력한 API를 제공하며, DL 프레임워크에 독립적이다.
생산성 및 공정성:
공통 GNN 모델 프런트엔드와 독립적인 GNNBENCH-System을 통해 연구자들이 시스템 혁신을 빠르게 프로토타이핑하고 공정한 평가를 받을 수 있다.
도메인 특화 언어(DSL)를 사용하여 통합 코드를 자동으로 생성하여 생산성을 높인다.
DL 프레임워크 독립성:
GNNBENCH-System은 PyTorch, TensorFlow 등 다양한 DL 프레임워크와 통합될 수 있다.
새로운 아티팩트:
GNNBENCH를 통해 기존 GNN 시스템의 오류를 자동으로 수정하고 새로운 아티팩트를 생성할 수 있다.
평가 통찰:
GNNBENCH 평가를 통해 다양한 통찰을 얻을 수 있다. 예를 들어 DGL의 프레임워크 오버헤드, 다양한 GNN 시스템의 성능 및 메모리 사용량 등을 확인할 수 있다.
Stats
DGL은 프레임워크 관련 메모리 오버헤드가 크다.
dgNN은 GAT 모델에서 GB Cusparse 대비 45.77% 느리지만 메모리는 16.44% 절감한다.