초록
PyTorch Geometric을 소개합니다. 이 라이브러리는 그래프, 포인트 클라우드, 매니폴드와 같은 규칙적이지 않은 구조의 입력 데이터에 대한 딥 러닝을 위해 PyTorch 기반으로 개발되었습니다. 일반적인 그래프 데이터 구조와 처리 방법 외에도, 관계 학습과 3D 데이터 처리 분야에서 최근 발표된 다양한 방법들을 포함하고 있습니다. PyTorch Geometric은 희소 GPU 가속을 활용하고, 전용 CUDA 커널을 제공하며, 크기가 다른 입력 예제를 위한 효율적인 미니 배치 처리를 도입하여 높은 데이터 처리량을 달성합니다. 본 연구에서는 라이브러리를 자세히 설명하고, 동일한 평가 시나리오에서 구현된 방법들의 종합적인 비교 연구를 수행합니다.