Rational Transductors

arXiv:2602.07599 · cs.LG · Submitted 2026-02-07 · Read on arXiv

Listen

Radio episode about this paper

Transcript

Introduction to the show: ident: AI Radio. Generated commentary on the latest Artificial Intelligence papers.

Tom: Today's paper: "Rational Transductors".

Jane: As a fastidious researcher, I have meticulously analyzed both provided texts from "Rational Transconductors." The goal is not merely to summarize,

Tom: First, who's behind it and why it matters.

Paper summary: Tom: So we talked about the setup. The main thing in this Rational Transconductors paper is that standard Transformers struggle with rigid sequential logic and state tracking over long sequences.

Jane: They show that theoretically, self-attention is limited to AC0 or TC0 complexity classes, which means they can't handle problems that need intermediate chain of thought for robust generalization.

Lu: The authors propose Rational Transconductors as a dual-stream architecture that adds a matrix-valued recurrence based on Weighted Finite Automata to the Transformer.

Meng: It claims this injection of rational state information into the attention mechanism, via their Deep Rational Injection scheme, allows the model to generalize its expressive power beyond standard Transformers.

Tom: Specifically, they claim this allows them to capture all Regular Languages and NC1-complete problems like Boolean Formula Evaluation, plus fundamental separations such as Parity and Modular Counting.

Jane: And the crucial part is that they maintain an O(L + log T) parallel training efficiency while achieving this expanded expressive power.

Lu: They back this up by proving that Random Rational Features are a universal basis for sequential dependencies, justifying their initialization strategy in a rigorous way.

Lalam: And they also establish that the Differentiable Rational Feature regime is required to bridge the gap in representational compactness between the Transformer and what's needed for these tasks.

Meng: So, if this holds up, it suggests we could build models with better sequential reasoning capabilities without having to rely on slow recurrent structures.

Tom: Right, so it’s about using that algebraic structure from WFAs to give the Transformer a new way to handle sequence problems that are too rigid for its current setup.

Jane: This is important because it moves the conversation toward models that can perform precise algorithmic reasoning rather than just general pattern matching.

Conclusion: Tom: So to wrap up this discussion on Rational Transconductors, the authors are really showing that you can augment an existing Transformer with a specific algebraic structure from Weighted Automata to make it much better at sequence tasks.

Jane: It’s about moving away from just semantic modeling toward models capable of precise algorithmic reasoning and state tracking over very long contexts.

Lu: The implication is that this approach provides a framework for building AI systems that can handle complex logical constraints more reliably than the existing Transformer architecture alone allows.

Meng: For deployment, it suggests a path where we can get powerful sequential reasoning capabilities while still keeping training efficient with parallel techniques.

Tom: So essentially, they've found a way to inject rational state information directly into the attention mechanism to solve the regular gap problem they're seeing in Transformers on tasks like counting.

Jane: It’s about giving AI tools more mathematical rigor when it comes to sequential data and making sure those long context models don't just fail spectacularly when pushed past their training limits.

Lu: The work suggests that this dual-stream architecture could become a key direction for building foundation models with stronger logical reasoning capabilities.

Lalam: Lalam’s reading is that they are providing a way to get the mathematical foundation for these more complex reasoning capabilities in deep learning setups.

Mehryar Mohri

Google Research

cs.LG

Submitted: 2026-02-07

Updated: 2026-10-14

License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/

Importance score: 89/100

The gist: As a fastidious researcher, I have meticulously analyzed both provided texts from "Rational Transconductors." The goal is not merely to summarize, but to synthesize these distinct perspectives—the

Key concepts

Deep Rational Injection
This scheme injects rational state information directly into every Transformer block's hidden state. It uses a specific formula to combine the standard Transformer output with a matrix-valued recurrence derived from WFAs. This ensures that the model's internal representation is strictly tied to algebraic, rational functions, enabling it to reason about sequential logic precisely.
Weighted Finite Automata (WFA)
A WFA is a mathematical structure used to define sequential logic. It consists of a transition matrix and an initial state vector. The hidden state of the RT model represents the distribution of weights across these automaton states after processing an input sequence, effectively encoding the history in a compact, linear algebraic form.
Regular Gap
This refers to the performance failure where standard Transformers fail catastrophically on long sequences (e.g., counting). The RT framework solves this by grounding the model in formal language theory, allowing it to maintain 100% accuracy on tasks like Modulo-5 Counting, whereas standard models become unreliable as sequence length increases.

Terminology

Summary

As a fastidious researcher, I have meticulously analyzed both provided texts from Rational Transconductors. The goal is not merely to summarize, but to synthesize these distinct perspectives—the high-level architectural vision (Text A) and the rigorous mathematical grounding (Text B)—into a comprehensive, detailed description suitable for understanding the core contribution of this work.

Here is the combined, detailed summary:


Comprehensive Research Summary: Rational Transconductors

The paper introduces Rational Transconductors (RT), a novel dual-stream architecture designed to overcome the fundamental limitations of standard Transformers when handling rigid sequential logic and state tracking, particularly in ultra-long-context modeling. The core innovation lies in augmenting the Transformer with a mechanism derived from Weighted Finite Automata (WFA) and Rational Power Series, thereby bridging the gap between semantic flexibility and rigorous algebraic reasoning.

I. Architectural Innovation: Rational Transconductors

The RT architecture is fundamentally a dual-stream system that integrates rational state information directly into the attention mechanism. This is achieved via a Deep Rational Injection scheme, where the rational features are injected into the hidden state of every Transformer block ((l) t = z(l) t + W(l) projh t).

The central component is a matrix-valued recurrence derived from WFAs. The rational state vector h t at time step t is formally defined as the forward state of the automaton after processing the input prefix x 1:t:

h t = M x 1:t alpha

where M is a transition matrix, and alpha represents an initial state vector.

Key Architectural Features:

  • Linear Recurrence vs. Non-linear RNNs: Unlike traditional Recurrent Neural Networks (RNNs) which rely on non-linear activations (like), the RT update is strictly linear: h t = M x 1:t alpha. This linearity is crucial as it guarantees that the features capture Regular Languages exactly.

  • Parallel Scalability: The linear recurrence allows for highly efficient computation using parallel associative scans (prefix sums), effectively avoiding the sequential computational bottlenecks inherent in standard RNNs, thus preserving an optimal ** O(L + T) parallel time complexity**.

  • Universal Rational Basis: The framework is grounded in the theory of Rational Power Series. The authors prove that Random Rational Features act as a universal basis for capturing sequential dependencies, justifying the initialization strategy.

II. Theoretical Foundation: Bridging the Regular Gap

The theoretical rigor of this work stems from established results in formal language theory and linear algebra concerning rational functions:

  • Rationality and Automata: A series S is rational if and only if its Hankel matrix (HS) has finite rank. This rank is equal to the dimension (d) of the minimal linear representation of the series.

  • WFA Isomorphism: The definition of a WFA via a linear representation of dimension d is mathematically isomorphic to a Weighted Finite Automaton with exactly d states. The hidden state vector h t in R d represents the distribution of accumulated weights over these d states at time t.

  • Expressive Power: By grounding the architecture in this theory, the RT framework is shown to strictly generalize the expressive power of standard Transformers. The authors establish a strict inclusion hierarchy: (FRat FTF) FRT FRNN, where TRat (Rational Transconductor functions) is characterized as the composition of the Transformer class (FTF) and the class of vector-valued rational functions (TRat). This proves that RT can capture all Regular Languages and NC1-complete problems (such as Boolean Formula Evaluation), including fundamental separations like Parity and Modular Counting, which standard Transformers fail to recognize.

III. Operational Modes: Universal Heads

The framework supports diverse dynamical behaviors through different transition matrix structures, instantiated in parallel heads:

  • Orthogonal Heads (Cayley): These learn strictly norm-preserving rotations, making them ideal for continuous group operations and tasks involving infinite memory or counting/parity checks.

  • Stochastic Heads (Simplex): Parameterized as column-stochastic matrices, these heads function as differentiable finite state automata. They are perfectly suited for discrete logic, state switching, and hard resets (e.g., addition).

IV. Empirical Validation: Solving the Performance Gaps

Empirical results decisively demonstrate that Rational Transconductors successfully resolve the known weaknesses of standard Transformers:

  1. The Regular Gap: In tasks like Modulo-5 Counting, the Standard Transformer exhibits catastrophic failure, collapsing to near-random chance (20%) as sequence length (L) increases beyond training limits. In stark contrast, the RT model converges to 100% accuracy almost immediately.

  2. Length Generalization: The RT architecture maintains robust performance over extended sequences, generalizing nearly perfectly even when tested at lengths L up to 25 times the training length, mitigating the catastrophic positional drift seen in standard models.

  3. Algorithmic Reasoning: Experiments on tasks like Long-Integer Addition show that the Universal Rational Transconductor (leveraging a Stochastic head) autonomously learns the correct finite state automaton logic for addition and generalizes perfectly to sequences of 1000 digits, confirming its capability to learn complex sequential decision-making.

Conclusion

Rational Transconductors represent a significant advancement by structurally implementing the Krohn-Rhodes decomposition of finite automata within a deep learning framework. By combining the semantic flexibility of Transformers with the algebraic precision of rational series and WFAs, this architecture achieves mathematically grounded, ultra-long-context modeling with superior parallel efficiency (O(L + T)). It offers a pathway toward bicameral foundation models capable of precise, infinite-horizon sequential reasoning alongside fluent language generation.

Improvements for AI systems

  1. Rational Transductor Architecture Implementation: Implement a dual-stream architecture where a Rational Feature Head computes matrix-valued recurrence via Weighted Finite Automata (WFA) to handle sequential logic, while the Transformer handles semantic flexibility, ensuring the entire rational state history is pre-computed via a single parallel scan, reducing the total parallel depth to O(L + log T).

  2. Algorithmic Generalization and Robustness: The system can solve tasks like Modulo-5 Counting and Long-Integer Addition exactly for arbitrary lengths, unlike standard Transformers which suffer from catastrophic positional drift. This is achieved by leveraging the WFA's ability to implement cyclic group operations or stochastic switching logic autonomously.

  3. Training Stability and Efficiency: Utilize the Unified Scaled Cayley Parameterization with a scalar gain parameter, allowing learning of "decay (gt < 1) for fading memory tasks, which ensures gradient stability by bounding the spectral radius and guaranteeing that the error signal from time t reaches time 0 without attenuation."

  4. Parallel Throughput Scaling: The architecture achieves sub-linear complexity, as it is designed to bypass sequential bottlenecks, allowing inference on long sequences up to T=32k with a wall-clock latency of O(log T), significantly outperforming both standard RNNs (O(T)) and standard Transformers (O(T2)).

  5. Expressivity for NC1 Problems: The system can solve NC1-complete problems (such as Boolean Formula Evaluation) by structurally implementing the Krohn-Rhodes decomposition, which allows it to capture the full hierarchy of regular languages that standard Transformers are limited to AC0 or TC0 complexity classes.

  6. Universal Transconductor for Task Switching: Implement a Universal Rational Transconductor combining orthogonal (Cayley) and stochastic (Simplex) heads, enabling the model to autonomously route signals to the dynamical system best suited for the underlying algorithm, thereby achieving Logical Completeness.

Sources

Related papers