精读笔记
Problem Setting
论文标题：Two-Layer Linear Auto-Regressive Models Estimate Latent States（arXiv preprint / 2026）。
这篇论文处理的是 partially observed linear dynamical system 中的一个 representation learning 问题：只给单条 input-output trajectory，用未来输出预测损失训练两层线性 AR 模型，hidden layer 是否会恢复 Kalman filter 的状态估计。
真正困难点不在预测本身。浅层 AR regression 预测未来输出是系统辨识里很成熟的事；困难在于从“预测 operator 学得好”推出“中间表示就是 latent state estimate”。这要求解决三个辨识层面的矛盾： 1. 状态不可唯一辨识，只能 up to similarity transform； 2. 训练信号只在 output space，没有 state supervision； 3. 两层线性模型是非凸 factorization，学到的 factor 不必先验等于 O,C。
以前的 subspace identification 通常绕开这个问题：先估 Hankel / Markov 参数，再显式做 Ho-Kalman/SVD 分解。那条路线的问题是 state extraction 是 post-processing，不是端到端学习中 hidden activation 的自然结果。本文要证明的是：在合适条件下，这个 post-processing 可以被两层线性网络的 hidden layer 隐式完成。
Motivation
作者的动机不是提高 LDS prediction accuracy，而是给“序列模型通过 next-token/future prediction 学到 latent state”这个深度学习叙事一个可证明的线性版本。
已有路线缺的是 representation-level guarantee。系统辨识文献能保证估计 Hankel operator 或 state-space realization，但通常需要显式 factorization；深度学习经验上看 hidden representation 可能编码世界状态，但缺少可辨识、有限样本、优化层面的闭环解释。本文的核心观察是：Kalman filter 的有限窗口展开天然产生一个低秩 operator OC，而两层线性网络的 G2G1 正是低秩 factorization。因此，如果 ERM 学到 OC，那么 G1 z 的语义很可能不是任意 embedding，而是 Kalman state estimate 的坐标变换。
关键缺口在于把三件事接起来：Kalman filtering ≈ finite AR；AR prediction operator ≈ low-rank Hankel；low-rank factor 的 first layer activation ≈ state estimate。论文主要贡献就是把这条链条做成 finite-sample statement。
Core Idea
论文的核心思想可以压缩成一句话：把 Kalman filter 看成一个稳定闭环系统的有限记忆 AR 展开，然后用两层线性网络对对应的 Hankel prediction operator 做低秩分解；hidden layer 因为处在 factorization 的中间位置，被迫实现与 Kalman state estimate 等价的 sufficient statistic。
具体地，稳态 Kalman predictor 可以展开为 x_hat_t = C z_t + Abar^L x_hat_{t-L}。只要 Abar = A-FC 稳定，后项随 L 指数衰减，于是过去 L 步 input-output 足以近似当前 filter state。未来 H 步输出又满足 y_{t:t+H-1} ≈ O x_hat_t + noise ≈ O C z_t + noise。于是 supervised learning 看到的是 z_t -> future y 的 regression，但这个 regression 的 population operator 是 rank-n 的 OC。
和 prior 的本质区别是：prior 多数显式估计 OC 后再分解；本文直接参数化为 G2G1，把“分解”变成训练过程的一部分。这个变化引入的 inductive bias 是低秩 bottleneck：hidden dimension 限制迫使模型把过去窗口压缩成预测未来所需的 n 维 statistic。在线性 Gaussian LDS 中，这个 statistic 正是 Kalman estimate。
Method
1. 有限窗口 AR 化：解决递归 filter 不直接匹配 feedforward AR 模型的问题。通过展开 Kalman predictor，把 x_hat_t 写成过去 L 步 y,u 的线性函数加 truncation residual。需要它是因为模型没有内部 recurrence，只能看固定历史窗口；核心变化是把 filtering 问题转成 supervised regression 问题。
2. 两层线性低秩参数化：解决 Hankel operator 的 latent factorization 问题。G2G1 拟合未来输出预测，hidden width h 对应候选系统阶数。需要 bottleneck，否则浅层 full-rank AR predictor 可以预测但不必形成 state-like representation；核心变化是把 state extraction 内化为网络 factor。
3. 优化 landscape 分析：解决非凸 ERM 是否真的能到全局解的问题。作者借用 shallow linear network 的 landscape 结果，在 Gram matrix 满秩/persistent excitation 下说明固定 h 时无 spurious local minima，saddle 是 strict saddle。注意这不是完整的实际 Adam 收敛证明，尤其 constrained norm ball 下的算法保证文中未充分说明。
4. 从 prediction 到 latent recovery 的误差传递：先用 martingale offset complexity 控制训练误差，再用 persistent excitation 把 in-sample prediction error 转为 Frobenius parameter error，最后用低秩 factor perturbation 将 OC 的误差转成 G1 与 C 的 alignment。这个链条是论文最实质的技术组织。
Key Insight / Why It Works
最关键 insight 是：future prediction 的 minimal sufficient statistic 在这个问题里就是 Kalman state estimate，而两层线性 bottleneck 让模型没有太多别的选择。只要 H 足够让 O full column rank、L 足够让 C full row rank，OC 的 rank-n factorization 在可逆变换外基本确定；因此 G1 z_t 必须对齐到 C z_t，也就是近似 x_hat_t。
有效性主要来自 latent structure + better inductive bias，不是 scaling。数据量 T 只负责把估计误差压下去；真正让 hidden representation 有语义的是低秩瓶颈和线性系统的 observability/controllability。没有这个 bottleneck，模型可以学 full AR predictor，但 representation 未必可解释；没有稳定 Kalman truncation，固定窗口也无法近似递归 filter。
最可能的核心贡献是把三个已知事实拼成一个 representation theorem：Kalman filter 可 AR 展开、Hankel operator 低秩、浅层线性网络 factorization landscape benign。单独看每一块都不是全新，但把它们组合成“ERM hidden activation recovers Kalman estimate”的 finite-sample guarantee 是实质新增信息。
哪些是辅助：optimization landscape 结果重要但不是现象的根因；它说明可训练性，而不是为什么 hidden state 有语义。实验也主要是 sanity check。architecture search 找到 h=n 的现象有趣，但理论上仍依赖 rank/width 条件，不能过度解读为自动 model order discovery 已解决。
一个值得注意的点：论文证明的是 recovery of Kalman filter estimates，而不是 recovery of true latent states。若系统状态本身只是一个 realization，Kalman estimate 已经是 input-output statistics 下的最优 linear sufficient statistic；这和“世界模型是否理解真实状态”之间仍有概念距离。
Relation To Prior Work
最接近的是 finite-sample LDS identification / ARX regression / subspace identification。传统 pipeline 是 estimate Hankel -> Ho-Kalman/SVD -> state-space realization；本文把 Ho-Kalman 式的 factorization 隐式放进两层网络训练里。因此它不是从零发明新辨识算法，而是把经典 subspace idea 重写为 end-to-end representation learning theorem。
和 Tsiamis/Pappas、Lale 等 AR regression 方向相比，本文同样关注 Kalman filtering 和有限样本，但关注点从 output prediction/parameter estimation 转向 hidden activation 的语义。和 nuclear norm / low-rank Hankel 方法相比，本文用显式 factorized nonconvex parameterization，而不是 convex relaxation 或后处理。
和 nonconvex LDS learning 工作相比，它更像 shallow linear network matrix factorization + system ID 的交叉。Hardt et al. 的 recurrent linear model直接学 state-space dynamics；本文不学 A,B,C,F，而是学 finite-window predictor 的低秩 factor。这个差异很重要：它避免显式模型辨识，但代价是历史窗口 L 和输出 horizon H 带来的维度膨胀。
和 Transformer approximates Kalman filter 的工作相比，本文更弱也更干净：模型是线性的、单系统训练、没有跨系统 generalization；但它给出了 hidden representation recovery 的统计保证，而不仅是表达能力。
Dataset / Evaluation
实验覆盖的是理论设定内的系统：随机生成线性系统和 ControlGym 中的线性 benchmark。没有真实世界、没有真机、没有非线性系统，也没有多系统 meta-learning 或 distribution shift 下的系统泛化。
评估基本验证了两个 claim：hidden dimension 接近真实状态维度时训练损失较优；训练后的 G1 activation 经线性变换后与 Kalman predicted state 高度对齐；训练数据变多时 recovery error 下降。这些都支持论文的主张，但支持范围严格限于线性 Gaussian/LDS-like regime。
实验没有真正检验更强叙事，例如“大型 AR 模型会自然学习世界状态”。它验证的是一个受控线性机制，不应外推到语言模型或视频 world model。benchmark leakage/implicit memorization 不是这里的主要问题，因为任务是合成动力系统；更大的问题是 evaluation 太贴近理论假设，不能说明 robustness 或 generalization。
Limitation
1. 前提强：线性系统、Gaussian 独立噪声、已进入稳态 Kalman regime、A 非爆炸、A-FC 稳定、observability/stabilizability、充分 excitation、O/C 满秩。这些条件缺一部分，证明链条就可能断。
2. 模型并不计算高效：true Kalman filter 递归维护 n 维状态；本文的 AR 模型需要 L 步历史，输入维度随 L 增长。它证明 representation emergence，不证明这是更好的 filtering algorithm。
3. recovery 是 up to similarity transform，且是对 Kalman estimate 的 recovery。对于真实 latent state 的语义解释仍依赖选择的 realization；这在系统辨识里正常，但在 representation learning 叙事里容易被过度宣传。
4. hidden dimension/model order 的选择没有完全解决。理论 ERM 里允许 h≤r 并做 architecture search，但实际如何稳定发现 n、过参数化时 hidden representation 是否仍可解释，文中未充分说明。
5. 优化保证和实际训练之间有 gap。论文证明 landscape benign，但 constrained problem 的 perturbed/projected GD 全局收敛没有给出；实验用 Adam + weight decay，理论和 practice 之间仍是经验连接。
6. H/L trade-off 是真实上限：H<n 时 O 不满秩无法恢复状态；H 增大又使噪声 Toeplitz 项和 bound 变差。L 需要随稳定裕度和 T 对数增长；filter 接近不稳定时窗口会变长，scaling 变差。
7. 对非高斯、非线性、多系统、闭环数据的泛化没有建立。若扩展到更复杂序列模型，增益来源可能会混入 data coverage、implicit regularization 和 architecture bias，当前理论无法归因。
Takeaway
- 1. 最值得记住的是：在可观可控的线性 Gaussian 系统中，future prediction + low-rank bottleneck 足以让 hidden layer 学到 Kalman sufficient statistic；state supervision 不是必要条件。
- 2. 这篇论文把 subspace identification 的 factorization 观点翻译成了 representation learning 语言。
- 未来很多“世界模型是否学状态”的理论工作，可以沿着“预测 operator 的低秩/可分解结构 + bottleneck representation”这条线做。
- 3. 可迁移的 insight 不是两层线性网络本身，而是证明路线：先刻画最优 predictor 的结构，再证明网络参数化的 factor 被迫对齐到某个 latent sufficient statistic，最后用 perturbation 把 operator error 传到 representation error。
一句话总结
这篇论文是把经典 Kalman filtering/subspace identification 低秩结构改写为端到端两层线性 AR representation learning 定理的工作，真正贡献在于证明输出预测损失下的 hidden activation 可有限样本恢复 Kalman 状态估计。