REVIEW 3 major objections 5 minor 23 references
Online learning of neural state-space models
T0 review · 3 major / 5 minor · reviewed 2026-08-01 · deepseek-v4-flash
Pith's one-line read This paper establishes an almost-sure convergence guarantee for a recursive Gauss–Newton online learning rule applied to encoder-based neural state-space models, enabling computationally cheap adaptation during operation.
desk verdict A credible recursive prediction-error scheme for encoder-based ANN-SS models with a convergence theorem imported from Ljung, but the theory's key condition is not verified for the simulation, and the empirical evidence is thinner than claimed. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The recursion (4): θ_i = θ_{i−1} + μ_i R_i^{−1} ψ_i^T(θ_{i−1}) ϵ_i(θ_{i−1}), with R_i updated as a weighted running covariance. The encoder network Ψ_θ reconstructs the initial state of each batch from past IO data, making the predictor depend only on a finite memory window, which is what allows the classical convergence analysis to apply. The ODE-based argument identifies V̄ as a Lyapunov function for the limiting trajectory, giving the almost-sure result.
What would settle it
Run recursion (4) on a system with slow or no exponential forgetting (e.g., an oscillator with zero damping) and observe whether θ_i diverges or cycles; the theorem predicts convergence only when Condition 1 holds, so a clear failure on such a plant would define the boundary of the claim. Additionally, test the fully unknown-output-map setting (learning h_θ from data) and compare convergence with the vehicle simulation.
Extended reading notes
Core claim
The central claim is Theorem 2: the recursion (4) — a Gauss–Newton update with step size, inverse Hessian approximation R_i, gradient ψ_i of the predictor, and prediction error ϵ_i — converges with probability 1 to either a stationary point of the limiting cost V̄(θ) or to the boundary of the parameter set Θ, provided the data-generating system satisfies exponential forgetting of initial states and the predictor is Lipschitz. This gives encoder-based ANN-SS models the same kind of almost-sure convergence guarantee long available for classical recursive prediction-error methods, and it applies to a whole family of subspace-encoder model structures.
Load-bearing premise
The convergence proof assumes Condition 1 — that the real system exponentially forgets its initial state — which the paper neither verifies for the simulated vehicle nor discusses for nonlinear systems where it fails; the simulation also supplies the model with the true output structure, an unstated favorable prior.
Editorial extensions
If this is right
- Online adaptation becomes feasible for encoder-based ANN-SS models at roughly 1 ms per batch update.
- The convergence guarantee extends to the whole SUBNET family—LPV, Koopman, and model-augmentation structures—not just plain ANN-SS models.
- The recursive scheme is robust to batch length, whereas the batch-wise method is sensitive to horizon and memory size.
- With the recommended forgetting-factor choices, the scheme can track slowly time-varying dynamics.
- The batch-wise pipeline provides a practical middle ground using a replay buffer when lower-latency recursive updates are not required.
Reading between the lines
- The simulation's choice to set the output map to the true linear structure ([1 0]x) is an unstated favorable prior; the claimed accuracy may not carry over to the fully unknown-output-map setting the paper frames as its target.
- Condition 1 (exponential forgetting) is plausible for the stable Pacejka vehicle model but could fail for near-marginally-stable systems; a natural test is to estimate the forgetting exponent from data and check whether it satisfies the bound.
- The proof leaves open the practical role of the regularization constant δ and the boundary-convergence case; users likely need to monitor whether iterations are in the interior of Θ.
- Because the recursion only needs the current batch and a running R_i, the method should scale to embedded or resource-constrained controllers where storing a replay buffer is infeasible.
Signed reviews
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes two methods for online identification of encoder-based neural state-space (ANN-SS) models. The first is a batch-wise pipeline that retrains on a sliding replay buffer using a JAX-based Adam+L-BFGS optimizer and an encoder for state initialization. The second is a recursive Gauss-Newton scheme (Eq. (4)) that updates the parameters each time a batch arrives, using the same encoder-based predictor. The main theoretical claim (Theorems 1 and 2) is that, under a global exponential-forgetting condition on the data-generating system and a Lipschitz/uniform-boundedness condition on the predictor, the recursive parameter estimates converge almost surely to stationary points of a limiting mean-square prediction-error criterion, or to the boundary of the parameter set. The methods are demonstrated on a nonlinear vehicle lateral-dynamics benchmark, with reported convergence-speed and wall-clock comparisons.
Significance. If the convergence result is valid under its hypotheses, the recursive scheme provides a computationally cheap online counterpart to state-of-the-art offline encoder-based ANN-SS identification (SUBNET), with a classical recursive-prediction-error type guarantee. This is a useful extension of an active research area. The batch-wise pipeline is also a practical engineering contribution, and the authors provide a public implementation. However, the significance is limited by a substantial gap between the theoretical assumptions and the simulation study: Condition 1 is not verified, and in fact appears to be violated by the benchmark vehicle model. The theoretical framework itself is standard and, conditional on the assumptions, likely correct, but the paper as written does not establish that the theory covers its own validation example.
major comments (3)
- [IV-C, Eq. (10); V, Eqs. (18)-(20)] Condition 1 (Eq. (10)) requires the data-generating system to exponentially forget initial conditions uniformly over all initial states in any delta-ball. The simulation model with saturating Pacejka tire forces does not satisfy this. For two initial states deep in the saturated regime with the same input and noise, the tire forces become identical constants, so Delta(dot omega)=0 and Delta(dot v_y) = -v_x Delta(omega). Hence Delta(omega) remains constant and Delta(v_y) grows linearly; fourth moments of the output difference do not decay, contradicting (10). Thus Theorem 2's almost-sure convergence guarantee does not apply to the benchmark used for validation. The paper should either replace the example with a system satisfying Condition 1, or explicitly state and justify that the simulation is an empirical demonstration outside the theorem's coverage.
- [V, output-map definition] The state is defined as x_k = [v_{y,k}, omega_k]^T and the measured output as y_k = omega_k + e_k, so the true output map is h(x) = [0 1]x. The text then says h_theta is chosen to match the true output structure, i.e., \hat y_k = [1 0]\hat x_k, which extracts v_y, not omega. As written, the model output is the wrong state component, making the simulation an invalid test of identifying the described system. Correct the typo or clarify the state ordering/output map.
- [IV-C, proof of Theorem 1] The proof of Theorem 1 consists of stating that the batch loss satisfies Condition C1 of [21] and that `the proof of [21, Lemma 3.1] applies`. This is too terse for a central result. Conditions 1-2 do not transparently imply that the loss sequence is quasi-stationary with geometrically summable covariances, nor that the gradient sequence psi_i has the bounded moments needed for uniform convergence of g(theta) and \bar R(theta). Please provide a self-contained verification of the cited lemma's hypotheses, or state the intermediate mixing/moment assumptions explicitly.
minor comments (5)
- [III-B, paragraph after parameter estimation] The sentence claiming that `under mild assumptions, the convergence and consistency properties of the SUBNET approach are inherited as m -> infinity` is unsupported. Please either give precise assumptions and a theorem, or soften this to a heuristic remark.
- [IV-D, Eq. (17)] The implemented update uses (R_i + delta I)^{-1} instead of the R_i^{-1} analyzed in Theorem 2. The effect of the constant regularization delta on the convergence guarantee should be discussed; as it stands, the practical algorithm is not exactly the recursion covered by the theorem.
- [IV-C, Theorem 2(b)] The condition R_i > 0 is assumed, but with the recommended mu_0 = 1, Eq. (4b) gives R_1 = psi_1^T psi_1, which can be singular. State explicitly that mu_i in (0,1) and R_0 > 0, or note that regularization is needed to satisfy (b).
- [IV-C, Condition 2] Condition 2 requires the derivative d/dtheta \hat gamma to satisfy a Lipschitz/uniform-bound condition. This is immediate for smooth tanh networks, but not for ReLU activations. The paper should state that the analysis assumes smooth activations, given that the differentiability assumption is only called `technical`.
- [V, Fig. 2] Minor presentation issues: the y-axis label `T est RMSE` has an unintended space; also, the AUCC values in Table III for divergent runs are reported as infinity, which is fine, but the table caption should define AUCC more explicitly.
Circularity Check
No circular derivation: the convergence theorem is an application of external stochastic-approximation theory and the simulation is evaluated on separate test data.
full rationale
The paper's derivation chain is not circular. The recursive update (4) is a Gauss-Newton-type parameter recursion on the batch prediction error. Theorem 1 is established by invoking Ljung's classical convergence analysis ([21, Lemma 3.1]), and Theorem 2 is proven by applying Ljung's recursive prediction-error identification theorem ([19, Theorem 2]) through an ODE/Lyapunov argument. These are external, established results, not self-citations, and they do not assume the paper's conclusion. The limiting criterion \bar V(\theta) is not a fitted quantity; it is defined as the almost-sure limit of the running average loss, and convergence to its stationary points is the theorem's claim, not an input. Condition 1 (Eq. 10) is an exponential-forgetting assumption on the data-generating system; even if the simulation model may violate it, that is a validity/coverage gap between theory and experiment, not a circular step. The simulation's choice of h_theta matching the true output structure ([1 0] x) is stated explicitly and evaluated on a separate test set; it is a favorable modeling prior, not a fitted parameter renamed as a prediction. Hyperparameters such as the step size and regularization are hand-selected and not fitted to the test data. The overlapping-author citations ([4], [17], [18]) are used for the encoder architecture, optimizer pipeline, and model-family examples, but they do not carry the convergence argument. No equation in the paper reduces, by construction or by self-citation, to its own inputs, so no circular step is identified.
Assumptions & free parameters
free parameters (7)
- mu_0 (initial step size) =
1
- lambda_0 (initial forgetting factor) =
0.75 (recommended range 0.75-0.95)
- bar_lambda (forgetting factor target) =
0.99
- delta (regularization for R_i) =
10^-3
- R_0 (initial information matrix) =
10^4 I_n_theta
- n (encoder lag) =
not specified
- Batch length N and replay memory m =
N=25, m=20 in the main simulation; varied in Table III
assumptions (7)
- domain assumption The data-generating system is a discrete-time nonlinear state-space model (1) with i.i.d. finite-variance output noise, and the input is quasi-stationary and independent of the noise.
- domain assumption Condition 1: exponential forgetting of initial conditions, E_e{||y_k - tilde_y_k||^4} < C(delta) lambda^{k-k0} (Eq. 10).
- standard math Condition 2: the predictor and its gradient are Lipschitz in the past IO window (Eq. 11), automatically satisfied by Lipschitz ANN activations on a compact parameter set.
- domain assumption The encoder reconstructability map exists, so an initial state can be estimated from n past IO samples (Eq. 2b), following [4].
- standard math Ljung's convergence theorems for recursive prediction-error identification (Refs. [19] and [21]) apply to recursion (4).
- ad hoc to paper Theta lies in a compact set Theta, projection keeps iterates inside, and the predictor is differentiable everywhere on Theta.
- ad hoc to paper For the batch-wise pipeline, convergence and consistency properties of SUBNET are inherited as m tends to infinity under unspecified mild assumptions.
Cite this review
Pith. "Pith review of Online learning of neural state-space models." pith.science (2026). https://pith.science/paper/ZAJMISSM
@misc{pith2026260717614,
author = {Pith},
title = {Pith review of: Online learning of neural state-space models},
year = {2026},
howpublished = {\url{https://pith.science/paper/ZAJMISSM}},
note = {Machine review of arXiv:2607.17614}
}
read the original abstract
Recent advances in deep-learning-based nonlinear system identification have led to encoder-based estimation of neural state-space (ANN-SS) models that achieve state-of-the-art performance in offline settings by estimating initial model states from past input-output data. These methods are typically used in multiple-shooting-based offline identification, and online learning of these models remains largely unexplored. This paper presents a batch-wise learning pipeline and a direct recursive identification algorithm for subspace encoder-based ANN-SS models. We provide convergence analysis of the recursive formulation and validate its performance through extensive simulation studies. The results demonstrate that the proposed approach enables computationally efficient online adaptation with high model accuracy.
Figures
Reference graph
Works this paper leans on
-
[21]
Convergence analysis of parametric identification methods,
——, “Convergence analysis of parametric identification methods,” IEEE Transactions on Automatic Control, vol. 23, no. 5, pp. 770–783, 1978
1978
-
[1]
Deep Learning and System Identification,
L. Ljung, C. Andersson, K. Tiels, and T. B. Sch ¨on, “Deep Learning and System Identification,” inProc. of the 21st IFAC World Congress, 2020, pp. 1175–1181
2020
-
[2]
An L-BFGS-B Approach for Linear and Nonlinear System Identification Underℓ 1 and Group-Lasso Regularization,
A. Bemporad, “An L-BFGS-B Approach for Linear and Nonlinear System Identification Underℓ 1 and Group-Lasso Regularization,” IEEE Transactions on Automatic Control, vol. 70, no. 7, pp. 4857– 4864, 2025
2025
-
[3]
Continuous-time system identification with neural networks: Model structures and fitting criteria,
M. Forgione and D. Piga, “Continuous-time system identification with neural networks: Model structures and fitting criteria,”European Journal of Control, vol. 59, pp. 69–81, 2021
2021
-
[4]
Deep subspace encoders for nonlinear system identification,
G. I. Beintema, M. Schoukens, and R. T ´oth, “Deep subspace encoders for nonlinear system identification,”Automatica, vol. 156, p. 111210, 2023
2023
-
[5]
Fast Recursive Identification of State Space Models via Exploitation of Displacement Structure,
Y . M. Cho, G. Xu, and T. Kailath, “Fast Recursive Identification of State Space Models via Exploitation of Displacement Structure,” Automatica, vol. 30, no. 1, pp. 45–59, 1994
1994
-
[6]
Recursive identification of a nonlinear state space model,
T. Wigren, “Recursive identification of a nonlinear state space model,” International Journal of Adaptive Control and Signal Processing, vol. 37, no. 2, pp. 447–473, 2023
2023
-
[7]
Online learning of nonlinear parametric models under non-smooth regularization using EKF and ADMM,
L. Frascati and A. Bemporad, “Online learning of nonlinear parametric models under non-smooth regularization using EKF and ADMM,” Automatica, vol. 185, p. 112787, 2026
2026
Show all 23 references
-
[8]
On the adaptation of recurrent neural networks for system identification,
M. Forgione, A. Muni, D. Piga, and M. Gallieri, “On the adaptation of recurrent neural networks for system identification,”Automatica, vol. 155, p. 111092, 2023
2023
-
[9]
Meta-Learning of Neural State-Space Models Using Data From Similar Systems,
A. Chakrabarty, G. Wichern, and C. R. Laughman, “Meta-Learning of Neural State-Space Models Using Data From Similar Systems,” in Proc. of the 22nd IFAC World Congress, 2023, pp. 1490–1495
2023
-
[10]
On the adaptation of in-context learners for system identification,
D. Piga, F. Pura, and M. Forgione, “On the adaptation of in-context learners for system identification,” inProc. of the 20th IFAC Sympo- sium on System Identification, 2024, pp. 277–282
2024
-
[11]
Ljung and T
L. Ljung and T. S ¨oderstr¨om,Theory and Practice of Recursive Identifi- cation, ser. Signal Processing, Optimization, and Control. Cambridge, MA, USA: MIT Press, 1987
1987
-
[12]
Deep-Learning-Based Identification of LPV Models for Nonlinear Systems,
C. Verhoek, G. I. Beintema, S. Haesaert, M. Schoukens, and R. T ´oth, “Deep-Learning-Based Identification of LPV Models for Nonlinear Systems,” inProc. of the 61st IEEE Conference on Decision and Control, 2022, pp. 3274–3280
2022
-
[13]
Deep Identification of Nonlinear Systems in Koopman Form,
L. C. Iacob, G. I. Beintema, M. Schoukens, and R. T ´oth, “Deep Identification of Nonlinear Systems in Koopman Form,” inProc. of the 60th IEEE Conference on Decision and Control, 2021, pp. 2288–2293
2021
-
[14]
Learning- based augmentation of first-principle models: A linear fractional representation-based approach,
J. H. Hoekstra, B. M. Gy ¨or¨ok, R. T ´oth, and M. Schoukens, “Learning- based augmentation of first-principle models: A linear fractional representation-based approach,”arXiv preprint arXiv:2602.17297, 2026
2026
-
[15]
Adam: A Method for Stochastic Opti- mization,
D. P. Kingma and J. Ba, “Adam: A Method for Stochastic Opti- mization,” inProc. of the 3rd International Conference on Learning Representations, 2015, pp. 1–15
2015
-
[16]
On the limited memory BFGS method for large scale optimization,
D. C. Liu and J. Nocedal, “On the limited memory BFGS method for large scale optimization,”Mathematical Programming, vol. 45, pp. 503–528, 1989
1989
-
[17]
Efficient identification of linear, parameter-varying, and nonlinear systems with noise models,
A. Bemporad and R. T ´oth, “Efficient identification of linear, parameter-varying, and nonlinear systems with noise models,” arXiv:2504.11982, 2025
2025 arXiv
-
[18]
Data-driven augmentation of first-principles models under constraint-free well-posedness and stability guarantees,
B. Gy ¨or¨ok, R. Drenth, C. Verhoek, T. P ´eni, M. Schoukens, and R. T ´oth, “Data-driven augmentation of first-principles models under constraint-free well-posedness and stability guarantees,”arXiv preprint arXiv:2604.11421, 2026
2026 arXiv
-
[19]
Analysis of a general recursive prediction error identifica- tion algorithm,
L. Ljung, “Analysis of a general recursive prediction error identifica- tion algorithm,”Automatica, vol. 17, no. 1, pp. 89–99, 1981
1981
-
[20]
Pearson Education, 1998
——,System Identification: Theory for the User, 2nd ed. Pearson Education, 1998
1998
-
[22]
C. E. Rasmussen and C. K. I. Williams,Gaussian Processes for Machine Learning. MIT Press, 2005
2005
-
[23]
H. B. Pacejka,Tyre and Vehicle Dynamics, 3rd ed. Butterworth- Heinemann, 2012
2012
Reviewed August 1, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.