Re:Vision Mamba: Efficient Visual Representation Learning wi...
by wityl on Thursday February 01, 2024 @03:48PM
(#39221234)
They were previously parallelizable (via fft), but performed poorly on language modeling tasks.
Mamba adds a dependence on the inputs that makes language modeling competitive with transformers, but that prevents using the fft approach. So they switch to a method using parallel prefix scan.
Mamba adds a dependence on the inputs that makes language modeling competitive with transformers, but that prevents using the fft approach. So they switch to a method using parallel prefix scan.