要約
PyTorch Geometricの紹介：これは、グラフ、ポイントクラウド、多様体などの不規則な構造を持つ入力データに対する深層学習を実現するためのライブラリで、PyTorchに基づいて構築されています。一般的なグラフデータ構造と処理方法に加えて、関係学習および3Dデータ処理の分野から最近発表された様々な手法を含んでいます。PyTorch Geometricは、スパースGPU加速の活用、専用のCUDAカーネルの提供、異なるサイズの入力例に対する効率的なミニバッチ処理の導入により、高いデータスループットを達成しています。本稿では、このライブラリについて詳細に説明し、均質な評価シナリオにおいて実装された手法の包括的な比較研究を行います。