atlas

Research map / Label-efficient and robust learning / Domain adaptation and generalization

Towards Cross Domain Generalization of Hamiltonian Representation via Meta Learning

ICLR 2024 · Label-efficient and robust learning › Domain adaptation and generalization · approach: Optimization-based domain generalization · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Meta-trains a GCN-based Hamiltonian network with MAML across six heterogeneous Hamiltonian systems so it can adapt to an unseen system type with few gradient steps.

Problem

Existing neural Hamiltonian and dynamics-learning methods are system-specific and generalize only across parameter environments of one system type, so they fail on systems with different functional-form Hamiltonians.

Approach

Apply MAML to a GCN Hamiltonian regressor trained over a distribution of different Hamiltonian system types (mass-spring, pendulum, Hénon-Heiles, magnetic-mirror, two-body, three-body), holding out one type for adaptation, and use CKA to argue the meta-trained model learns a shared Hamiltonian representation.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use meta-training instead of ordinary pre-trainingGNN Hamiltonian model · Held-out Hamiltonian systemsrelative errorHelped
Use meta-training instead of random initializationGNN Hamiltonian model · Held-out Hamiltonian systemsadaptation speedHelped
Use meta-training instead of pre-training or random initializationGNN Hamiltonian model · Mass-springtrajectory prediction qualityMixed
Use meta-training instead of pre-training or random initializationGNN Hamiltonian model · Pendulumtrajectory prediction qualityHelped
Use meta-training instead of pre-training or random initializationGNN Hamiltonian model · Hénon-Heiles and magnetic-mirrortrajectory prediction qualityHelped
Use meta-training instead of CoDACross-domain Hamiltonian dynamics model · Held-out Hamiltonian systemstrajectory prediction qualityHelped
Use the meta-trained model instead of DyAdMagnetic-mirror adaptation · Magnetic-mirroradaptation performanceHelped
Adapt a model meta-trained only on conservative systems to a dissipative systemMeta-trained Hamiltonian model · Damped mass-springadaptation performanceHurt
Adapt the meta-trained model to additional held-out system typesMeta-trained Hamiltonian model · 2D harmonic oscillator and Keplerrelative errorMixed
Use a symplectic integrator instead of LSODA (ablation)Adapted Hamiltonian models · Integrator ablation across adaptation tasksrelative errorHurt
Increase adaptation data size K (ablation)Meta-trained model · Held-out adaptation tasksrelative errorHelped
Use meta-training for adaptation with low K (ablation)Hamiltonian model · Held-out adaptation tasksrelative errorHelped

Surprising findings

Limitations

Experimental setup

Datasets: simulated trajectories for mass-spring, pendulum, Hénon-Heiles, magnetic-mirror, two-body, three-body systems

Benchmarks: mass-spring pendulum Hénon-Heiles magnetic-mirror two-body three-body 2D harmonic oscillator Kepler damped mass-spring

Metrics: relative error of predicted coordinates relative error of energy 1-CKA

Baselines: pre-trained model without meta learning randomly initialized model (HNN) CoDA DyAd

Builds on: MAML Hamiltonian Neural Network CoDA DyAd

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy