arXiv reaDer
胸部CTからの気道セグメンテーションのための共同3D UNet-Graphニューラルネットワークベースの方法
A joint 3D UNet-Graph Neural Network-based method for Airway Segmentation from chest CTs
3D UNetアーキテクチャをグラフニューラルネットワーク(GNN)モデルと組み合わせることにより、エンドツーエンドの深層学習セグメンテーション手法を提示します。このアプローチでは、UNetの最も深いレベルの畳み込み層が、一連のグラフ畳み込みを備えたGNNベースのモジュールに置き換えられます。このレベルの密な特徴マップは、GNNモジュールへのグラフ入力に変換されます。 UNetにグラフコンボリューションを組み込むと、ダウンサンプリングされたパスを介して学習されたローカルフィーチャに加えて、ノードの接続性に基づく情報がグラフ内のノードに提供されます。この情報は、セグメンテーションの決定を改善するのに役立ちます。複数のグラフ畳み込み層を積み重ねることにより、ノードは、計算コストを大幅に増加させることなく、高次の近隣情報にアクセスできます。グラフの隣接における2種類のノード接続を提案します。i)事前定義され、通常のノード周辺に基づいているものと、ii)トレーニング中に動的に計算され、特徴空間の最近傍ノードを使用するものです。この方法を、胸部CTスキャンから気道ツリーをセグメント化するタスクに適用しました。デンマークの肺がんスクリーニング試験データセットの32個のCTで実験が行われました。 2種類のグラフ隣接関係を持つUNet-GNNモデルのパフォーマンスを評価し、ベースラインUNetと比較します。
We present an end-to-end deep learning segmentation method by combining a 3D UNet architecture with a graph neural network (GNN) model. In this approach, the convolutional layers at the deepest level of the UNet are replaced by a GNN-based module with a series of graph convolutions. The dense feature maps at this level are transformed into a graph input to the GNN module. The incorporation of graph convolutions in the UNet provides nodes in the graph with information that is based on node connectivity, in addition to the local features learnt through the downsampled paths. This information can help improve segmentation decisions. By stacking several graph convolution layers, the nodes can access higher order neighbourhood information without substantial increase in computational expense. We propose two types of node connectivity in the graph adjacency: i) one predefined and based on a regular node neighbourhood, and ii) one dynamically computed during training and using the nearest neighbour nodes in the feature space. We have applied this method to the task of segmenting the airway tree from chest CT scans. Experiments have been performed on 32 CTs from the Danish Lung Cancer Screening Trial dataset. We evaluate the performance of the UNet-GNN models with two types of graph adjacency and compare it with the baseline UNet.
updated: Thu Aug 22 2019 20:32:01 GMT+0000 (UTC)
published: Thu Aug 22 2019 20:32:01 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト