MambaOut

MambaOut是一个基于PyTorch实现的计算机视觉模型，旨在探讨在视觉任务中是否真的需要复杂的Mamba架构。该项目的核心功能是通过使用Gated CNN块构建一系列简化模型，以较少的参数和计算量达到甚至超越现有复杂模型（如ViT-S）在ImageNet上的分类精度。技术特点包括高效的Gated CNN设计及对Transformer FLOPs的优化计算方法。MambaOut特别适合于那些追求高效性能同时又希望保持高准确率的应用场景，比如移动设备或边缘计算环境中的图像识别任务。