マルチGPUトレーニングで3Dポイントクラウドを処理するための効率的で非同期の並列処理アプローチであるSparsePipeを提案します。 SparsePipeは、点群などの3Dスパースデータをサポートするように構築されています。これは、スパーステンソル表現を使用した一般化された畳み込みを採用して、表現力豊かな高次元の畳み込みニューラルネットワークを構築することで実現されます。高密度ソリューションと比較して、新しいモデルは、空間全体を高密度にスライドすることなく、不規則な点群を効率的に処理できるため、メモリ要件が大幅に削減され、基盤となる3Dボリュームの解像度が高くなり、パフォーマンスが向上します。 SparsePipeは、入力データを複数のプロセッサに分割するバッチ内並列処理を活用し、通信とコンピューティングをオーバーラップさせるバッチ間パイプライン化によってトレーニングスループットをさらに向上させます。さらに、GPUが異種である場合にモデルを適切に分割し、コンピューティングの負荷が分散され、通信オーバーヘッドが削減されます。 8 GPUプラットフォームでの実験結果を使用して、SparsePipeが効果的に並列化し、トレーニングと推論の両方で、高密度ソリューションと比較して、現在のポイントクラウドベンチマークでより優れたパフォーマンスを取得できることを示します。
We propose SparsePipe, an efficient and asynchronous parallelism approach for handling 3D point clouds with multi-GPU training. SparsePipe is built to support 3D sparse data such as point clouds. It achieves this by adopting generalized convolutions with sparse tensor representation to build expressive high-dimensional convolutional neural networks. Compared to dense solutions, the new models can efficiently process irregular point clouds without densely sliding over the entire space, significantly reducing the memory requirements and allowing higher resolutions of the underlying 3D volumes for better performance. SparsePipe exploits intra-batch parallelism that partitions input data into multiple processors and further improves the training throughput with inter-batch pipelining to overlap communication and computing. Besides, it suitably partitions the model when the GPUs are heterogeneous such that the computing is load-balanced with reduced communication overhead. Using experimental results on an eight-GPU platform, we show that SparsePipe can parallelize effectively and obtain better performance on current point cloud benchmarks for both training and inference, compared to its dense solutions.