arXiv reaDer
医療画像セグメンテーション用の混合変圧器U-Net
Mixed Transformer U-Net For Medical Image Segmentation
U-Netは、医療画像のセグメンテーションタスクで大きな成功を収めていますが、長距離の依存関係を明示的にモデル化する機能がありません。したがって、Vision Transformerは、Self-Attention(SA)を介して長距離の相関関係をキャプチャする固有の機能により、最近、代替のセグメンテーション構造として登場しました。ただし、Transformerは通常、大規模な事前トレーニングに依存しており、計算が非常に複雑です。さらに、SAは、データセット全体の潜在的な相関関係を無視して、単一のサンプル内の自己親和性のみをモデル化できます。これらの問題に対処するために、アフィニティ間およびアフィニティ内の同時学習用に、Mixed Transformer Module(MTM)という名前の新しいTransformerモジュールを提案します。 MTMはまず、適切に設計されたローカル-グローバルガウス加重自己注意(LGG-SA)を使用して自己親和性を効率的に計算します。次に、External Attention(EA)を介してデータサンプル間の相互接続をマイニングします。 MTMを使用して、正確な医療画像セグメンテーションのために、Mixed Transformer U-Net(MT-UNet)という名前のU字型モデルを構築します。 2つの異なる公開データセットでメソッドをテストし、実験結果は、提案されたメソッドが他の最先端のメソッドよりも優れたパフォーマンスを達成することを示しています。コードはhttps://github.com/Dootmaan/MT-UNetで入手できます。
Though U-Net has achieved tremendous success in medical image segmentation tasks, it lacks the ability to explicitly model long-range dependencies. Therefore, Vision Transformers have emerged as alternative segmentation structures recently, for their innate ability of capturing long-range correlations through Self-Attention (SA). However, Transformers usually rely on large-scale pre-training and have high computational complexity. Furthermore, SA can only model self-affinities within a single sample, ignoring the potential correlations of the overall dataset. To address these problems, we propose a novel Transformer module named Mixed Transformer Module (MTM) for simultaneous inter- and intra- affinities learning. MTM first calculates self-affinities efficiently through our well-designed Local-Global Gaussian-Weighted Self-Attention (LGG-SA). Then, it mines inter-connections between data samples through External Attention (EA). By using MTM, we construct a U-shaped model named Mixed Transformer U-Net (MT-UNet) for accurate medical image segmentation. We test our method on two different public datasets, and the experimental results show that the proposed method achieves better performance over other state-of-the-art methods. The code is available at: https://github.com/Dootmaan/MT-UNet.
updated: Mon Nov 08 2021 09:03:46 GMT+0000 (UTC)
published: Mon Nov 08 2021 09:03:46 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アソシエイト