ZNO: Stable Rational Neural Operators in the Z-Domain for Discrete-Time Dynamics

TL;DR

ZNO introduces a stable, rational neural operator in the z-domain for discrete-time dynamics, excelling in identifying stable rational systems with near-unit-circle poles and long-memory effects.

Contribution

The paper presents ZNO, a novel neural operator leveraging z-domain rational filters with stability constraints, tailored for discrete-time system identification.

Findings

ZNO outperforms in stable rational systems with lightly damped poles near the unit circle.

ZNO achieves the lowest mean error across memory regimes from short to long.

On benchmarks aligned with stable rational filters, ZNO is competitive with existing neural operator and state-space models.

Abstract

We introduce the Z-Domain Neural Operator (ZNO), a causal neural operator whose layers are stable low-rank multiple-input multiple-output (MIMO) rational filters parameterized directly in the $z$-plane. ZNO addresses a limitation of existing operator learning methods, many of which are primarily tailored for continuous-time problems, while a large class of system-identification problems is intrinsically discrete-time. The $z$-domain form expresses stability as a unit-disk pole constraint and makes learned discrete-time poles directly readable. The model combines low-rank channel mixing, smooth stable pole reparameterization, causal recurrence, and an optional short finite impulse response (FIR) branch in a single $z$-domain rational recurrent layer. Across controlled discrete system-identification experiments, ZNO's advantage is most evident when the target dynamics are stable rational systems with lightly damped poles near the unit circle. Under matched parameter budgets, ZNO is not uniformly dominant; however, with validation-selected configurations, the same architecture can achieve the lowest mean error across the controlled tasks. A five-bin difficulty sweep over near-unit-circle / long-memory dynamics shows that ZNO has the lowest mean error across memory regimes, from short (approximately 10 steps) to long (approximately 100-200 steps). On five public nonlinear system-identification benchmarks, ZNO is competitive with neural operator and state-space baselines, achieving the lowest mean error on benchmarks whose dynamics align with stable rational discrete-time filters, while classical or state-space baselines remain preferable on some systems. These results position ZNO as a strong model for stable rational discrete-time dynamics, especially in near-unit-circle and long-memory regimes, but not as a universal replacement for specialized system-identification methods.