この作業では、DA-BDense-UNetと呼ばれる注意メカニズムとともにDenseUNetと双方向LSTMを統合することにより、新しいセグメンテーションネットワークを提案します。 DenseUNetを使用すると、情報の流れを調整することで、十分に多様な機能を学習し、ネットワークの代表的な能力を強化できます。双方向LSTMは、エンコードされた機能と、エンコードおよびデコードパスでアップサンプリングされた機能との関係を調査する役割を果たします。一方、DenseUNetにアテンションゲート(AG)を導入して、無関係な背景領域の応答を減らし、顕著な領域の応答を徐々に拡大します。さらに、双方向LSTMでの注意は、セグメンテーションの改善において、エンコードされた特徴とアップサンプリングされた特徴の寄与の違いを考慮に入れます。これにより、これら2種類の特徴の適切な重みを調整できます。複数の病院から収集された肝臓CT画像データセットを最新のセグメンテーションモデルと比較することにより、実験を行います。実験結果は、提案した方法DA-BDense-UNetがダイス係数に関して比較性能を達成したことを示しており、その有効性を示しています。
In this work, we propose a new segmentation network by integrating DenseUNet and bidirectional LSTM together with attention mechanism, termed as DA-BDense-UNet. DenseUNet allows learning enough diverse features and enhancing the representative power of networks by regulating the information flow. Bidirectional LSTM is responsible to explore the relationships between the encoded features and the up-sampled features in the encoding and decoding paths. Meanwhile, we introduce attention gates (AG) into DenseUNet to diminish responses of unrelated background regions and magnify responses of salient regions progressively. Besides, the attention in bidirectional LSTM takes into account the contribution differences of the encoded features and the up-sampled features in segmentation improvement, which can in turn adjust proper weights for these two kinds of features. We conduct experiments on liver CT image data sets collected from multiple hospitals by comparing them with state-of-the-art segmentation models. Experimental results indicate that our proposed method DA-BDense-UNet has achieved comparative performance in terms of dice coefficient, which demonstrates its effectiveness.