Graph Signal Processing Meets Mamba2: Adaptive Filter Bank via Delta Modulation
Organizations: KAIST
Abstract
State-space models (SSMs) offer efficient alternatives to attention with linear-time recurrence. Mamba2, a recent SSM-based language model, uses selective input gating and a multi-head structure, enabling parallel computation and strong benchmark performance. However, its multi-head recurrence operates independently without structured utilization or analysis. In this work, we propose a novel method called Hierarchical ADaptive filter bank for Efficient SSMs (HADES), a Graph Signal Processing (GSP)-inspired framework that reinterprets Mamba2 as an adaptive filter bank on a line graph. Our hierarchical architecture introduces two filter types: shared filters for global low-pass behavior and expert filters for local high-pass behavior, achieved through structured bias on the parameter Δ. HADES achieves comparable performance to baseline models including Mamba2 across various benchmarks in language modeling, commonsense reasoning, and long-context retrieval, while using only 58.9% of the original parameters. In this regard, HADES bridges GSP and neural sequence modeling, enabling efficient, hierarchical, and interpretable filtering within state-space models.
Figures & tables
| Model | Train. | Wiki. | LMB. | LMB. | BoolQ | Hella. | Wino. | ARC-e | ARC-c | PIQA | OBQA. | Avg. |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| ppl | ppl | ppl | acc | acc | acc_n | acc | acc | acc | acc | acc_n | 8 tasks | |
| Linear Transformer | 2.49 | 45.43 | 73.93 | 24.06 | 61.50 | 28.20 | 51.30 | 42.05 | 21.76 | 60.55 | 27.60 | 39.63 |
| RetNet | 2.41 | 34.12 | 29.46 | 35.36 | 55.57 | 31.31 | 51.70 | 44.49 | 2 3.46 | 62.40 | 28.00 | 41.54 |
| DeltaNet | 2.29 | 33.25 | 26.82 | 35.75 | 54.07 | 31.40 | 49.96 | 44.11 | 22.18 | 63.60 | 29.60 | 41.33 |
| Mamba1 | 2.53 | 47.51 | 85.53 | 22.43 | 62.17 | 28.71 | 50.67 | 42.09 | 22.35 | 60.72 | 26.60 | 39.47 |
| Mamba2 | 2.33 | 31.34 | 2 4.38 | 3 6.46 | 53.88 | 3 2.62 | 50.83 | 45.29 | 24.15 | 63.44 | 26.40 | 4 1.63 |
| Methods | Wiki. | LMB. | Avg. |
|---|---|---|---|
| ppl | ppl | 8 tasks | |
| HADES (Ours) | 31.51 | 21.74 | 42.91 |
| w/o | 34.73 | 26.84 | 41.57 |
| w/o | 33.83 | 27.40 | 42.15 |
| Only Shared Filters | 34.55 | 27.64 | 42.21 |
| Only Expert Filters | 36.34 | 30.12 | 41.68 |
Appendix figures & tables18 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Wiki. | LMB. | LMB. | BoolQ | Hella. | Wino. | ARC-e | ARC-c | PIQA | OBQA. | Avg. |
|---|---|---|---|---|---|---|---|---|---|---|---|
| ppl | ppl | acc | acc | acc_n | acc | acc | acc | acc | acc_n | 8 tasks | |
| Linear Transformer | 45.43 | 73.93 | 24.06 | 61.50 | 28.20 | 51.30 | 42.05 | 21.76 | 60.55 | 27.60 | 39.63 |
| (N/A) | (4.168) | (0.006) | (0.009) | (0.005) | (0.014) | (0.010) | (0.012) | (0.012) | (0.020) | (0.011) | |
| RetNet | 34.12 | 29.46 | 35.36 | 55.57 | 31.31 | 51.70 | 44.49 | 2 3.46 | 62.40 | 28.00 | 41.54 |
| (N/A) | (1.046) | (0.007) | (0.009) | (0.005) | (0.014) | (0.010) | (0.012) | (0.011) | (0.020) | (0.011) | |
| DeltaNet | 33.25 | 26.82 | 35.75 | 54.07 | 31.40 | 49.96 | 44.11 | 22.18 | 63.60 | 29.60 | 41.33 |
| Model | Wiki. | LMB. | LMB. | PIQA | Hella. | Wino. | ARC-e | ARC-c | BoolQ | OBQA. | Avg. |
| ppl | ppl | acc | acc | acc_n | acc | acc | acc | acc | acc_n | 8 tasks | |
| RetNet (1.3B) | 22.45 | 21.84 | 38.70 | 69.04 | 47.73 | 52.72 | 63.68 | 33.36 | 60.61 | 36.60 | 50.31 |
| Mamba2 (1.3B) | 19.47 | 17.40 | 40.68 | 70.29 | 53.24 | 56.04 | 69.87 | 36.35 | 55.81 | 37.40 | 52.46 |
| DeltaNet (1.3B) | 1 9.77 | 16.64 | 41.78 | 70.95 | 51.09 | 54.70 | 67.63 | 34.47 | 61.19 | 38.40 | 52.53 |
| HADES (1B) | 20.41 | 1 7.22 | 4 1.18 | 71.33 | 5 1.85 | 56.35 | 6 8.48 | 3 4.81 | 6 0.73 | 38.60 | 52.92 |
| Model | Wiki. | LMB. | LMB. | PIQA | Hella. | Wino. | ARC-e | ARC-c | BoolQ | OBQA. | Avg. |
|---|---|---|---|---|---|---|---|---|---|---|---|
| ppl | ppl | acc | acc | acc_n | acc | acc | acc | acc | acc_n | 8 tasks | |
| MoM | 31.58 | 23.28 | 40.40 | 62.57 | 32.99 | 52.64 | 44.91 | 23.89 | 52.78 | 27.20 | 42.17 |
| HADES | 31.48 | 21.74 | 39.24 | 63.93 | 32.82 | 52.64 | 45.03 | 22.01 | 58.84 | 28.80 | 42.91 |
| Model | Wiki. | LMB. | LMB. | BoolQ | Hella. | Wino. | ARC-e | ARC-c | PIQA | OBQA. | Avg. |
|---|---|---|---|---|---|---|---|---|---|---|---|
| ppl | ppl | acc | acc | acc_n | acc | acc | acc | acc | acc_n | 8 tasks | |
| Mamba2 | 30.64 | 22.57 | 37.85 | 54.63 | 33.27 | 51.15 | 45.34 | 23.29 | 63.04 | 27.93 | 42.06 |
| (0.752) | (1.662) | (0.014) | (0.007) | (0.007) | (0.006) | (0.004) | (0.009) | (0.004) | (0.014) | (0.004) | |
| HADES | 33.41 | 25.57 | 37.10 | 59.67 | 31.70 | 51.85 | 44.84 | 22.67 | 63.15 | 27.87 | 42.37 |
| (1.721) | (3.482) | (0.022) | (0.009) | (0.010) | (0.011) | (0.008) | (0.006) | (0.007) | (0.013) | (0.005) |
| Methods | Wiki. | LMB. | LMB. | PIQA | Hella. | Wino. | ARC-e | ARC-c | BoolQ | OBQA. | Avg. |
| ppl | ppl | acc | acc | acc_n | acc | acc | acc | acc | acc_n | 8 tasks | |
| HADES (Ours) | 31.48 | 21.74 | 39.24 | 63.93 | 32.82 | 52.64 | 45.03 | 22.01 | 58.84 | 28.80 | 42.91 |
| w/o | 34.73 | 26.84 | 36.77 | 62.68 | 30.96 | 50.75 | 43.22 | 22.70 | 59.27 | 26.20 | 41.57 |
| w/o | 33.83 | 27.40 | 36.04 | 62.46 | 31.48 | 51.38 | 44.87 | 22.61 | 59.94 | 28.40 | 42.15 |
| Only Shared Filters | 34.55 | 27.64 | 35.75 | 62.40 | 31.39 | 52.88 | 44.23 | 24.23 | 60.83 | 26.00 | 42.21 |
| Only Expert Filters | 36.34 | 30.12 | 34.89 | 61.53 | 30.30 | 52.41 | 44.49 | 22.70 | 58.29 | 28.80 | 41.68 |
| Hyper | Wiki. | LMB. | LMB. | PIQA | Hella. | Wino. | ARC-e | ARC-c | BoolQ | OBQA. | Avg. | |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| param. | ppl | ppl | acc | acc | acc_n | acc | acc | acc | acc | acc_n | 8 tasks | |
| 8 | 39.52 | 37.58 | 32.93 | 60.88 | 29.63 | 50.28 | 42.00 | 22.27 | 61.10 | 24.60 | 40.46 | |
| 16 | 31.48 | 21.74 | 39.24 | 63.93 | 32.82 | 52.64 | 45.03 | 22.01 | 58.84 | 28.80 | 42.91 | |
| 24 | 33.11 | 26.73 | 35.40 | 62.57 | 31.99 | 50.75 | 44.36 | 23.29 | 48.93 | 28.60 | 40.74 | |
| 0.15 | 34.81 | 29.07 | 34.78 | 61.86 | 31.33 | 51.22 | 43.64 | 23.04 | 60.34 | 26.60 | 41.60 | |
| 0.25 | 31.48 | 21.74 | 39.24 | 63.93 | 32.82 | 52.64 | 45.03 | 22.01 | 58.84 | 28.80 | 42.91 | |
| Model | Linear Transformer | RetNet | DeltaNet | Mamba | Mamba2 | HADES |
|---|---|---|---|---|---|---|
| # Params. | 370M | 370M | 370M | 370M | 370M | 218M |
| Inference Time (sec) | 2.19 | 2.86 | 2.57 | 4.73 | 3.43 | 2.49 |
| Memory (GB) | 5.41 | 6.17 | 5.59 | 5.56 | 6.20 | 4.50 |
| Model | Setting | Selection Ratio | ||
|---|---|---|---|---|
| 50% | 25% | 75% | ||
| 370M | Prefill (1K) | 0.3997 | 0.3374 | 0.4969 |
| Prefill (2K) | 0.5936 | 0.4572 | 0.7869 | |
| Prefill (3K) | 0.6043 | 0.4765 | 0.8033 | |
| Decode | 0.0004 | 0.0003 | 0.0004 | |
| 1.3B | Prefill (1K) | 0.9981 | 0.5165 | 0.7742 |
| Metric | Value |
|---|---|
| Avg Routing Time (ms) | 0.0071 |
| Avg Total Time (ms) | 0.2900 |
| Avg Routing Time Ratio (%) | 2.4371% |
| Operation | Prefill |
|---|---|
| In Projection | |
| 1D Convolution | |
| SSM Kernel | |
| Out Projection | |
| RMS Norm | |
| HADES Ops. |
| Component | Parameters |
|---|---|
| in_proj linear | |
| conv1d | |
| out_proj | |
| rms norm | |
| ssm params | |
| Added params in HADES |
| Operation | Mamba2 | HADES | (Reduction) |
|---|---|---|---|
| In-projection | |||
| 1D Convolution | |||
| Out-projection | |||
| RMS Norm | |||
| SSD (SSM Core) |
| HADES Operation | FLOPs/token |
|---|---|
| Residual computation | |
| Selection score projection | |
| Top- selection | |
| Spectral bias | |
| Delta modulation |