Latent Field Discovery In Interacting Dynamical Systems With Neural Fields

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Kofinas, Miltiadis, Bekkers, Erik J., Nagaraja, Naveen Shankar, Gavves, Efstratios
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914721362345984
author Kofinas, Miltiadis
Bekkers, Erik J.
Nagaraja, Naveen Shankar
Gavves, Efstratios
author_facet Kofinas, Miltiadis
Bekkers, Erik J.
Nagaraja, Naveen Shankar
Gavves, Efstratios
contents Systems of interacting objects often evolve under the influence of field effects that govern their dynamics, yet previous works have abstracted away from such effects, and assume that systems evolve in a vacuum. In this work, we focus on discovering these fields, and infer them from the observed dynamics alone, without directly observing them. We theorize the presence of latent force fields, and propose neural fields to learn them. Since the observed dynamics constitute the net effect of local object interactions and global field effects, recently popularized equivariant networks are inapplicable, as they fail to capture global information. To address this, we propose to disentangle local object interactions -- which are $\mathrm{SE}(n)$ equivariant and depend on relative states -- from external global field effects -- which depend on absolute states. We model interactions with equivariant graph networks, and combine them with neural fields in a novel graph network that integrates field forces. Our experiments show that we can accurately discover the underlying fields in charged particles settings, traffic scenes, and gravitational n-body problems, and effectively use them to learn the system and forecast future trajectories.
format Preprint
id arxiv_https___arxiv_org_abs_2310_20679
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Latent Field Discovery In Interacting Dynamical Systems With Neural Fields
Kofinas, Miltiadis
Bekkers, Erik J.
Nagaraja, Naveen Shankar
Gavves, Efstratios
Machine Learning
Systems of interacting objects often evolve under the influence of field effects that govern their dynamics, yet previous works have abstracted away from such effects, and assume that systems evolve in a vacuum. In this work, we focus on discovering these fields, and infer them from the observed dynamics alone, without directly observing them. We theorize the presence of latent force fields, and propose neural fields to learn them. Since the observed dynamics constitute the net effect of local object interactions and global field effects, recently popularized equivariant networks are inapplicable, as they fail to capture global information. To address this, we propose to disentangle local object interactions -- which are $\mathrm{SE}(n)$ equivariant and depend on relative states -- from external global field effects -- which depend on absolute states. We model interactions with equivariant graph networks, and combine them with neural fields in a novel graph network that integrates field forces. Our experiments show that we can accurately discover the underlying fields in charged particles settings, traffic scenes, and gravitational n-body problems, and effectively use them to learn the system and forecast future trajectories.
title Latent Field Discovery In Interacting Dynamical Systems With Neural Fields
topic Machine Learning
url https://arxiv.org/abs/2310.20679