arXiv reaDer
Point-Unet:ボリュームセグメンテーションのためのコンテキストアウェアポイントベースのニューラルネットワーク
Point-Unet: A Context-aware Point-based Neural Network for Volumetric Segmentation
ディープラーニングを使用した医療画像分析が最近普及しており、医療画像セグメンテーションとその兄弟であるボリューム画像セグメンテーションを含むさまざまなダウンストリームタスクで優れたパフォーマンスを示しています。特に、典型的な体積セグメンテーションネットワークは、体積データを個々のボクセル「スライス」のスタックとして扱うボクセルグリッド表現に強く依存しています。これにより、既存の画像ベースのセグメンテーションネットワークを拡張するのと同じくらい簡単にボクセルグリッドをセグメント化することを学習できます。 3Dドメイン。ただし、ボクセルグリッド表現を使用するには、大きなメモリフットプリント、高価なテスト時間、およびソリューションのスケーラビリティの制限が必要です。本論文では、3D点群を用いた深層学習の効率を体積セグメンテーションに組み込んだ新しい方法であるPoint-Unetを提案します。私たちの重要なアイデアは、最初に注意確率マップを学習することによってボリューム内の関心領域を予測することです。これは、ボリュームをスパースポイントクラウドにサンプリングするために使用され、その後、ポイントベースのニューラルネットワークを使用してセグメント化されます。小規模データセットの膵臓と大規模データセットのBraTS18、BraTS19、およびBraTS20の両方の課題を使用して、医療ボリュームセグメンテーションタスクの実験を実施しました。さまざまなメトリックに関する包括的なベンチマークは、コンテキストアウェアなPoint-Unetが、精度、トレーニング中のメモリ使用量、およびテスト中の時間消費の両方で、SOTAボクセルベースのネットワークを確実に上回っていることを示しています。私たちのコードはhttps://github.com/VinAIResearch/Point-Unetで入手できます。
Medical image analysis using deep learning has recently been prevalent, showing great performance for various downstream tasks including medical image segmentation and its sibling, volumetric image segmentation. Particularly, a typical volumetric segmentation network strongly relies on a voxel grid representation which treats volumetric data as a stack of individual voxel `slices', which allows learning to segment a voxel grid to be as straightforward as extending existing image-based segmentation networks to the 3D domain. However, using a voxel grid representation requires a large memory footprint, expensive test-time and limiting the scalability of the solutions. In this paper, we propose Point-Unet, a novel method that incorporates the efficiency of deep learning with 3D point clouds into volumetric segmentation. Our key idea is to first predict the regions of interest in the volume by learning an attentional probability map, which is then used for sampling the volume into a sparse point cloud that is subsequently segmented using a point-based neural network. We have conducted the experiments on the medical volumetric segmentation task with both a small-scale dataset Pancreas and large-scale datasets BraTS18, BraTS19, and BraTS20 challenges. A comprehensive benchmark on different metrics has shown that our context-aware Point-Unet robustly outperforms the SOTA voxel-based networks at both accuracies, memory usage during training, and time consumption during testing. Our code is available at https://github.com/VinAIResearch/Point-Unet.
updated: Wed Mar 16 2022 22:02:08 GMT+0000 (UTC)
published: Wed Mar 16 2022 22:02:08 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト