arXiv Machine Learning

Adaptive Interaction Graphs for Particle Simulation

The paper introduces AdaptGNS, a particle simulation framework that dynamically adjusts the interaction graph based on per-particle uncertainty estimates. By training a variance head jointly with the acceleration head using a heteroscedastic Gaussian NLL loss, the model expands the neighborhood of high‑uncertainty particles, improving long‑horizon accuracy. Experiments show a strict Pareto improvement on the WaterDrop benchmark and a modest gain on Sand, suggesting adaptive graphs are especially beneficial in spatially complex regions.

arXiv Machine Learning
Sep 4

TRACE: Spatiotemporal Contact Memory Graph Network Simulator for Granular Dynamics

TRACE is a graph‑network simulator that stores granular contact history on edges using attention‑based message passing and a gated recurrent unit, with an edge‑identity dictionary to preserve memory as contacts change. It predicts normal and tangential forces while enforcing Coulomb friction and internal force balance, and is trained via single‑step pretraining followed by autoregressive fine‑tuning. On 2D and 3D granular column‑collapse benchmarks, TRACE achieves significantly lower long‑rollout position and deposit errors than existing simulators, uses fewer parameters, maintains near‑zero particle interpenetration, and outperforms the material point method by 12.2× (2D) and 8.9× (3D).

By Changjian Zhou, Negin Yousefpour, Jie Qi, Junfeng Fang, Guillermo A. Narsilio, Hans Petter Jostad
arXiv Machine Learning
Sep 10

Mind the Gap: Navigating Inference with Optimal Transport Maps

The paper introduces a model calibration method using optimal transport to address discrepancies between simulation and experimental data in high-dimensional machine learning applications. Applied to jet tagging in particle physics, the technique calibrates a 128‑dimensional latent representation from a general‑purpose classifier, ensuring downstream derived quantities are properly calibrated. This enables more reliable use of foundation models for jet flavor analysis in LHC experiments and offers a general framework for correcting high‑dimensional simulations across scientific fields.

By Malte Algren, Tobias Golling, Francesco Armando Di Bello, Christopher Pollard
arXiv Machine Learning
Sep 17

ADAPT: Lightweight, Long-Range Machine Learning Force Fields Without Graphs

The paper introduces ADAPT, a lightweight machine‑learning force field that replaces graph neural networks with a direct coordinates‑in‑space Transformer encoder to model all pairwise atomic interactions. Applied to silicon point defects, ADAPT reduces force prediction error by about 22% and energy prediction error by roughly 40% compared to a state‑of‑the‑art GNN model, while also cutting computational cost. This approach addresses common GNN issues such as oversmoothing, oversquashing, and poor long‑range interaction representation, which are especially problematic for point defect modeling.

By Evan Dramko, Yihuang Xiong, Yizhi Zhu, Geoffroy Hautier, Thomas Reps, Christopher Jermaine, Anastasios Kyrillidis