Saved in:
Bibliographic Details
Main Authors: Vowels, Matthew J., Rochat, Mathieu, Akbari, Sina
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.14485
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908789118074880
author Vowels, Matthew J.
Rochat, Mathieu
Akbari, Sina
author_facet Vowels, Matthew J.
Rochat, Mathieu
Akbari, Sina
contents Artificial Neural Networks (ANNs), including fully-connected networks and transformers, are highly flexible and powerful function approximators, widely applied in fields like computer vision and natural language processing. However, their inability to inherently respect causal structures can limit their robustness, making them vulnerable to covariate shift and difficult to interpret/explain. This poses significant challenges for their reliability in real-world applications. In this paper, we introduce Causal Transformers (CaTs), a general model class designed to operate under predefined causal constraints, as specified by a Directed Acyclic Graph (DAG). CaTs retain the powerful function approximation abilities of traditional neural networks while adhering to the underlying structural constraints, improving robustness, reliability, and interpretability at inference time. This approach opens new avenues for deploying neural networks in more demanding, real-world scenarios where robustness and explainability is critical.
format Preprint
id arxiv_https___arxiv_org_abs_2410_14485
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle CaTs and DAGs: Integrating Directed Acyclic Graphs with Transformers for Causally Constrained Predictions
Vowels, Matthew J.
Rochat, Mathieu
Akbari, Sina
Machine Learning
Artificial Neural Networks (ANNs), including fully-connected networks and transformers, are highly flexible and powerful function approximators, widely applied in fields like computer vision and natural language processing. However, their inability to inherently respect causal structures can limit their robustness, making them vulnerable to covariate shift and difficult to interpret/explain. This poses significant challenges for their reliability in real-world applications. In this paper, we introduce Causal Transformers (CaTs), a general model class designed to operate under predefined causal constraints, as specified by a Directed Acyclic Graph (DAG). CaTs retain the powerful function approximation abilities of traditional neural networks while adhering to the underlying structural constraints, improving robustness, reliability, and interpretability at inference time. This approach opens new avenues for deploying neural networks in more demanding, real-world scenarios where robustness and explainability is critical.
title CaTs and DAGs: Integrating Directed Acyclic Graphs with Transformers for Causally Constrained Predictions
topic Machine Learning
url https://arxiv.org/abs/2410.14485