Bidirectional Mamba state-space model for anomalous diffusion

Lavaud, Maxime, Shokeeb, Yosef, Lacherez, Juliette, Amarouchene, Yacine, Salez, Thomas

arXiv.org Machine Learning 

Characterizing anomalous diffusion is crucial in order to understand the evolution of complex stochastic systems, from molecular interactions to cellular dynamics. In this work, we characterize the performances regarding such a task of Bi-Mamba, a novel state-space deep-learning architecture articulated with a bidirectional scan mechanism. Our implementation is tested on the AnDi-2 challenge datasets among others. As such, our results indicate the potential practical use of the Bi-Mamba architecture for anomalousdiffusion characterization. Deep-learning methods for advanced microscopy have thus emerged as a promising change of paradigm [13].