让模型预见分布漂移：动态系统颠覆性设计引领时域泛化新革命｜NeurIPS 2024

tayun

新智元报道

编辑：LRST

【新智元导读】研究人员提出了一种方法，能够在领域数据分布持续变化的动态环境中，基于随机时刻观测的数据分布，在任意时刻生成适用的神经网络，实现前所未有的泛化能力。

在实际应用中，数据集的数据分布往往随着时间而不断变化，预测模型需要持续更新以保持准确性。


时域泛化旨在预测未来数据分布，从而提前更新模型，使模型与数据同步变化。

然而，传统方法假设领域数据在固定时间间隔内收集，忽视了现实任务中数据集采集的随机性和不定时性，无法应对数据分布在连续时间上的变化。

此外，传统方法也难以保证泛化过程在整个时间流中保持稳定和可控。

为此，研究人员提出了连续时域泛化任务，并设计了一个基于模型动态系统的时域泛化框架Koodos，使得模型在连续时间中与数据分布的变化始终保持协调一致。

下图展示了模型在领域数据随时间发生旋转和膨胀时的泛化表现。通过在一些随机时间点（蓝色标记点）的观测，模型可以在任意时刻生成适用的神经网络，其决策边界始终与数据分布保持协调一致。

Koodos通过库普曼算子将模型的复杂非线性动态转化为可学习的连续动态系统，同时利用先验常识以确保泛化过程的稳定性和可控性。实验表明，Koodos显著超越现有方法，为时域泛化开辟了全新的研究方向。

论文链接：https://arxiv.org/pdf/2405.16075

开源代码：https://github.com/Zekun-Cai/Koodos/

逐步教程：

https://github.com/Zekun-Cai/Koodos/blob/main/Tutorial_for_Koodos.ipynb

研究人员在代码库中提供了详细的逐步教程，涵盖了Koodos的实现、核心概念的解读以及可视化演示。整个教程流程紧凑，十分钟即可快使掌握Koodos的使用方法，力荐尝试！

研究背景

在实际应用中，训练数据的分布通常与测试数据不同，导致模型在训练环境之外的泛化能力受限。领域泛化（Domain Generalization, DG）作为一种重要的机器学习策略，旨在学习一个能够在未见目标领域中也保持良好表现的模型。

近年来研究人员发现，在动态环境中，领域数据（Domain Data）分布往往具有显著的时间依赖性，这促使了时域泛化（Temporal Domain Generalization, TDG）技术的快速发展。

时域泛化将多个领域视为一个时间序列而非一组独立的静态个体，利用历史领域预测未来领域，从而实现对模型参数的提前调整，显著提升了传统DG方法的效果。

然而，现有的时域泛化研究集中在「离散时间域」假设下，即假设领域数据在固定时间间隔（如逐周或逐年）收集。

基于这一假设，概率模型被用于预测时域演变，例如通过隐变量模型生成未来数据，或利用序列模型（如LSTM）预测未来的模型参数。然而在现实中，领域数据的观测并不总是在离散、规律的时间点上，而是随机且稀疏地分布在连续时间轴上。

例如，图1展示了一个典型的例子——基于推文数据进行社交媒体舆情预测。与传统TDG假设的领域在时间轴上规律分布不同，实际中只能在特定事件（如总统辩论）发生时获得一个域，而这些事件的发生时间并不固定。

同时，概念漂移（Concept Drift）在时间轴上发生，即领域数据分布随着时间不断演变：如活跃用户增加、新交互行为形成、年龄与性别分布变化等。理想情况下，每个时态域对应的预测模型也应随时间逐渐调整，以应对这种概念漂移。

最后，由于未来的域采集时间未知，预测模型需要泛化到连续时间的任意时刻。

图1：连续时域泛化示意图。图中展示了通过推文训练分类模型进行舆情预测。其中训练域仅能在特定政治事件（如总统辩论）前后采集。研究人员的目标是通过这些不规律时间分布的训练域来捕捉分布漂移，最终使模型能够推广到任意未来时刻

事实上，领域分布在连续时间上的场景十分常见，例如：

1. 事件驱动的数据采集：仅在特定事件发生时采集领域数据，事件之间没有数据。

2. 流数据的随机观测：领域数据在数据流的任意时间点开始或结束采集，而非持续进行。

3. 离散时态域但缺失：尽管领域数据基于离散时间点采集，但部分时间节点的领域数据缺失。

为了应对这些场景中的模型泛化，研究人员提出了「连续时域泛化」（Continuous Temporal Domain Generalization, CTDG）任务，其中观测和未观测的领域均分布于连续时间轴上随机的时间点。

CTDG关注于如何表征时态领域的连续动态，使得模型能够在任意时间点实现稳定、适应性的调整，从而完成泛化预测。

核心挑战

CTDG任务的挑战远超传统的TDG方法。CTDG不仅需要处理不规律时间分布的训练域，还能够让模型泛化到任意时刻，即要求在连续时间的每个点上都能精确描述模型状态。

而TDG方法则仅关注未来的单步泛化：在观测点优化出当前模型状态后，只需将其外推一步即可。

该特性使得CTDG区别于TDG任务：CTDG的关键在于如何在连续时间轴上同步数据分布和模型参数的动态演变，而不是仅局限于未来某一特定时刻的模型表现。

具体而言，与TDG任务相比，CTDG的复杂性主要来自以下几个尚未被充分探索的核心挑战：

1. 如何建模数据动态并同步模型动态

CTDG要求在连续时间轴上捕捉领域数据的动态，并据此同步调整模型状态。然而，数据动态本身难以直接观测，需要通过观测时间点来学习。此外，模型动态的演变过程也同样复杂。理解数据演变如何驱动模型演变构成了CTDG的首要挑战。

2. 如何在高度非线性模型动态中捕捉主动态

领域数据的预测模型通常依赖过参数化（over-parametrized）的深度神经网络，模型动态因此呈现出高维、非线性的复杂特征，导致模型的主动态嵌藏在大量潜在维度中。如何有效提取并将这些主动态映射到可学习的空间，是CTDG任务中的另一重大挑战。

3. 如何确保长期泛化的稳定性和可控性

为实现未来任意时刻的泛化，CTDG必须确保模型的长期稳定性。此外，在许多情况下，用户可能拥有数据动态的高层次先验常识。如何将这些先验常识嵌入CTDG的优化过程中，进而提升泛化的稳定性和可控性，是一个重要的开放性问题。

技术方法

问题定义

在CTDG中，一个域D(t)表示在时间t采集的数据集，由实例集 组成，其中 和N(t)分别为特征值，目标值和实例数。

研究重点关注连续时间上的渐进性概念漂移，表示为领域数据的条件概率分布P(Y(t)|X(t))随时间平滑变化。

在训练阶段，模型接收一系列在不规律时间点T={t1,t2,…,tT}上收集的观测域{D(t1),D(t2),…,D(tT)}，其中每个时间点ti∈T是定义在连续时间轴R+上的实数，且满足t1 2<… T

在每个ti∈T上，模型学习到领域数据D(ti)的预测函数g(;θ(ti))，其中θ(ti)表示ti时刻的模型参数；CTDG的目标是建模参数的动态变化，以便在任意给定时刻sT上预测模型参数θ(s)，从而得到泛化模型g(;θ(s))。

在后续部分中，使用简写符号Di、Xi、Yi和θi，分别表示在时间ti上的D(ti)、X(ti)、Y(ti)和θ(ti)

设计思路

该方法通过模型与数据的同步、动态简化表示，以及高效的联合优化展开。具体思路如下：

1. 同步数据和模型的动态


研究人员证明了连续时域中模型参数的连续性，而后借助神经微分方程（Neural ODE）建立模型动态系统，从而实现模型动态与数据动态的同步。

2. 表征高维动态到低维空间

将高维模型参数映射到一个结构化的库普曼空间（Koopman Space）中。该空间通过可学习的低维线性动态来捕捉模型的主要动态。

3. 联合优化模型与其动态

将单个领域的模型学习与各时间点上的连续动态进行联合优化，并设计了归纳偏置的约束接口，通过端到端优化保证泛化的稳定性和可控性。

解决方案

Step 1. 数据动态建模与模型动态同步

1. 分布变化的连续性假设


首先假设数据分布在时间上具有连续演化的特性，即条件概率分布Pt(Y|X)随时间平滑变化，其演化规律可由一个函数f所描述的动态系统刻画。

尽管真实世界中的渐进概念漂移可能较为复杂，但因概念漂移通常源于底层的连续过程（如自然、生物、物理、社会或经济因素），这一假设不失普适性。

2. 分布变化引发的模型参数连续演化


基于上述假设，模型的函数功能空间g(;θt)应随数据分布变化同步调整。

借助常微分方程来描述这一过程：

由此可推导出模型参数θt的演化满足：

其中，Jg(θt)是g对θt的雅可比矩阵。


这一结果表明，如果数据分布的演化在时间上具有连续性，那么θt的演化过程也具有连续性，即模型参数会随数据分布的变化而平滑调整。上式为θt建立了一个由微分方程描述的模型动态系统。

3、模型动态系统学习


由于数据动态f的具体形式未知，直接求解上述微分方程并不可行。为此，引入一个由神经网络定义的连续动态系统，用可学习的函数h(θt,t;)描述模型参数θt的变化。通过鼓励模型动态和数据动态之间的拓扑共轭（Topological Conjugation）关系使h逼近真实动态。

具体而言，拓扑共轭要求通过泛化获得的模型参数与直接训练得到的参数保持一致。为此，设定以下优化目标，以学习h的参数：


其中，θi通过在时刻ti的领域上直接训练获得， 则表示从时间t j 通过动态h演变至t i 的泛化参数：


通过这一优化过程，可以建立模型动态与数据动态之间的同步机制；借助动态函数h，可以在任意时刻精确求解模型的状态。

Step 2. 通过库普曼算子简化模型动态

非线性动态线性化


在实际任务中，预测模型通常依赖于过参数化的深度神经网络，使得模型动态h呈现为在高维空间中纠缠的非线性动态。直接对h建模不仅计算量大，且极易导致泛化不稳定。

然而，h受数据动态f的支配，而数据动态通常是简单、可预测的。这意味着在过参数化空间中，模型的主动态（Principal Dynamics）可以在适当转换的空间内进行更易于管理的表示。

受此驱动，研究人员引入库普曼理论（Koopman Theory）来简化复杂的模型动态。库普曼理论在保持动态系统特征的同时将复杂的非线性动态线性化。


具体而言，定义一个库普曼嵌入函数，将原始的高维参数空间映射到一个低维的库普曼空间中：


其中，z表示库普曼空间中的低维表示。通过库普曼算子K，可以在线性空间中刻画z的动态：

一旦获得了简化的动态表示，可以在库普曼空间中更新模型参数，而后将其反映射回原始参数空间：

最终，通过库普曼算子的引入，实现了对模型动态的简化，保证了泛化过程的稳健性。

Step 3. 联合优化与先验常识结合

模型及其动力学的联合优化


研究人员对多个组件同时施加约束确保模型能稳定泛化，其包含以下关键项：

1. 预测准确性：通过最小化预测误差，使预测模型在每个观测时间点都能准确预测实际数据。

2. 泛化准确性：通过最小化预测误差，使泛化模型在每个观测时间点都能准确预测实际数据。

3. 重构一致性：确保模型参数在原始空间与库普曼空间之间的转换具有一致性。

4. 动态保真性：约束库普曼空间的动态行为，使得映射后的空间符合预期的动态系统特征。

5. 参数一致性：确保泛化模型参数映射回原始空间后与预测模型参数保持一致。

利用库普曼算子评估和控制泛化过程


引入库普曼理论的另一优势在于，可以通过库普曼算子的谱特性来评估模型的长期稳定性。此外，还可以在库普曼算子中施加约束来控制模型的动态行为。

- 系统稳定性评估

通过观察库普曼算子的特征值，可以判断系统是否稳定：

1、若所有特征值实部为负，系统会稳定地趋向于一个平衡状态。

2、若存在特征值实部为正，系统将变得不稳定，模型在未来可能会崩塌。

3、若特征值实部为零，系统可能表现出周期性行为。通过分析这些特征值的分布，可以预测系统的长期行为，识别模型在未来是否可能出现崩溃的风险。

- 泛化过程约束

可以通过对库普曼算子施加显式约束来调控模型的动态行为。例如：

1. 周期性约束：当数据动态为周期性时，可将库普曼算子K设为反对称矩阵，使其特征值为纯虚数，从而使模型表现出周期性行为。

2. 低秩近似：将K表示为低秩矩阵，有助于控制模型的自由度，避免过拟合到次要信息。

通过这些手段，不仅提高了泛化的长期稳定性，还增强了模型在特定任务中的可控性。

实验

实验设置

为验证算法效果，研究人员使用了合成数据集和多种真实世界场景的数据集：

合成数据集

包括Rotated 2-Moons和Rotated MNIST数据集，通过在连续时间区间内随机生成时间戳，并对Moons和MNIST数据按时间戳逐步旋转生成连续时域。

真实世界数据集

1. 事件驱动数据集Cyclone：基于热带气旋的卫星图像预测风力强度，气旋发生日期对应连续时域。

