arXiv reaDer
3D生物医学画像セグメンテーションのための動的線形変圧器
Dynamic Linear Transformer for 3D Biomedical Image Segmentation
トランスフォーマーベースのニューラルネットワークは、自己注意メカニズムからのより優れたグローバル情報モデリングにより、多くの生物医学画像セグメンテーションタスクで有望なパフォーマンスを上回りました。ただし、ほとんどの方法は、重要な3Dボリューム情報を無視して、2D医療画像用に設計されています。 3Dトランスフォーマーベースのセグメンテーション手法の主な課題は、自己注意メカニズムvaswani2017attentionによって導入された2次の複雑さです。この論文では、線形の複雑さを備えたエンコーダ-デコーダスタイルのアーキテクチャを使用して、3D医療画像セグメンテーションのための新しいトランスアーキテクチャを提案します。さらに、動的トークンの概念を新たに導入して、自己注意計算のトークン数をさらに削減します。グローバル情報モデリングを利用して、さまざまな階層段階からの不確実性マップを提供します。複数の挑戦的なCT膵臓セグメンテーションデータセットでこの方法を評価します。私たちの有望な結果は、私たちの新しい3D Transformerベースのセグメンターが、単一の注釈を使用して、有望な非常に実現可能なセグメンテーションパフォーマンスと正確な不確実性の定量化を提供できることを示しています。コードはhttps://github.com/freshman97/LinTransUNetで入手できます。
Transformer-based neural networks have surpassed promising performance on many biomedical image segmentation tasks due to a better global information modeling from the self-attention mechanism. However, most methods are still designed for 2D medical images while ignoring the essential 3D volume information. The main challenge for 3D transformer-based segmentation methods is the quadratic complexity introduced by the self-attention mechanism vaswani2017attention. In this paper, we propose a novel transformer architecture for 3D medical image segmentation using an encoder-decoder style architecture with linear complexity. Furthermore, we newly introduce a dynamic token concept to further reduce the token numbers for self-attention calculation. Taking advantage of the global information modeling, we provide uncertainty maps from different hierarchy stages. We evaluate this method on multiple challenging CT pancreas segmentation datasets. Our promising results show that our novel 3D Transformer-based segmentor could provide promising highly feasible segmentation performance and accurate uncertainty quantification using single annotation. Code is available https://github.com/freshman97/LinTransUNet.
updated: Wed Feb 01 2023 17:58:54 GMT+0000 (UTC)
published: Wed Jun 01 2022 21:15:01 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アソシエイト