arXiv reaDer
Graph Refinement based Airway Extraction using Mean-Field Networks and Graph Neural Networks
 グラフの改良、または完全すぎるグラフから対象のサブグラフを取得するタスクには、多くのさまざまなアプリケーションがあります。この作業では、画像データからツリーまたはサブツリーのコレクションを抽出します。まず、ボリュームデータのグラフベースの表現を導出し、次にツリー抽出をグラフ改良タスクとしてポーズを設定します。グラフの改良を実行する2つの方法を紹介します。最初に、平均場近似(MFA)を使用して、対象の最適サブグラフを推定できるサブグラフの事後密度を近似します。平均フィールドネットワーク(MFN)は、MFAの反復はニューラルネットワークのフィードフォワード操作と見なすことができるという解釈に基づく推論に使用されます。これにより、勾配降下法を使用してモデルパラメーターを学習できます。次に、MFNの一般化と見なすことができるグラフニューラルネットワーク(GNN)を使用した教師あり学習アプローチを紹介します。サブグラフは、GNNベースのグラフ改良モデルをトレーニングして、エッジ確率を直接予測することによって取得されます。 2つの方法のクラス間の関係を議論し、3D、低線量、胸部CTデータから気道を抽出するタスクのためにそれらを比較します。 MFNモデルとGNNモデルの両方が、EXACT'09チャレンジの最高パフォーマンスメソッドと同様の1つのベースラインメソッド、および3D U-Netベースの気道セグメンテーションモデルと比較すると、誤検知が少なくなります。
Graph refinement, or the task of obtaining subgraphs of interest from over-complete graphs, can have many varied applications. In this work, we extract trees or collection of sub-trees from image data by, first deriving a graph-based representation of the volumetric data and then, posing the tree extraction as a graph refinement task. We present two methods to perform graph refinement. First, we use mean-field approximation (MFA) to approximate the posterior density over the subgraphs from which the optimal subgraph of interest can be estimated. Mean field networks (MFNs) are used for inference based on the interpretation that iterations of MFA can be seen as feed-forward operations in a neural network. This allows us to learn the model parameters using gradient descent. Second, we present a supervised learning approach using graph neural networks (GNNs) which can be seen as generalisations of MFNs. Subgraphs are obtained by training a GNN-based graph refinement model to directly predict edge probabilities. We discuss connections between the two classes of methods and compare them for the task of extracting airways from 3D, low-dose, chest CT data. We show that both the MFN and GNN models show significant improvement when compared to one baseline method, that is similar to a top performing method in the EXACT'09 Challenge, and a 3D U-Net based airway segmentation model, in detecting more branches with fewer false positives.
updated: Tue Jun 02 2020 16:14:58 GMT+0000 (UTC)
published: Wed Nov 21 2018 10:50:31 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト