Learning Lifted Action Models from Unsupervised Visual Traces

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Xi, Kai, Gould, Stephen, Thiébaux, Sylvie
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909021072523264
author Xi, Kai
Gould, Stephen
Thiébaux, Sylvie
author_facet Xi, Kai
Gould, Stephen
Thiébaux, Sylvie
contents Efficient construction of models capturing the preconditions and effects of actions is essential for applying AI planning in real-world domains. Extensive prior work has explored learning such models from high-level descriptions of state and/or action sequences. In this paper, we tackle a more challenging setting: learning lifted action models from sequences of state images, without action observation. We propose a deep learning framework that jointly learns state prediction, action prediction, and a lifted action model. We also introduce a mixed-integer linear program (MILP) to prevent prediction collapse and self-reinforcing errors among predictions. The MILP takes the predicted states, actions, and action model over a subset of traces and solves for logically consistent states, actions, and action model that are as close as possible to the original predictions. Pseudo-labels extracted from the MILP solution are then used to guide further training. Experiments across multiple domains show that integrating MILP-based correction helps the model escape local optima and converge toward globally consistent solutions.
format Preprint
id arxiv_https___arxiv_org_abs_2604_19043
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Learning Lifted Action Models from Unsupervised Visual Traces
Xi, Kai
Gould, Stephen
Thiébaux, Sylvie
Artificial Intelligence
Efficient construction of models capturing the preconditions and effects of actions is essential for applying AI planning in real-world domains. Extensive prior work has explored learning such models from high-level descriptions of state and/or action sequences. In this paper, we tackle a more challenging setting: learning lifted action models from sequences of state images, without action observation. We propose a deep learning framework that jointly learns state prediction, action prediction, and a lifted action model. We also introduce a mixed-integer linear program (MILP) to prevent prediction collapse and self-reinforcing errors among predictions. The MILP takes the predicted states, actions, and action model over a subset of traces and solves for logically consistent states, actions, and action model that are as close as possible to the original predictions. Pseudo-labels extracted from the MILP solution are then used to guide further training. Experiments across multiple domains show that integrating MILP-based correction helps the model escape local optima and converge toward globally consistent solutions.
title Learning Lifted Action Models from Unsupervised Visual Traces
topic Artificial Intelligence
url https://arxiv.org/abs/2604.19043