面向监督因果学习的测试时训练方法 / Test Time Training for Supervised Causal Learning
1️⃣ 一句话总结
本文提出一种测试时训练框架,通过在测试阶段动态生成与当前样本匹配的训练数据,有效解决了监督因果学习方法在真实场景中泛化能力差、分布偏移敏感和组合推理失败的问题。
Supervised Causal Learning (SCL) has shown promise in causal discovery by framing it as a supervised learning problem. However, it suffers from significant out-of-distribution generalization challenges. We reveal three limitations of previous SCL practices: a significant performance gap between synthetic benchmarks and real-world data, fragility to distribution shifts, and failure in compositional generalization, collectively questioning its real-world applicability. To address this, we propose Test-Time Training for Supervised Causal Learning (TTT-SCL), a novel framework that dynamically generates training sets explicitly aligned with any specific test instance. We demonstrate the correlation between TTT-SCL and score-based methods, and design an efficient module for generating training sets based on the classic scoring function. Experiments on synthetic benchmarks, pseudo-real and real-world datasets demonstrate that TTT-SCL significantly outperforms existing SCL and traditional causal discovery methods.
面向监督因果学习的测试时训练方法 / Test Time Training for Supervised Causal Learning
本文提出一种测试时训练框架,通过在测试阶段动态生成与当前样本匹配的训练数据,有效解决了监督因果学习方法在真实场景中泛化能力差、分布偏移敏感和组合推理失败的问题。
源自 arXiv: 2605.30015