Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to understand the relationship between mamba2 and GLA? #380

Open
alpacaduby opened this issue Jun 10, 2024 · 0 comments
Open

How to understand the relationship between mamba2 and GLA? #380

alpacaduby opened this issue Jun 10, 2024 · 0 comments

Comments

@alpacaduby
Copy link

alpacaduby commented Jun 10, 2024

Hi, author, thank you for sharing your fantastic work.

When I was reading things about mamba, I found that in mamba-mini , it says that GLA is a special case of mamba and when the dimension of w changed from [batch, seqlen, dtstate, dim] to [batch, seqlen, dtstate], then they are equalivent.

image

The author of VMamba also suggested that in the arxiv paper:
image

Then I found that in the SSD, which is the core component of mamba2, the dimension of matrix A $\odot$ dt is also reduced to [..., nheads], which is may suggests that the matrix w has been reduced to [batch, seqlen]. So my question is, is mamba2 a special case of GLA?

Moreover, I did an experiment testing mamba2 and GLA, and found that they almost share the same result with each other, only with numerical differences (1e-5).

glavsmamba

So, How to understand the relationship between mamba1, mamba2 and GLA?

@alpacaduby alpacaduby changed the title What is the difference between mamba2 and GLA? How to understand the relationship between mamba2 and GLA? Jun 10, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant