search

wurenkai / UltraLight-VM-UNet

[arXiv] The official code for "UltraLight VM-UNet: Parallel Vision Mamba Significantly Reduces Parameters for Skin Lesion Segmentation".
235 stars 33 forks source link

Issue on PVMlayer in the model #12

Open QuantumLuckin opened 7 months ago

QuantumLuckin commented 7 months ago

Good work! But it seems that in the model you proposed, the PVMlayer directly uses Albert Gu's Mamba instead of the SS2D in Vmamba. However, in Figure 4 (b) of your article, the PVMlayer uses Vmamba's VSS block and SS2D. May I ask if I am right?

wurenkai commented 7 months ago

Hi. Yes, you are right. The application of VSS Block and SS2D will increase the memory burden, but it is still to significantly reduce the memory burden on this basis using our parallel approach (which we tried), which is still a variant of Mamba at their core. Here, the parallel approach is the winner for lightweighting. We are still conducting richer experiments with this work recently, and we will be modifying this in a future arXiv version. Thanks.