Are Vision xLSTM Embedded UNet More Reliable in Medical 3D Image Segmentation?

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Dutta, Pallabi, Bose, Soham, Roy, Swalpa Kumar, Mitra, Sushmita
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909725531045888
author Dutta, Pallabi
Bose, Soham
Roy, Swalpa Kumar
Mitra, Sushmita
author_facet Dutta, Pallabi
Bose, Soham
Roy, Swalpa Kumar
Mitra, Sushmita
contents The development of efficient segmentation strategies for medical images has evolved from its initial dependence on Convolutional Neural Networks (CNNs) to the current investigation of hybrid models that combine CNNs with Vision Transformers (ViTs). There is an increasing focus on creating architectures that are both high-performing and computationally efficient, capable of being deployed on remote systems with limited resources. Although transformers can capture global dependencies in the input space, they face challenges from the corresponding high computational and storage expenses involved. This research investigates the integration of CNNs with Vision Extended Long Short-Term Memory (Vision-xLSTM)s by introducing the novel U-VixLSTM. The Vision-xLSTM blocks capture the temporal and global relationships within the patches extracted from the CNN feature maps. The convolutional feature reconstruction path upsamples the output volume from the Vision-xLSTM blocks to produce the segmentation output. Our primary objective is to propose that Vision-xLSTM forms an appropriate backbone for medical image segmentation, offering excellent performance with reduced computational costs. The U-VixLSTM exhibits superior performance compared to the state-of-the-art networks in the publicly available Synapse, ISIC and ACDC datasets. Code provided: https://github.com/duttapallabi2907/U-VixLSTM
format Preprint
id arxiv_https___arxiv_org_abs_2406_16993
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Are Vision xLSTM Embedded UNet More Reliable in Medical 3D Image Segmentation?
Dutta, Pallabi
Bose, Soham
Roy, Swalpa Kumar
Mitra, Sushmita
Image and Video Processing
Computer Vision and Pattern Recognition
The development of efficient segmentation strategies for medical images has evolved from its initial dependence on Convolutional Neural Networks (CNNs) to the current investigation of hybrid models that combine CNNs with Vision Transformers (ViTs). There is an increasing focus on creating architectures that are both high-performing and computationally efficient, capable of being deployed on remote systems with limited resources. Although transformers can capture global dependencies in the input space, they face challenges from the corresponding high computational and storage expenses involved. This research investigates the integration of CNNs with Vision Extended Long Short-Term Memory (Vision-xLSTM)s by introducing the novel U-VixLSTM. The Vision-xLSTM blocks capture the temporal and global relationships within the patches extracted from the CNN feature maps. The convolutional feature reconstruction path upsamples the output volume from the Vision-xLSTM blocks to produce the segmentation output. Our primary objective is to propose that Vision-xLSTM forms an appropriate backbone for medical image segmentation, offering excellent performance with reduced computational costs. The U-VixLSTM exhibits superior performance compared to the state-of-the-art networks in the publicly available Synapse, ISIC and ACDC datasets. Code provided: https://github.com/duttapallabi2907/U-VixLSTM
title Are Vision xLSTM Embedded UNet More Reliable in Medical 3D Image Segmentation?
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2406.16993