Causal Imitation Learning under Temporally Correlated Noise
Gokul Swamy, Sanjiban Choudhury, Drew Bagnell, Steven Wu
Abstract
We develop algorithms for imitation learning from policy data that was corrupted by temporally correlated noise in expert actions. When noise affects multiple timesteps of recorded data, it can manifest as spurious correlations between states and actions that a learner might latch on to, leading to poor policy performance. To break up these spurious correlations, we apply modern variants of the instrumental variable regression (IVR) technique of econometrics, enabling us to recover the underlying policy without requiring access to an interactive expert. In particular, we present two techniques, one of a generative-modeling flavor (DoubIL) that can utilize access to a simulator, and one of a game-theoretic flavor (ResiduIL) that can be run entirely offline. We find both of our algorithms compare favorably to behavioral cloning on simulated control tasks.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 07280a32-b72d-4232-b5c7-91c63d86c1f2Cited by top-tier papers10
- QueST: Self-Supervised Skill Abstractions for Learning Continuous ControlAtharva Mete, Haotian Xue, Albert Wilcox, Yongxin Chen et al.NeurIPS 2024 · 76 citations
- Sequence Model Imitation Learning with Unobserved ContextsGokul Swamy, Sanjiban Choudhury, J. Andrew Bagnell, Zhiwei Steven WuNeurIPS 2022 · 39 citations
- Multi-Agent Imitation Learning: Value is Easy, Regret is HardJingwu Tang, Gokul Swamy, Fei Fang, Zhiwei Steven WuNeurIPS 2024 · 13 citations
- Causal Imitation for Markov Decision Processes: a Partial Identification ApproachKangrui Ruan, Junzhe Zhang, Xuan Di, Elias BareinboimNeurIPS 2024 · 12 citations
- Causal Imitability Under Context-Specific Independence RelationsFateme Jamshidi, Sina Akbari, Negar KiyavashNeurIPS 2023 · 8 citations
Builds on5
- Exploring the Limitations of Behavior Cloning for Autonomous DrivingFelipe Codevilla, Eder Santana, Antonio M. López, Adrien GaidonICCV 2019 · 666 citations
- Minimax Estimation of Conditional Moment ModelsNishanth Dikkala, Greg Lewis, Lester Mackey, Vasilis SyrgkanisNeurIPS 2020 · 125 citations
- Of Moments and Matching: A Game-Theoretic Framework for Closing the Imitation GapGokul Swamy, Sanjiban Choudhury, J. Andrew Bagnell, Steven WuICML 2021 · 90 citations
- Causal Imitation Learning With Unobserved ConfoundersJunzhe Zhang, Daniel Kumor, Elias BareinboimNeurIPS 2020 · 86 citations
- Sequential Causal Imitation Learning with Unobserved ConfoundersDaniel Kumor, Junzhe Zhang, Elias BareinboimNeurIPS 2021 · 53 citations
Related papers
- Causal Imitation Learning under Expert-Observable and Expert-Unobservable ConfoundingDaqian Shao, Thomas Kleine Buening, Marta KwiatkowskaICLR 2026 · 1 citation
- Learning Decision Policies with Instrumental Variables through Double Machine LearningDaqian Shao, Ashkan Soleymani, Francesco Quinzan, Marta KwiatkowskaICML 2024 · 4 citations
- Fighting Copycat Agents in Behavioral Cloning from Observation HistoriesChuan Wen, Jierui Lin, Trevor Darrell, Dinesh Jayaraman et al.NeurIPS 2020 · 103 citations
- Learning Human Driving Behaviors with Sequential Causal Imitation LearningKangrui Ruan, Xuan DiAAAI 2022 · 28 citations
- An Instrumental Variable Approach to Confounded Off-Policy EvaluationYang Xu, Jin Zhu, Chengchun Shi, Shikai Luo et al.ICML 2023 · 24 citations
