arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2601.21135cs.LG
收藏

TRACE:因果表示学习中连续机制演化的轨迹恢复

TRACE: Trajectory Recovery for Continuous Mechanism Evolution in Causal Representation Learning

Shicheng Fan, Kun Zhang, Lu Cheng

AI总结:

研究现实世界系统中因果机制的连续转变问题,提出TRACE框架,将过渡机制建模为原子机制的凸组合,通过专家混合框架学习原子机制,能恢复机制轨迹,实验证明其在恢复混合轨迹上性能远超离散切换基线。

摘要:

时间因果表示学习方法假设因果机制在离散域之间瞬间切换,但现实世界系统常呈现连续机制转变。例如车辆动力学在转弯时逐渐演变,人类步态从行走平稳过渡到跑步。我们通过将过渡机制建模为有限多个原子机制的凸组合(由时变混合系数控制)来形式化此设置。理论上证明了潜在因果变量和连续混合轨迹可联合识别。我们还提出了TRACE,一个专家混合框架,每个专家在训练时学习一种原子机制,能在测试时恢复机制轨迹。实验表明TRACE恢复混合轨迹的相关性高达0.99,显著优于离散切换基线。

英文摘要:

Temporal causal representation learning methods assume that causal mechanisms switch instantaneously between discrete domains, yet real-world systems often exhibit continuous mechanism transitions. For example, a vehicle's dynamics evolve gradually through a turning maneuver, and human gait shifts smoothly from walking to running. We formalize this setting by modeling transitional mechanisms as convex combinations of finitely many atomic mechanisms, governed by time-varying mixing coefficients. Our theoretical contributions establish that both the latent causal variables and the continuous mixing trajectory are jointly identifiable. We further propose TRACE, a Mixture-of-Experts framework where each expert learns one atomic mechanism during training, enabling recovery of mechanism trajectories at test time. This formulation generalizes to intermediate mechanism states never observed during training. Experiments on synthetic and real-world data demonstrate that TRACE recovers mixing trajectories with up to 0.99 correlation, substantially outperforming discrete-switching baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