Confounded Causal Imitation Learning with Instrumental Variables

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zeng, Yan, Nie, Shenglan, Xie, Feng, Huang, Libo, Wu, Peng, Geng, Zhi
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909702023020544
author Zeng, Yan
Nie, Shenglan
Xie, Feng
Huang, Libo
Wu, Peng
Geng, Zhi
author_facet Zeng, Yan
Nie, Shenglan
Xie, Feng
Huang, Libo
Wu, Peng
Geng, Zhi
contents Imitation learning from demonstrations usually suffers from the confounding effects of unmeasured variables (i.e., unmeasured confounders) on the states and actions. If ignoring them, a biased estimation of the policy would be entailed. To break up this confounding gap, in this paper, we take the best of the strong power of instrumental variables (IV) and propose a Confounded Causal Imitation Learning (C2L) model. This model accommodates confounders that influence actions across multiple timesteps, rather than being restricted to immediate temporal dependencies. We develop a two-stage imitation learning framework for valid IV identification and policy optimization. In particular, in the first stage, we construct a testing criterion based on the defined pseudo-variable, with which we achieve identifying a valid IV for the C2L models. Such a criterion entails the sufficient and necessary identifiability conditions for IV validity. In the second stage, with the identified IV, we propose two candidate policy learning approaches: one is based on a simulator, while the other is offline. Extensive experiments verified the effectiveness of identifying the valid IV as well as learning the policy.
format Preprint
id arxiv_https___arxiv_org_abs_2507_17309
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Confounded Causal Imitation Learning with Instrumental Variables
Zeng, Yan
Nie, Shenglan
Xie, Feng
Huang, Libo
Wu, Peng
Geng, Zhi
Machine Learning
Artificial Intelligence
Imitation learning from demonstrations usually suffers from the confounding effects of unmeasured variables (i.e., unmeasured confounders) on the states and actions. If ignoring them, a biased estimation of the policy would be entailed. To break up this confounding gap, in this paper, we take the best of the strong power of instrumental variables (IV) and propose a Confounded Causal Imitation Learning (C2L) model. This model accommodates confounders that influence actions across multiple timesteps, rather than being restricted to immediate temporal dependencies. We develop a two-stage imitation learning framework for valid IV identification and policy optimization. In particular, in the first stage, we construct a testing criterion based on the defined pseudo-variable, with which we achieve identifying a valid IV for the C2L models. Such a criterion entails the sufficient and necessary identifiability conditions for IV validity. In the second stage, with the identified IV, we propose two candidate policy learning approaches: one is based on a simulator, while the other is offline. Extensive experiments verified the effectiveness of identifying the valid IV as well as learning the policy.
title Confounded Causal Imitation Learning with Instrumental Variables
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2507.17309