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

PVM in decoder blocks #11

Open fceex49 opened 7 months ago

fceex49 commented 7 months ago

Hi

one question regarding decoder blocks. In the LightMamba paper they are not using S6 layers in the decoder blocks.

You are however using S6 there. Any intuitions?

wurenkai commented 7 months ago

Hi, you can learn about our ablation experiment 2. After replacing the PVM Layer of the decoder, there is a degradation in performance and an increase in operational burden.