Stefan A Baumann

ZigMa: A DiT-style Zigzag Mamba Diffusion Model

"The diffusion model has long been plagued by scalability and quadratic complexity issues, especially within transformer-based structures. In this study, we aim to leverage the long sequence modeling capability of a State-Space Model called Mamba to extend its applicability to visual data generation…