Zero-shot Safety Prediction for Autonomous Robots with Foundation World Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Mao, Zhenjiang, Dai, Siqi, Geng, Yuang, Ruchkin, Ivan
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910433051410432
author Mao, Zhenjiang
Dai, Siqi
Geng, Yuang
Ruchkin, Ivan
author_facet Mao, Zhenjiang
Dai, Siqi
Geng, Yuang
Ruchkin, Ivan
contents A world model creates a surrogate world to train a controller and predict safety violations by learning the internal dynamic model of systems. However, the existing world models rely solely on statistical learning of how observations change in response to actions, lacking precise quantification of how accurate the surrogate dynamics are, which poses a significant challenge in safety-critical systems. To address this challenge, we propose foundation world models that embed observations into meaningful and causally latent representations. This enables the surrogate dynamics to directly predict causal future states by leveraging a training-free large language model. In two common benchmarks, this novel model outperforms standard world models in the safety prediction task and has a performance comparable to supervised learning despite not using any data. We evaluate its performance with a more specialized and system-relevant metric by comparing estimated states instead of aggregating observation-wide error.
format Preprint
id arxiv_https___arxiv_org_abs_2404_00462
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Zero-shot Safety Prediction for Autonomous Robots with Foundation World Models
Mao, Zhenjiang
Dai, Siqi
Geng, Yuang
Ruchkin, Ivan
Machine Learning
Robotics
A world model creates a surrogate world to train a controller and predict safety violations by learning the internal dynamic model of systems. However, the existing world models rely solely on statistical learning of how observations change in response to actions, lacking precise quantification of how accurate the surrogate dynamics are, which poses a significant challenge in safety-critical systems. To address this challenge, we propose foundation world models that embed observations into meaningful and causally latent representations. This enables the surrogate dynamics to directly predict causal future states by leveraging a training-free large language model. In two common benchmarks, this novel model outperforms standard world models in the safety prediction task and has a performance comparable to supervised learning despite not using any data. We evaluate its performance with a more specialized and system-relevant metric by comparing estimated states instead of aggregating observation-wide error.
title Zero-shot Safety Prediction for Autonomous Robots with Foundation World Models
topic Machine Learning
Robotics
url https://arxiv.org/abs/2404.00462