UNetとTransUNetのような最新の拡張機能は、近年、主要な医療画像セグメンテーション手法となっています。ただし、これらのネットワークは、パラメータが多く、計算が複雑で、使用が遅いため、ポイントオブケアアプリケーションでの迅速な画像セグメンテーションに効果的に採用することはできません。この目的のために、画像セグメンテーションのための畳み込み多層パーセプトロン(MLP)ベースのネットワークであるUNeXtを提案します。 UNeXtは、初期の畳み込み段階と潜在段階のMLP段階を使用して効果的に設計します。畳み込み特徴を効率的にトークン化して投影し、MLPを使用して表現をモデル化するトークン化されたMLPブロックを提案します。パフォーマンスをさらに向上させるために、ローカル依存関係の学習に焦点を合わせるために、MLPにフィードしながら入力のチャネルをシフトすることを提案します。潜在空間でトークン化されたMLPを使用すると、パラメーターの数と計算の複雑さが軽減されると同時に、セグメンテーションに役立つより良い表現が得られます。ネットワークは、さまざまなレベルのエンコーダーとデコーダー間のスキップ接続でも構成されています。複数の医療画像セグメンテーションデータセットでUNeXtをテストし、パラメータの数を72倍削減し、計算の複雑さを68倍削減し、推論速度を10倍向上させると同時に、最先端のセグメンテーションパフォーマンスを向上させることを示しています。医療画像セグメンテーションアーキテクチャ。コードはhttps://github.com/jeya-maria-jose/UNeXt-pytorchで入手できます。
UNet and its latest extensions like TransUNet have been the leading medical image segmentation methods in recent years. However, these networks cannot be effectively adopted for rapid image segmentation in point-of-care applications as they are parameter-heavy, computationally complex and slow to use. To this end, we propose UNeXt which is a Convolutional multilayer perceptron (MLP) based network for image segmentation. We design UNeXt in an effective way with an early convolutional stage and a MLP stage in the latent stage. We propose a tokenized MLP block where we efficiently tokenize and project the convolutional features and use MLPs to model the representation. To further boost the performance, we propose shifting the channels of the inputs while feeding in to MLPs so as to focus on learning local dependencies. Using tokenized MLPs in latent space reduces the number of parameters and computational complexity while being able to result in a better representation to help segmentation. The network also consists of skip connections between various levels of encoder and decoder. We test UNeXt on multiple medical image segmentation datasets and show that we reduce the number of parameters by 72x, decrease the computational complexity by 68x, and improve the inference speed by 10x while also obtaining better segmentation performance over the state-of-the-art medical image segmentation architectures. Code is available at https://github.com/jeya-maria-jose/UNeXt-pytorch