Saved in:
Bibliographic Details
Main Authors: Lange, Moritz, Engelhardt, Raphael C., Konen, Wolfgang, Melnik, Andrew, Wiskott, Laurenz
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2507.04920
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915373820936192
author Lange, Moritz
Engelhardt, Raphael C.
Konen, Wolfgang
Melnik, Andrew
Wiskott, Laurenz
author_facet Lange, Moritz
Engelhardt, Raphael C.
Konen, Wolfgang
Melnik, Andrew
Wiskott, Laurenz
contents Reasoning about the trajectories of multiple, interacting objects is integral to physical reasoning tasks in machine learning. This involves conditions imposed on the objects at different time steps, for instance initial states or desired goal states. Existing approaches in physical reasoning generally rely on autoregressive modeling, which can only be conditioned on initial states, but not on later states. In fields such as planning for reinforcement learning, similar challenges are being addressed with denoising diffusion models. In this work, we propose an object-centric denoising diffusion model architecture for physical reasoning that is translation equivariant over time, permutation equivariant over objects, and can be conditioned on arbitrary time steps for arbitrary objects. We demonstrate how this model can solve tasks with multiple conditions and examine its performance when changing object numbers and trajectory lengths during inference.
format Preprint
id arxiv_https___arxiv_org_abs_2507_04920
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Object-centric Denoising Diffusion Models for Physical Reasoning
Lange, Moritz
Engelhardt, Raphael C.
Konen, Wolfgang
Melnik, Andrew
Wiskott, Laurenz
Machine Learning
Artificial Intelligence
Reasoning about the trajectories of multiple, interacting objects is integral to physical reasoning tasks in machine learning. This involves conditions imposed on the objects at different time steps, for instance initial states or desired goal states. Existing approaches in physical reasoning generally rely on autoregressive modeling, which can only be conditioned on initial states, but not on later states. In fields such as planning for reinforcement learning, similar challenges are being addressed with denoising diffusion models. In this work, we propose an object-centric denoising diffusion model architecture for physical reasoning that is translation equivariant over time, permutation equivariant over objects, and can be conditioned on arbitrary time steps for arbitrary objects. We demonstrate how this model can solve tasks with multiple conditions and examine its performance when changing object numbers and trajectory lengths during inference.
title Object-centric Denoising Diffusion Models for Physical Reasoning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2507.04920