arXiv reaDer
MALUNet: 皮膚病変セグメンテーションのためのマルチアテンション軽量 UNet
MALUNet: A Multi-Attention and Light-weight UNet for Skin Lesion Segmentation
最近、一部の先駆的な研究では、セグメンテーションのパフォーマンスを向上させるために、より複雑なモジュールを適用することが好まれています。ただし、コンピューティング リソースが限られているため、実際の臨床環境には適していません。この課題に対処するために、軽量モデルを提案して、これまでのところ最小のパラメーターコストと計算複雑さで皮膚病変セグメンテーションの競争力のあるパフォーマンスを達成します。簡単に言えば、4 つのモジュールを提案します。 (2) IEA。これは、データセット全体を特徴付け、サンプル間のつながりを強化するための外部の注意に基づいています。 (3) CAB は 1D 畳み込みと完全に接続されたレイヤーで構成され、マルチステージ機能のグローバルおよびローカル融合を実行して、チャネル軸でアテンション マップを生成します。 (4) SAB。共有 2D 畳み込みによって多段階の機能を操作して、空間軸でアテンション マップを生成します。 4 つのモジュールを U 字型アーキテクチャと組み合わせて、MALUNet と呼ばれる軽量の医用画像セグメンテーション モデルを取得します。 UNet と比較すると、私たちのモデルは mIoU と DSC の指標をそれぞれ 2.39% と 1.49% 改善し、パラメーターの数と計算の複雑さを 44 倍と 166 倍削減しています。さらに、2 つの皮膚病変セグメンテーション データセット (ISIC2017 および ISIC2018) の比較実験を行います。実験結果は、私たちのモデルがパラメーターの数、計算の複雑さ、およびセグメンテーションのパフォーマンスのバランスを取ることで最先端を達成することを示しています。コードは https://github.com/JCruan519/MALUNet で入手できます。
Recently, some pioneering works have preferred applying more complex modules to improve segmentation performances. However, it is not friendly for actual clinical environments due to limited computing resources. To address this challenge, we propose a light-weight model to achieve competitive performances for skin lesion segmentation at the lowest cost of parameters and computational complexity so far. Briefly, we propose four modules: (1) DGA consists of dilated convolution and gated attention mechanisms to extract global and local feature information; (2) IEA, which is based on external attention to characterize the overall datasets and enhance the connection between samples; (3) CAB is composed of 1D convolution and fully connected layers to perform a global and local fusion of multi-stage features to generate attention maps at channel axis; (4) SAB, which operates on multi-stage features by a shared 2D convolution to generate attention maps at spatial axis. We combine four modules with our U-shape architecture and obtain a light-weight medical image segmentation model dubbed as MALUNet. Compared with UNet, our model improves the mIoU and DSC metrics by 2.39% and 1.49%, respectively, with a 44x and 166x reduction in the number of parameters and computational complexity. In addition, we conduct comparison experiments on two skin lesion segmentation datasets (ISIC2017 and ISIC2018). Experimental results show that our model achieves state-of-the-art in balancing the number of parameters, computational complexity and segmentation performances. Code is available at https://github.com/JCruan519/MALUNet.
updated: Thu Nov 03 2022 13:19:22 GMT+0000 (UTC)
published: Thu Nov 03 2022 13:19:22 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アソシエイト