arXiv AI

KPPax: preprocess differentiable kinetic ODE for machine learning in atmospheric chemistry

KPPax is a new kinetic preprocessor that converts atmospheric chemistry mechanisms written in the Kinetic PreProcessor (KPP) language into differentiable, just‑in‑time compiled, vectorisable code in JAX. It evaluates rate laws in log space to maintain differentiability and integrates stiff systems using implicit solvers with forward‑ and reverse‑mode adjoints from the diffrax library, enabling exact sensitivities without hand‑derived adjoint code. Demonstrations with the POLLU benchmark show that KPPax can predict which rate coefficients are identifiable and improve neural‑network corrections of emission inventories, reducing errors by up to a factor of five.

arXiv Machine Learning
Sep 2

Accelerating Chemical Kinetics for Exoplanet Atmospheres using Neural Networks

The paper introduces a neural‑network based local‑box chemical kinetics solver for exoplanet atmospheres, employing a residual flow‑map architecture. It achieves microsecond‑scale inference with percent‑level accuracy across a wide range of temperatures, pressures, time steps, and compositional variations, outperforming other machine‑learning models and handling the extreme stiffness of atmospheric chemistry. The surrogate model offers a flexible, efficient alternative to classical solvers for state‑to‑state flow‑map problems in numerical simulations.

By Isaac Malsky, Xi Zhang, Tiffany Kataria, Matthew Graham, Ziyu Huang, Boris Bonev, Shang-Min Tsai, Elspeth K. H. Lee
arXiv Machine Learning
Aug 4

A Physics-Chemistry-Informed Neural Network (PCINN) for Real-Time Spatial-ALD Coverage Prediction and Reliable Kinetics Inversion

arXiv:2608. 00212v1 Announce Type: new Abstract: Spatial atomic layer deposition (SALD) is a leading atmospheric-pressure, high-throughput route to industrial ALD, but design and control are limited by the cost of predicting surface coverage: high-fidelity CFD is far too slow for operating-window scans, while analytic models miss transport modulation such as the gas curtain.

By Ning Hu, Chang Liu, Yunlei Jiang, Yuan Dong
Hugging Face Trending Papers
Aug 5

DASyR-LLM: Domain-Aware Symbolic Regression with LLMs for Kinetic Model Discovery

Kinetic model discovery is a central challenge in chemical engineering, as accurate rate expressions are essential for understanding and controlling chemical and biological processes. Symbolic regression (SR) has emerged as a powerful data-driven approach for identifying interpretable kinetic models, but usually operates without domain knowledge, often exploring physicochemically implausible models.

arXiv Machine Learning
Oct 8

Physics-Aligned Electronic Ground-State Learning Improves Generalization

The paper introduces Physics‑Aligned Electronic Ground‑State Models (GSMs) that sit between machine‑learned interatomic potentials and Kohn‑Sham density functional theory in terms of computational cost and accuracy. By aligning GSM learning objectives with KS‑DFT equations and enforcing physical constraints, the authors develop new training techniques—OrthoNormal‑Loss (ON‑Loss), Grassmann Restricted Occupied‑Orbital Training (GROOT), and Residual Optimal‑gauge Conditioning‑aware KS‑Eq. Training (ROCKET)—which dramatically reduce energy and force errors in size‑extrapolation tests from QM9 to QM40. The models also achieve near‑chemical‑accuracy predictions on reactive chemistry benchmarks and demonstrate efficient label‑free self‑consistency fine‑tuning. whyItMatters":"The work advances the cost‑accuracy frontier for electronic structure predictions, enabling more reliable out‑of‑distribution generalization in drug and material discovery."

By Eike S. Eberhard, Xaver Kainz, Viktor Kotsev, Abdulrahman Aldossary, Stephan G\"unnemann
arXiv AI
Sep 17

BOOM: Benchmarking Out-Of-distribution Molecular Property Predictions of Machine Learning Models

BOOM is a new benchmark for evaluating out‑of‑distribution (OOD) molecular property predictions in machine learning. It provides chemically‑informed tests across common property prediction tasks and assesses over 150 model‑task combinations. The study shows that current models, including chemical foundation models, struggle to generalize OOD, with the best model still exhibiting three times higher error than in‑distribution predictions.

By Evan R. Antoniuk, Shehtab Zaman, Tal Ben-Nun, Peggy Li, James Diffenderfer, Busra Sahin, Obadiah Smolenski, Everett Grethel, Tim Hsu, Anna M. Hiszpanski, Kenneth Chiu, Bhavya Kailkhura, Brian Van Essen
arXiv Machine Learning
Sep 22

Role-Aware Morgan Fingerprints for Reaction Yield Prediction

The paper introduces MFP, a reaction yield prediction method that uses role-aware Morgan fingerprints. It computes count-based circular fingerprints for each reaction component, aggregates them by chemical role, and combines them with transformation-sensitive difference features into a fixed-length descriptor for a feed-forward neural regressor. On the Suzuki‑Miyaura and Buchwald‑Hartwig benchmarks, MFP achieves R² scores of 0.878 and 0.969 respectively, while training an order of magnitude faster than graph or Transformer-based alternatives.

By Chinmay Mirji, Prashant Shekhar, Foram Madiyar, Hao Peng
arXiv Machine Learning
Sep 4

Differentiable Hybrid Modelling for Learning and Optimising Chemical Transport Processes from Experimental Data

The paper presents a differentiable hybrid modelling framework that combines a JAX finite volume population balance solver with learnable neural network components. This approach learns constitutive laws and initial conditions directly from experimental data, improving the fidelity of transport models in chemical engineering. The framework’s differentiability also enables optimisation of experimental settings for desired process outcomes.

By Arthur Jessop, Mohammed Alsubeihi, Ben Moseley, Ashwin Kumar Rajagopalan