Cumulative Memory Recurrent Unit (CMRU)
- CMRU is a recurrent unit that implements a cumulative update mechanism, preserving hysteretic, discrete memory and enabling parallelizable, hardware-friendly computation.
- Its design replaces destructive state overwrites with additive quantized contributions, thereby creating skip-connections through time that improve gradient flow.
- CMRU bridges persistent multistability with modern RNN architectures, offering robust long-range dependency retention and efficient low-power analog implementation.
The Cumulative Memory Recurrent Unit (CMRU) is a parallelizable recurrent cell introduced to preserve the persistent, discrete memory and hardware-friendliness of the Bistable Memory Recurrent Unit (BMRU) while restoring good gradient flow and improving learning stability and performance to be competitive with modern parallelizable RNNs such as LRUs, minGRUs, and SSM-based models (Brandoit et al., 12 May 2026). Its defining change is a cumulative update formulation that replaces BMRU’s destructive overwrite step with an update that adds a quantized contribution on top of a scaled previous state, thereby creating skip-connections through time while retaining event-driven, hysteretic memory dynamics. The same work also introduces CMRU, a relaxed variant with input-dependent update scale that trades strict quantization for higher expressivity (Brandoit et al., 12 May 2026).
1. Historical and conceptual setting
CMRU emerged from the line of work on Memory Recurrent Units (MRUs), which were proposed to combine persistent memory via multistability with parallelizable computation over sequence length (Geeter et al., 14 Jan 2026). In that framework, the BMRU was introduced as a proof-of-concept recurrent cell whose hidden state is bistable, quantized, and compatible with the parallel scan algorithm. BMRU addressed two issues simultaneously: persistent memory through multistability with two quantized attractors per dimension, and analog hardware mapping through direct implementability with Schmitt triggers in CMOS, enabling sub-W power, including 100 nW keyword spotting (Brandoit et al., 12 May 2026).
The motivation for CMRU is tied to the limitations of existing sequence models. Modern parallelizable RNNs and Transformers are described through linear or linearizable state dynamics of the form
with eigenvalues of strictly inside the unit circle for stability; this implies monostability, a unique attracting fixed point, and fading memory, meaning that information stored in decays exponentially with elapsed time (Brandoit et al., 12 May 2026). Parallelizable gated RNNs such as minGRU avoid nonlinear recurrent dependence and remain scan-compatible, but because their input-only gate satisfies , they have no strict retain or overwrite mode and therefore no true persistent memory (Brandoit et al., 12 May 2026).
Against that backdrop, BMRU supplied exact retain behavior and hysteretic, discrete state transitions, but its performance lagged behind LRUs and minGRUs on complex sequential problems. The central diagnosis leading to CMRU is that gradient blocking during state updates is the main reason BMRU underperforms modern parallelizable RNNs on demanding tasks (Brandoit et al., 12 May 2026). A related but distinct tradition appears in HiPPO, where a “cumulative memory” unit can be understood as a memory core that stores an optimal polynomial projection of the full input history, especially in the HiPPO-LegS construction; this suggests that the term cumulative memory has both a quantized hysteretic realization in CMRU and a polynomial-projection interpretation in HiPPO-style recurrent memory (Gu et al., 2020).
2. From BMRU to the cumulative update rule
In the original BMRU, the hidden state is quantized so that each component is constrained to two values , with learned . Given input 0, BMRU defines
1
2
3
and the state update
4
Here, 5 is the componentwise Heaviside step function, 6 is the sign function, and 7 denotes the Hadamard product (Brandoit et al., 12 May 2026).
This yields two operating modes per dimension. If 8, then 9 and the cell is in retain mode: 0 If 1, then 2 and the cell is in update mode: 3 The result is a bistable hidden state with hysteretic thresholds and exact persistence whenever threshold crossings do not occur (Brandoit et al., 12 May 2026).
CMRU changes only the update part of this mechanism. The candidate, threshold, and gate remain unchanged: 4 The new state equation is
5
with scalar hyperparameter 6, restricted to 7 in most experiments (Brandoit et al., 12 May 2026).
Per dimension, retain mode is unchanged: 8 Update mode becomes
9
BMRU therefore performs full replacement during updates, whereas CMRU performs a cumulative update, similar to a residual connection through time (Brandoit et al., 12 May 2026).
3. Gradient flow, memory regimes, and state-space interpretation
The principal theoretical advantage of CMRU is the restoration of gradient flow during updates. For BMRU, treating 0 as constant in the local Jacobian gives
1
Hence, in retain mode,
2
but in update mode,
3
Any update event therefore cuts the gradient path from 4 to 5, which the paper links to poor convergence, high sensitivity to initialization, and brittle learning on long sequences (Brandoit et al., 12 May 2026).
For CMRU, the recurrence can be written as
6
so the Jacobian becomes
7
In update mode,
8
Thus, whenever 9, updates no longer block gradients, and when 0 the gradient magnitude is exactly 1 in both retain and update modes, giving skip-connection-like behavior through time (Brandoit et al., 12 May 2026).
The paper interprets 2 as selecting among three regimes. When 3, CMRU reduces to the original BMRU, with bistable states 4 and gradient blocking preserved. When 5, the model implements event-driven fading memory: memory decays with the number of update events rather than with elapsed time, while retain intervals remain perfectly persistent. When 6, the cell becomes a quantized cumulative integrator: 7 in update mode, with identity Jacobian in both modes (Brandoit et al., 12 May 2026).
This modification also changes the reachable state structure. In BMRU, each dimension can only take 8, so the overall state space is limited to at most 9 points. In CMRU with 0,
1
so reachable states form a lattice of integer multiples of 2, yielding countably infinite discrete attractors (Brandoit et al., 12 May 2026). The paper presents this as a substantial increase in representational capacity while preserving discrete spacing desirable for analog circuits.
CMRU also admits a selective state-space representation
3
with
4
In retain mode, 5 and all eigenvalues are 6, directly providing persistent memory. In update mode, 7, so all eigenvalues equal 8; for 9 they lie inside the unit circle, and for 0 they remain 1 (Brandoit et al., 12 May 2026). This places CMRU at an explicit interface between persistent-memory dynamics and the state-space vocabulary used for linear RNNs and SSMs.
4. 2CMRU and representational trade-offs
The 3CMRU variant relaxes the fixed quantization scale of CMRU by introducing an input-dependent scale
4
and update
5
The sign pattern remains discrete through 6, but the magnitude can vary continuously with the input (Brandoit et al., 12 May 2026).
This changes the geometry of the memory state. In CMRU, hidden states remain tied to fixed multiples of a global 7; in 8CMRU, hidden states are no longer restricted to fixed multiples of a global 9 and can densely cover 0 (Brandoit et al., 12 May 2026). The trade-off is explicit: strict quantization and explicit bistability at fixed levels are lost, but the model gains smooth, high-precision continuous memory while preserving event-driven accumulation and exact persistence in retain mode.
The paper positions the two variants differently. CMRU retains quantized states, persistent memory, and noise-resilient dynamics essential for analog implementation. 1CMRU is described as better suited for large-scale digital ML where quantization is not a hardware constraint (Brandoit et al., 12 May 2026). A common misconception is therefore to treat the two cells as interchangeable: the reported results indicate that they share the cumulative-update principle, but only CMRU preserves the strict discrete-state structure and explicit analog-hardware mapping of the BMRU lineage.
A second misconception is that the gains of CMRU arise simply from larger state expressivity. On sequential MNIST, the paper concludes instead that gradient flow from cumulative updates, not increased representational capacity, is the crucial improvement (Brandoit et al., 12 May 2026). This point matters especially when comparing CMRU to 2CMRU, because the fixed-scale CMRU still improves dramatically relative to BMRU once cumulative updates are enabled.
5. Training setup, architecture, and computational characteristics
The experimental architecture uses a common backbone into which all recurrent cells are plugged. The model dimension is 3, the recurrent cell has state dimension 4 varied across experiments from 5 to 6, and the number of recurrent blocks is 7 depending on the task (Brandoit et al., 12 May 2026). Each block contains a recurrent sublayer producing
8
followed by a pointwise MLP with GLU,
9
with hidden size 0, and skip connections with learned gain
1
Pooling depends on the task. Last pooling is used for tasks explicitly testing long-range memory, namely sMNIST, copy-first, and parity, whereas mean pooling is used for LRA tasks and audio tasks (Brandoit et al., 12 May 2026). Standard sinusoidal positional encodings are concatenated to inputs and linearly projected.
Training uses AdamW with 2, 3, 4, and weight decay 5. The learning rate is 6 with 7 linear warmup and cosine decay to 8. Gradient norm is clipped to 9, batch size is 0, no task-specific hyperparameter tuning is performed, and each configuration is run with 1 seeds with mean 2 min–max reported (Brandoit et al., 12 May 2026).
Because the Heaviside and sign operators are non-differentiable, training uses surrogate gradients: 3 so the forward dynamics remain exactly discrete while the backward pass is smoothed (Brandoit et al., 12 May 2026). Computationally, CMRU, 4CMRU, BMRU, LRU, and minGRU are all affine in 5, which means they support associative scan over sequence length and can therefore be parallelized on GPUs and TPUs (Brandoit et al., 12 May 2026). This inherits the scan-compatibility emphasized in the earlier MRU formulation, where recurrences of the form 6 admit a Blelloch-style scan implementation (Geeter et al., 14 Jan 2026).
For models intended to be deployed with 7 on existing Schmitt-trigger BMRU hardware, the paper proposes 8-annealing: start at 9, keep it for 00 of steps, linearly decay to 01 over the next 02, and hold at 03 for the final 04; only checkpoints after 05 are saved and evaluated (Brandoit et al., 12 May 2026). This is presented as a way to preserve the deployment dynamics of BMRU while benefiting from the gradient flow of CMRU during training.
6. Empirical performance and benchmark profile
On sequential MNIST with last pooling, 06, and one layer, the effect of 07 is direct. For CMRU, 08, 09 yields 10 accuracy, high variance, whereas 11 yields 12 accuracy, low variance. For 13CMRU, 14, 15 yields 16 accuracy, and 17 yields 18 accuracy (Brandoit et al., 12 May 2026). The paper reports that increasing 19 monotonically improves both mean accuracy and cross-seed stability, with the largest gains on the harder sMNIST relative to pMNIST.
At small state dimension, the reported scaling behavior is notable. With 20 and a single layer on sMNIST, at 21, CMRU reaches 22 and 23CMRU 24, while LRU and minGRU are described as significantly worse and more variable. At 25, all methods are 26–27, but the CMRU variants still show lower variance (Brandoit et al., 12 May 2026). This indicates that the cumulative memory mechanism is especially advantageous in the small-model regime emphasized by ultra-low-power applications.
The strongest specialization appears on copy-first tasks. In the discrete 15-class copy-first, noise-free setting with 28 and last pooling, CMRU (29) achieves 30 accuracy up to sequences of length 31, while LRU/minGRU collapse to near random for large 32 (Brandoit et al., 12 May 2026). In the continuous, noise-free version, CMRU with 33 achieves MAE near the theoretical quantization bound 34, and 35CMRU achieves essentially zero MAE. In the continuous with heavy noise variant, with 36 and later inputs i.i.d. from the same distribution, 37CMRU with 38 remains robust across all 39 and noise with MAE 40, whereas LRU and minGRU degrade rapidly with 41 and noise and at 42 essentially fail, with MAE near the 43 baseline (Brandoit et al., 12 May 2026).
On larger-scale noisy copy-first with 44, 45, 46, the paper reports that CMRU and 47CMRU solve the task robustly for all seeds, while LRU and minGRU still fail entirely (Brandoit et al., 12 May 2026). The interpretation given is architectural rather than merely parametric: the advantage comes from persistent-memory structure rather than only from small-model efficiency.
On the Long Range Arena-style tasks, the performance profile is mixed. On IMDb sentiment, LRU has the best mean at 48, minGRU is slightly below, and CMRU/49CMRU are 50–51, within 52. On sCIFAR10, LRU and minGRU are best at 53–54, especially at larger depth, while CMRU and 55CMRU are slightly behind but close, with the gap shrinking with depth (Brandoit et al., 12 May 2026). By contrast, on ListOps, at 56, CMRU and 57CMRU reach 58 accuracy, LRU is 59 and largely fails, and minGRU is intermediate at 60–61. On Pathfinder, CMRU reaches 62 and 63CMRU 64, while LRU and minGRU are 65, which is chance at this small capacity of 66 (Brandoit et al., 12 May 2026). The paper interprets these results as evidence that persistent-memory cells have a clear advantage on discrete and long-range compositional tasks.
For character-level language modeling on Shakespeare, the reported cross-entropy losses are: 67CMRU: 1.443, CMRU: 1.455, LRU: 1.504, minGRU: 1.453, Hybrid minGRU+CMRU: 1.442, and Hybrid minGRU+68CMRU: 1.441 (Brandoit et al., 12 May 2026). This supports the claim that CMRU and 69CMRU are competitive on smooth statistical tasks and that persistent + fading memory hybrids are complementary.
In the audio keyword spotting regime with very small state dimensions, the differences are small. On KWS Digits (11 classes), all models are 70–71 accuracy even at 72; CMRU at 73 reaches 74, while LRU at 75 reaches 76. On KWS All (35 classes), at 77, CMRU is 78 and LRU/minGRU are 79–80; at 81, CMRU is 82, LRU 83, and minGRU 84 (Brandoit et al., 12 May 2026). The important point emphasized is that a single-layer CMRU with 85 still fits existing hardware assumptions while maintaining competitive accuracy.
7. Analog implementation, model landscape, and related cumulative-memory frameworks
CMRU is explicitly designed for always-on sensors, implants, and related settings requiring sub-86W operation for years (Brandoit et al., 12 May 2026). In BMRU, quantized bistability and hysteresis map directly to Schmitt trigger circuits, where states are stored as discrete logic levels and do not accumulate analog errors (Brandoit et al., 12 May 2026, Geeter et al., 14 Jan 2026). CMRU with 87 extends this hardware picture to a hysteretic charge accumulator: cumulative updates correspond to adding discrete charges or currents to an integrator, and the paper proposes a native analog circuit based on a cascade of BMRU cells interleaved with Differential Pair Integrators, with staircase states at multiples of 88 (Brandoit et al., 12 May 2026). The reported Cadence simulations show event-driven increments or decrements of output by 89 every 90 ms in update mode, perfect retain in retain mode, and instantaneous power in the deep nW regime, including 91 nW in the example.
Within the broader model landscape, the paper contrasts CMRU and 92CMRU with LRUs, SSMs, and minGRU. LRUs and SSMs are strong on tasks such as sCIFAR and IMDb, but are limited to fading memory and typically rely on digital implementations or capacitor-based analog realizations (Brandoit et al., 12 May 2026). minGRU is fully parallelizable and strong on sCIFAR, IMDb, and language modeling, but its gates remain in 93 and therefore do not provide true retain or overwrite modes (Brandoit et al., 12 May 2026). CMRU and 94CMRU, by contrast, supply explicit persistent memory via multistability and exact retain mode, and at 95 they provide perfect gradient flow through time while remaining affine in the previous state (Brandoit et al., 12 May 2026).
A useful clarification concerns the relation between CMRU and HiPPO. HiPPO-LegS defines a memory state as the coefficient vector of the optimal projection of the entire history onto a polynomial basis under the measure
96
leading to the ODE
97
and discrete recurrence
98
for full-history cumulative memory (Gu et al., 2020). This is a different construction from the quantized, hysteretic CMRU introduced in 2026, but the juxtaposition is informative: HiPPO provides a cumulative-memory mechanism through optimal polynomial projection, whereas CMRU provides cumulative memory through event-driven discrete accumulation with hysteresis. This suggests two distinct technical meanings of cumulative recurrent memory: one based on analytically derived compression of all history, the other on quantized multistable state transitions.
The main contributions claimed for CMRU are therefore fourfold: diagnosis of gradient blocking in BMRU as the primary training bottleneck; introduction of cumulative updates that restore gradient flow while preserving persistent memory; a selective SSM representation in which 99 directly controls eigenvalues; and an explicit path toward ultra-low-power analog RNNs that combine quantized, noise-resilient persistent memory with gradient-friendly training (Brandoit et al., 12 May 2026). A plausible implication is that CMRU occupies a specific niche rather than replacing fading-memory models wholesale: it is strongest where discrete, long-range retention and hardware-constrained efficiency are both first-order design criteria.