dorsal/arxiv
View SchemaTransformer Is Inherently a Causal Learner
| Authors | Xinyue Wang, Stephen Wang, Biwei Huang |
|---|---|
| Categories | |
| ArXiv ID | 2601.05647vv1 |
| URL | https://arxiv.org/abs/2601.05647 |
| License | http://creativecommons.org/licenses/by/4.0/ |
Abstract
We reveal that transformers trained in an autoregressive manner naturally encode time-delayed causal structures in their learned representations. When predicting future values in multivariate time series, the gradient sensitivities of transformer outputs with respect to past inputs directly recover the underlying causal graph, without any explicit causal objectives or structural constraints. We prove this connection theoretically under standard identifiability conditions and develop a practical extraction method using aggregated gradient attributions. On challenging cases such as nonlinear dynamics, long-term dependencies, and non-stationary systems, this approach greatly surpasses the performance of state-of-the-art discovery algorithms, especially as data heterogeneity increases, exhibiting scaling potential where causal accuracy improves with data volume and heterogeneity, a property traditional methods lack. This unifying view lays the groundwork for a future paradigm where causal discovery operates through the lens of foundation models, and foundation models gain interpretability and enhancement through the lens of causality.
{
"annotation_id": "e705772a-2ad8-45d5-8b88-926df4ccc7ad",
"date_created": "2026-02-17T05:53:03.993000Z",
"date_modified": "2026-02-17T05:53:03.993000Z",
"file_hash": "3a4a731e5a60cb1d34ab50fd82f8a4215de9e5f2843c601a60ba69dd9053727a",
"private": false,
"record": {
"abstract": "We reveal that transformers trained in an autoregressive manner naturally encode time-delayed causal structures in their learned representations. When predicting future values in multivariate time series, the gradient sensitivities of transformer outputs with respect to past inputs directly recover the underlying causal graph, without any explicit causal objectives or structural constraints. We prove this connection theoretically under standard identifiability conditions and develop a practical extraction method using aggregated gradient attributions. On challenging cases such as nonlinear dynamics, long-term dependencies, and non-stationary systems, this approach greatly surpasses the performance of state-of-the-art discovery algorithms, especially as data heterogeneity increases, exhibiting scaling potential where causal accuracy improves with data volume and heterogeneity, a property traditional methods lack. This unifying view lays the groundwork for a future paradigm where causal discovery operates through the lens of foundation models, and foundation models gain interpretability and enhancement through the lens of causality.",
"arxiv_id": "2601.05647",
"authors": [
"Xinyue Wang",
"Stephen Wang",
"Biwei Huang"
],
"categories": [
"cs.LG",
"cs.AI"
],
"license": "http://creativecommons.org/licenses/by/4.0/",
"title": "Transformer Is Inherently a Causal Learner",
"url": "https://arxiv.org/abs/2601.05647",
"version": "v1"
},
"schema_id": "dorsal/arxiv",
"source": {
"execution_id": "ad03e806-ac98-4590-845e-0594211d4b85",
"id": "arXiv Dataset",
"type": "Model",
"variant": "snapshot-2026-01-17",
"version": "0.1.0"
},
"user_id": 1000002
}