arXiv reaDer
エンコーダ-デコーダアーキテクチャおよびそれ以降の双方向スキップ接続に向けて
Towards Bi-directional Skip Connections in Encoder-Decoder Architectures and Beyond
U-Netは、フォワードスキップ接続を備えたエンコーダ-デコーダアーキテクチャとして、さまざまな医療画像分析タスクで有望な結果を達成しています。最近の多くのアプローチでは、U-Netをより複雑なビルディングブロックで拡張しており、通常、ネットワークパラメータの数が大幅に増加しています。このような複雑さは、推論段階を臨床応用にとって非常に非効率的にします。効果的でありながら経済的なセグメンテーションネットワーク設計に向けて、この作業では、デコードされた機能をエンコーダに戻す後方スキップ接続を提案します。私たちの設計は、追加のパラメーターを導入することなく、繰り返し構造を形成するエンコーダー-デコーダーアーキテクチャーのフォワードスキップ接続と一緒に採用できます。後方スキップ接続を使用して、U-Netベースのネットワークファミリ、つまり双方向O字型ネットワークを提案します。これは、複数の公共医療画像セグメンテーションデータセットに新しいベンチマークを設定します。一方、最も単純なアーキテクチャ(BiO-Net)では、ネットワーク計算は、事前に設定された繰り返し時間とともに必然的に増加します。したがって、このような反復設計の欠陥のボトルネックを調査し、新しい2フェーズニューラルアーキテクチャ検索(NAS)アルゴリズム、つまりBiX-NASを提案して、最適なマルチスケール双方向スキップ接続を検索します。効果のないスキップ接続は、計算コストを削減し、ネットワーク推論を高速化するために破棄されます。最終的に検索されたBiX-Netは、ネットワークの複雑さを最小限に抑え、他の最先端の製品よりも大幅に優れています。合計6つのデータセットで、2Dと3Dの両方のセグメンテーションタスクでメソッドを評価します。提案された方法の包括的な分析を提供するために、広範なアブレーション研究も実施されました。
U-Net, as an encoder-decoder architecture with forward skip connections, has achieved promising results in various medical image analysis tasks. Many recent approaches have also extended U-Net with more complex building blocks, which typically increase the number of network parameters considerably. Such complexity makes the inference stage highly inefficient for clinical applications. Towards an effective yet economic segmentation network design, in this work, we propose backward skip connections that bring decoded features back to the encoder. Our design can be jointly adopted with forward skip connections in any encoder-decoder architecture forming a recurrence structure without introducing extra parameters. With the backward skip connections, we propose a U-Net based network family, namely Bi-directional O-shape networks, which set new benchmarks on multiple public medical imaging segmentation datasets. On the other hand, with the most plain architecture (BiO-Net), network computations inevitably increase along with the pre-set recurrence time. We have thus studied the deficiency bottleneck of such recurrent design and propose a novel two-phase Neural Architecture Search (NAS) algorithm, namely BiX-NAS, to search for the best multi-scale bi-directional skip connections. The ineffective skip connections are then discarded to reduce computational costs and speed up network inference. The finally searched BiX-Net yields the least network complexity and outperforms other state-of-the-art counterparts by large margins. We evaluate our methods on both 2D and 3D segmentation tasks in a total of six datasets. Extensive ablation studies have also been conducted to provide a comprehensive analysis for our proposed methods.
updated: Fri Mar 11 2022 01:38:52 GMT+0000 (UTC)
published: Fri Mar 11 2022 01:38:52 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アソシエイト