arXiv reaDer
NexToU: 医用画像セグメンテーションのための効率的なトポロジー対応 U-Net
NexToU: Efficient Topology-Aware U-Net for Medical Image Segmentation
畳み込みニューラル ネットワーク (CNN) と Transformer のバリアントは、主要な医療画像セグメンテーション バックボーンとして台頭しています。それにもかかわらず、これらのバックボーンは、全体的な画像のコンテキストを保存したり、視覚オブジェクトの不規則な形状を効率的に処理したりすることに限界があるため、さまざまな解剖学的領域からの情報を効果的に統合し、特に血管系の個人間のばらつきを軽減するのに苦労しています。さまざまな分野にわたるトポロジー特性と非ユークリッド関係の捕捉におけるグラフ ニューラル ネットワーク (GNN) の成功したブレークスルーを動機として、私たちは医療画像セグメンテーションのための新しいハイブリッド アーキテクチャである NexToU を提案します。 NexToU は、計算コストを最小限に抑えながらグローバルとローカルのトポロジ表現の両方を学習するための、Vision GNN (ViG) の改良された Pool GNN および Swin GNN モジュールで構成されています。さまざまな解剖学的構造間の包含と排除の関係に対処するために、バイナリ ツリーの性質に基づいてトポロジカル インタラクション (TI) モジュールを再定式化し、トポロジカル制約を NexToU に迅速にエンコードします。 3 つのデータセット (異なる画像次元、疾患の種類、および画像モダリティを含む) に対して行われた広範な実験により、私たちの手法が他の最先端 (SOTA) アーキテクチャよりも一貫して優れていることが実証されました。すべてのコードは https://github.com/PengchengShi1220/NexToU で公開されています。
Convolutional neural networks (CNN) and Transformer variants have emerged as the leading medical image segmentation backbones. Nonetheless, due to their limitations in either preserving global image context or efficiently processing irregular shapes in visual objects, these backbones struggle to effectively integrate information from diverse anatomical regions and reduce inter-individual variability, particularly for the vasculature. Motivated by the successful breakthroughs of graph neural networks (GNN) in capturing topological properties and non-Euclidean relationships across various fields, we propose NexToU, a novel hybrid architecture for medical image segmentation. NexToU comprises improved Pool GNN and Swin GNN modules from Vision GNN (ViG) for learning both global and local topological representations while minimizing computational costs. To address the containment and exclusion relationships among various anatomical structures, we reformulate the topological interaction (TI) module based on the nature of binary trees, rapidly encoding the topological constraints into NexToU. Extensive experiments conducted on three datasets (including distinct imaging dimensions, disease types, and imaging modalities) demonstrate that our method consistently outperforms other state-of-the-art (SOTA) architectures. All the code is publicly available at https://github.com/PengchengShi1220/NexToU.
updated: Thu May 25 2023 10:18:57 GMT+0000 (UTC)
published: Thu May 25 2023 10:18:57 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アソシエイト