May, 2024
Mamba-R:Vision Mamba 也需要寄存器
Mamba-R: Vision Mamba ALSO Needs Registers
Feng Wang, Jiahao Wang, Sucheng Ren, Guoyizhe Wei, Jieru Mei...
TL;DRVision Mamba 的特征图中存在与低信息背景区域中出现的高范数令牌对应的众多伪像,此问题通过向 Vision Mamba 引入寄存器令牌解决,改进后的 Mamba-R 架构在性能和扩展性方面具有更强的表现。