精读笔记
Problem Setting
《Koopman operator learning for predictive control via Khatri-Rao kernel regression》（arXiv preprint / 2026）处理的是 controlled Koopman model 在 MPC 中的一个实际瓶颈：模型既要能表达非输入仿射、强 state-input coupling 的非线性动力学，又要能递推多步、低维、数值稳定、可被在线优化器使用。
真正困难点是两个矛盾叠加：一方面，越 general 的 lifting 越容易维度爆炸、秩亏和过拟合；另一方面，MPC 需要的是短到中期 rollout 准确性，而标准 EDMD/Koopman regression 优化的是 isolated one-step residual。对 GeKo 来说这个问题更尖锐，因为递推项是 z⊗v，误差不仅沿 lifted dynamics 传播，还会通过输入 lifting 与状态 lifting 的乘积放大。
以前路线的卡点很明确：lifted linear / DMDc 便宜但输入表达弱；bilinear Koopman 适合 control-affine 结构但不自然覆盖任意输入非线性；deep Koopman 可以拟合复杂表示但训练非凸、理论和可控性弱；fixed-horizon multi-step predictors 降低了 rollout error，但牺牲了单一 Koopman operator 的可迭代结构。本文试图保留 GeKo 的 generality，同时把它改造成 MPC 友好的 surrogate。
Motivation
作者的核心观察是：GeKo 的理论表达已经足够 general，但原始 Khatri-Rao kernel regression 没有解决 MPC 使用场景里的两个关键缺口：多步递推误差和模型复杂度。换句话说，缺的不是另一个 Koopman definition，而是一个让 GeKo 在有限数据、有限字典、有限算力下可控可用的 learning pipeline。
更具体地说，一步回归只让 K 看到训练样本中的真实 one-step pairs；但 MPC rollout 时 K 会反复作用在自己预测出来的 lifted states 上，分布会偏离训练支撑。作者没有直接做非凸 recursive loss，而是用 trajectory time-shifted snapshots 近似覆盖 rollout 中会遇到的 evolved states。这是一个务实折中：保持 ridge regression 的闭式凸结构，同时缓解多步误差。
另一个动机是 kernel/RFF dictionary 通常故意过参数化，理论上提升表达能力，实践上却制造大量冗余方向。若不压缩，Khatri-Rao feature 维度按 state/input lifting 乘积增长，MPC 直接不可用。因此需要一种不破坏 z⊗v separability 的降维方式。
Core Idea
论文真正核心的思想不是“用 RFF 做 Koopman”，而是把 GeKo 的 product-Hilbert-space 表达变成一种可训练、可压缩、可递推的张量双线性控制模型：state 和 input 分别 lift，然后通过 z_t⊗v_t 进入同一个线性 operator K。这个 inductive bias 很强：它假设复杂 controlled dynamics 可以在 state features 与 input features 的乘积空间中近似线性演化，同时保留输入与状态的可分结构。
相比 prior，关键变化在训练信息流：不是从孤立 snapshot pairs 学一个 one-step map，而是从同一批轨迹中抽取多段窗口，把窗口内所有时间偏移的一步关系堆叠起来拟合同一个 K。它没有像 fixed-horizon predictor 那样为每个 horizon 学不同映射，因此仍是一个可任意递推的 Koopman-like operator；但它又让 K 在训练时看到 trajectory evolution 后的状态分布，减少 one-step training 和 multi-step deployment 的 mismatch。
第二个核心思想是结构化压缩：不直接对 Khatri-Rao 大矩阵做黑盒降维，而是分别压缩 state lifting 和 input lifting，再在低维乘积空间中 refit K。这样保留 GeKo 的 separable realization，也把维度从 n_z n_v 降到 r_z r_v。这个设计是可迁移的：它本质上是对 product feature model 的 factor-wise rank control。
Method
1. GeKo / Khatri-Rao realization：解决 controlled Koopman 中输入非线性表达不足的问题。通过 z_{t+1}=K(z_t⊗v_t)，任意输入非线性可以通过 Ψ_u(u) 进入乘积特征，而不是被限制成 additive forcing 或 input-affine bilinear form。核心变化是把控制变量从外部 forcing 变成 product Hilbert space 的一个因子。
2. Time-sequenced multi-step regression：解决 one-step regression 与 MPC rollout 之间的不一致。它把多个 trajectory windows 内的 shifted one-step pairs 横向堆叠，仍然解一个 ridge LS。需要它的原因是递推模型在第 k 步面对的是 evolved state distribution；只训练起点 one-step pairs 会低估 rollout error。核心变化是训练分布更接近部署时的递推分布，但仍保持单一 K。
3. Window-aware selection：解决全窗口数据过多以及混沌系统中同起点附近轨迹可能快速分叉的问题。用 start/mid/end anchors 做聚类，本质是覆盖 trajectory segment 而非覆盖单点 state。这个机制偏 engineering，但对 chaotic rollout 的数据覆盖可能很关键。
4. Structured SVD reduction：解决 RFF/kernel dictionary 的冗余、病态和 MPC 维度问题。分别对 state lifting 与 input lifting 做有效秩截断，再在 reduced product space refit K。核心变化是把过参数字典变成 data-adapted coordinate system，同时不破坏 Khatri-Rao separability。
5. Decoder-based MPC objective：解决 lifted feature error 与物理状态控制目标不对齐的问题。线性 decoder D 把 reduced lifted state 映射回 x，MPC cost 只惩罚可解码到物理状态的方向。这一点很重要，因为 feature-space accurate 不等于 control-relevant accurate。
Key Insight / Why It Works
最可能真正有效的部分是“训练分布对齐”而不是某个新的 kernel trick。多步 stacking 仍然是 one-step LS，但它改变了 one-step pairs 的采样方式：K 不只在原始 snapshot distribution 上拟合，而是在轨迹推进后的多个 time-shift distributions 上拟合。对于 recursive predictor，这相当于弱化版的 rollout-aware training，却避免了非凸多步递推 loss。这是本文最实质的贡献。
第二个有效来源是 representation alignment。RFF 提供一个足够大的候选特征空间，SVD 再把它投影到训练数据实际占据的子空间；decoder 又把 MPC cost 对齐到 physical state。这三者组合后，模型不会浪费容量去拟合大量 control-irrelevant feature drift。文中观察到 decoded-state error 明显小于 lifted-state error，说明很多 lifted 方向虽然漂移，但不影响控制目标。这一点很有价值：MPC surrogate 不需要 feature-space 全局准确，只需要 control-relevant projection 准确。
第三个来源可能是 data coverage / scaling。Lorenz 实验用了较长轨迹、覆盖 attractor box 的初值、多频输入、较大的 RFF 字典，再通过 SVD 选有效秩。闭环成功很可能依赖这些数据覆盖条件。论文没有 ablation，因此不能判断增益到底来自 GeKo 的 product structure、多步 stacking、SVD 压缩，还是 simply enough data + overcomplete features + nonlinear MPC。
需要直说的是：这里的“multi-step”并不是直接最小化 recursive multi-step prediction error。它只是把真实轨迹上的 shifted one-step residual 加入同一个 LS。它能降低 compounded error，是因为训练样本覆盖了未来状态分布，而不是因为优化目标显式建模误差传播。因此如果 closed-loop 访问到训练窗口未覆盖的区域，优势可能迅速消失。
SVD reduction 也不是普通压缩那么简单。它实际引入了一个 data manifold prior：只有训练数据中高能量的 lifted directions 被保留。这通常有助于数值稳定和 MPC 速度，但也可能删除低能量但控制关键的方向。文中未充分说明这一风险如何检测。
Relation To Prior Work
这篇属于 Koopman-based predictive control 谱系中“controlled Koopman realization + convex regression + MPC deployment”的路线，最接近 GeKo [1]、KCF、bilinear Koopman、kernel EDMD、以及近期 multi-step Koopman LS。
和 DMDc / lifted linear Koopman 的本质差异在输入建模：DMDc 把输入作为 additive forcing，GeKo 把输入 lift 后与 state lift 做 product，因此天然表达输入非线性和 state-input coupling。和 bilinear Koopman 的差异是 bilinear Koopman 通常对应 control-affine 或每个输入通道一个 operator，而这里输入通过 Ψ_u(u) 进入任意 nonlinear basis，形式上更 general。
和 KCF 的差异更微妙。KCF 是按 constant input indexing 得到 Koopman operator family，再建立 separable 表达；GeKo 是直接在 H_x⊗H_u 中定义单个 operator。两者都能覆盖 linear/bilinear control forms。本文的新增不在理论定义，而在如何让 GeKo 经由 Khatri-Rao regression、multi-step stacking 和 structured reduction 变成 MPC model。
和 fixed-horizon multi-step Koopman predictor 相比，本文保留一个可递推 K，而不是为每个 horizon 拟合 condensed mapping。这是实质差异：它牺牲了一部分 horizon-specific 最优性，换来 operator consistency 和 arbitrary-horizon rollout。
看似新的部分中，RFF、ridge regression、SVD 截断、MPC 都不是新思想；实质创新是这些组件在 GeKo/Khatri-Rao product structure 下的组合方式，尤其是 factor-wise SVD 保持 separability，以及 shifted-window LS 保持单 operator 的同时改善 multi-step behavior。
Dataset / Evaluation
实验只覆盖一个系统：controlled Lorenz。这个系统有代表性，因为混沌、非线性强、目标是不稳定平衡点，能测试短期预测和闭环 stabilization。但它仍然是低维、单输入、仿真环境，不能支撑关于高维系统、多输入复杂约束或真实部署的强 claim。
评估支持的核心 claim 是：在数据覆盖充分的低维混沌系统上，reduced GeKo model 可以作为 nonlinear MPC surrogate 工作，并且 multi-step decoded error 可作为 horizon selection 的经验诊断。它没有充分支持的 claim 是：该方法系统性优于已有 Koopman MPC 或 SINDy-MPC；也没有证明 GeKo 的 generality 在实验中带来不可替代收益，因为没有与 lifted linear、bilinear、KCF、one-step GeKo、no-SVD GeKo 做严谨 ablation。
闭环求解时间也暴露了 scalability 问题：即使低维 Lorenz、reduced 后 r_z=150、r_v=13，通用 SQP 也远慢于采样周期。作者承认 tailored solvers 是 future work。这意味着当前结果更像 proof-of-concept，而不是实时 MPC 方案。
没有真实世界实验，没有跨系统验证，没有噪声/扰动/模型不确定性分析，没有 region of attraction 估计。evaluation 的强项是 pipeline 完整；弱项是 claim attribution 不清。
Limitation
第一，方法成立强依赖数据覆盖。multi-step stacking 只能覆盖训练轨迹经过的 evolved states；如果 MPC 把系统带到未覆盖区域，K 的 recursive error 仍可能爆炸。所谓泛化主要是 kernel/RFF 插值意义上的局部泛化，不应理解成跨任务或大范围 extrapolation。
第二，scalability 上限明显。GeKo 的表达能力来自 z⊗v，但代价也是 r_z r_v。多输入系统中 input lifting 维度会迅速增加，在线 MPC 变成非凸 NLP，且每步还要计算 lifted input features。论文没有展示高维状态或多输入场景，无法判断是否可扩展。
第三，缺少稳定性和可行性保证。近期 Koopman MPC 文献已经开始强调 robust stability / recursive feasibility；本文主要是 empirical pipeline，没有 terminal ingredients、误差界或闭环 guarantee。因此它更像 identification-for-MPC，而不是 certified Koopman MPC。
第四，增益归因不清。多步 regression、SVD、RFF 字典大小、window clustering、输入激励覆盖、decoder objective 都可能贡献结果。没有 ablation 时，不能判断核心提升来自哪一项。部分增益可能主要来自 scaling / data，而非 GeKo 结构本身。
第五，SVD 截断有隐含风险。它按数据能量保留方向，但控制相关方向未必高能。尤其在 stabilization 到不稳定平衡点时，局部不稳定模态可能在全局训练数据中占能量不高。文中未充分说明如何避免删掉这些方向。
第六，decoded-state error 作为 horizon selection 是合理工程诊断，但不是控制性能保证。小预测误差不等于优化 landscape 正确，也不等于约束处理可靠。MPC 决策对梯度、局部非凸性和输入敏感性都很敏感，这些没有被诊断覆盖。
Takeaway
- 1. 对 Koopman MPC 来说，关键不只是 one-step model accuracy，而是训练分布与 recursive rollout 分布的对齐。
- 本文的 shifted-window LS 是一个保持凸闭式解的务实做法，值得迁移到其他 operator learning surrogate。
- 2. Product-structured lifting 的压缩应尽量 factor-wise 做，而不是直接对乘积空间黑盒降维。
- 分别压缩 state/input lifting 可以保留结构、降低维度，并保持后续控制问题的可解释组织。
一句话总结
这篇论文把 GeKo 从一个 general controlled Koopman 表达推进为一个可用于 MPC 的结构化学习 pipeline，其实质贡献是用轨迹时移回归和因子化 SVD 压缩来缓解递推误差与维度爆炸，属于 Koopman-control 从理论 realization 走向工程可用 surrogate 的方法演化。