arXiv reaDer
低GPUメモリ要件でSpatialConfiguration-Netを使用した効率的な多臓器セグメンテーション
Efficient Multi-Organ Segmentation Using SpatialConfiguration-Net with Low GPU Memory Requirements
多くの医療データセットで適切に実行できる多くのセマンティックセグメンテーション方法が存在しますが、多くの場合、それらは臨床診療で直接使用するように設計されていません。 2つの主な懸念事項は、異なるスキャナーを使用して取得した画像など、外観が異なる目に見えないデータへの一般化と、計算時間と必要なグラフィックスプロセッシングユニット(GPU)メモリの観点からの効率です。この作業では、SpatialConfiguration-Net(SCN)に基づく多臓器セグメンテーションモデルを採用します。これは、ラベル付けされた臓器間の空間構成に関する事前知識を統合して、ネットワーク出力の偽の応答を解決します。さらに、セグメンテーションモデルのアーキテクチャを変更して、予測の品質に大幅な影響を与えることなく、メモリフットプリントを可能な限り削減しました。最後に、実行時間と必要なGPUメモリの両方を最適化するための最小限の推論スクリプトを実装しました。
Even though many semantic segmentation methods exist that are able to perform well on many medical datasets, often, they are not designed for direct use in clinical practice. The two main concerns are generalization to unseen data with a different visual appearance, e.g., images acquired using a different scanner, and efficiency in terms of computation time and required Graphics Processing Unit (GPU) memory. In this work, we employ a multi-organ segmentation model based on the SpatialConfiguration-Net (SCN), which integrates prior knowledge of the spatial configuration among the labelled organs to resolve spurious responses in the network outputs. Furthermore, we modified the architecture of the segmentation model to reduce its memory footprint as much as possible without drastically impacting the quality of the predictions. Lastly, we implemented a minimal inference script for which we optimized both, execution time and required GPU memory.
updated: Tue Jan 04 2022 11:00:03 GMT+0000 (UTC)
published: Fri Nov 26 2021 17:47:10 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アソシエイト