2. 流数据集Twitter和House：分别从任意时间段抽取推文和房价数据流构成一个领域，多次随机抽取形成连续时域。

3. 不规则离散数据集Yearbook：人像图片预测性别，从84年中随机抽取40年数据作为连续时域。

实验结果与分析

定量分析

研究人员首先对比了Koodos方法与各基线方法的定量性能。

表1显示，Koodos方法在所有数据集上展现了显著的性能提升。在合成数据集上，Koodos能够轻松应对持续的概念漂移，而所有基线方法在这种场景下全部失效。

在真实世界数据集上，尽管某些基线方法（如CIDA、DRAIN和DeepODE）在少数场景中略有表现，但其相较于简单方法（如Offlne）的改进非常有限。

相比之下，Koodos显著优于所有现有方法，彰显出在时域泛化任务中考虑分布连续变化的关键作用。

定性分析

1. 决策边界


为直观展示泛化效果，研究人员在Rotated 2-Moons数据集上进行了决策边界的可视化。

该任务具有极高难度：模型需在0到35秒左右的35个连续时域上训练，随后泛化到不规律分布在35到50秒的15个测试域。而现有方法通常只能泛化至未来的一个时域（T+1），且难以处理不规律的时间分布。

图2：2-Moons数据集决策边界的可视化（紫色和黄色表示数据区域，红线表示决策边界）。从上到下比较了两种基线方法和Koodos；从左到右显示了部分测试域（15选7，所有测试域的分布在时间轴上用红点标记）。

图2从15个测试域中选取了7个进行可视化。

结果清晰地表明，基线方法在应对连续时域的动态变化时表现不足。随着时间推进，决策边界逐渐偏离理想状态。尤其是最新的DRAIN方法（ICLR23）在多步泛化任务中明显失效。

相比之下，Koodos在所有测试域上展现出卓越的泛化能力，始终保持清晰、准确的决策边界，与实际数据分布变化高度同步。这一效果突显了Koodos在时域泛化任务中的革命性优势。

2. 模型演变轨迹


为更深入地分析模型的泛化能力，通过t-SNE降维，将不同方法的模型参数的演变过程（Model Evolution Trajectory）在隐空间中可视化（图3）。

图3：模型状态在隐空间中的时空轨迹。Koodos展现出与数据动态和谐同步的模型动态

可以看出，Koodos的轨迹呈现出平滑而有规律的螺旋式上升路径，从训练域平滑延伸至测试域。

这一轨迹表明，Koodos能够在隐空间中有效捕捉数据分布的连续变化，并随时间自然地扩展泛化。相比之下，基线模型的轨迹在隐空间中缺乏清晰结构，随着时间推移，逐渐出现明显的偏离，未能形成一致的动态模式。

时域泛化的分析与控制

在Koodos模型中，库普曼算子为分析模型动态提供了有效手段。

对Koodos在2-Moons数据集上分析表明，库普曼算子的特征值在复平面上分布在稳定区和不稳定区，这意味着Koodos在中短期内能稳定泛化，但在极长时间的预测上将会逐渐失去稳定性，偏离预期路径（图4b）。

为提升模型的稳定性，研究人员通过将库普曼算子配置为反对称矩阵（即Koodos+版本），确保所有特征值为纯虚数，使模型具有周期性稳定特性。

在这一配置下，Koodos+展现出高度一致的轨迹，即使在长时间外推过程中依然保持稳定和准确，证明了引入先验常识对增强模型稳健性的效果（图4c）。

图4：非受控和受控条件下的极长期泛化预测模型轨迹。a：部分训练域数据；b：不受控，模型最终偏离预期；c：受控，模型始终稳定且准确。

结论

研究人员设计了一种基于模型连续动态系统的时域泛化方法，能够在数据域随时间逐渐演变的环境中，实现泛化模型的稳定性与可控性，未来还计划从多个方向进一步拓展这一技术的应用：

1. 生成式模型扩展：时域泛化与生成式模型任务有天然的关联，Koodos所具备的泛化能力能够为神经网络生成技术带来新的可能。

2. 非时态泛化任务：Koodos的应用并不局限于时域泛化，也可以适用于其他分布变化的任务中，研究人员计划探索其在非时态领域的应用。

3. 大模型集成：将探索时域泛化在大模型中的集成，帮助LLM在复杂多变的分布中保持鲁棒性和稳定性。

参考资料：

https://github.com/Zekun-Cai/Koodos/


来源：网易