医療画像は、ピクセルの認識が困難なクラス間の区別、高い変動性、およびノイズによって特徴付けられます。 1つのレベルからコンテキスト情報をキャプチャする以前の自己注意ベースの方法とは異なり、高次グラフの観点から自己注意メカニズムを再定式化し、Hierarchical Attention Network(HANet)の問題に対処する新しい方法を提案します。医用画像のセグメンテーション。具体的には、HANetに組み込まれたHAモジュールは、複数のレベルの隣接ノードからコンテキスト情報をキャプチャします。これらの隣接ノードは、高次グラフから抽出されます。高次グラフでは、2つのノード間の相関が十分に高い場合にのみ2つのノード間にエッジがあります。これにより、クラス間の不区別によって引き起こされるノイズの多い注意情報が自然に減少します。提案されたHAモジュールは、入力の分散に対してロバストであり、既存の畳み込みニューラルネットワークに柔軟に挿入できます。視神経乳頭/カップのセグメンテーション、血管のセグメンテーション、肺のセグメンテーションを含む3つの医療画像セグメンテーションタスクの実験を行います。広範な結果は、私たちの方法が既存の最先端の方法よりも効果的で堅牢であることを示しています。
The medical image is characterized by the inter-class indistinction, high variability, and noise, where the recognition of pixels is challenging. Unlike previous self-attention based methods that capture context information from one level, we reformulate the self-attention mechanism from the view of the high-order graph and propose a novel method, namely Hierarchical Attention Network (HANet), to address the problem of medical image segmentation. Concretely, an HA module embedded in the HANet captures context information from neighbors of multiple levels, where these neighbors are extracted from the high-order graph. In the high-order graph, there will be an edge between two nodes only if the correlation between them is high enough, which naturally reduces the noisy attention information caused by the inter-class indistinction. The proposed HA module is robust to the variance of input and can be flexibly inserted into the existing convolution neural networks. We conduct experiments on three medical image segmentation tasks including optic disc/cup segmentation, blood vessel segmentation, and lung segmentation. Extensive results show our method is more effective and robust than the existing state-of-the-art methods.