Learning Aerosol Coagulation Dynamics in Latent Space with a Scale-Covariant Neural ODE

Particle-resolved aerosol models preserve the joint size-composition structure that governs cloud activation, optical properties, and freezing, but their computational cost limits their use in large-scale atmospheric models. We extend AeroMELD from a compact representation of aerosol populations to a prognostic model of coagulation. The resulting AeroMELD-Coag advances nine learned coordinates for population shape and one for total particle number through a scale-covariant neural ordinary differential equation that incorporates the known concentration dependence of coagulation. Across 2,000 held-out trajectories spanning 48 h, median symmetric errors are 3.5% for latent shape and 0.52% for total number. This ten-coordinate state also retains the evolving size-resolved composition and associated cloud-condensation-nuclei (CCN) activation, optical properties, and frozen fraction. Compared with the evaluated 20-bin sectional model with 320 prognostic coordinates, AeroMELD-Coag lowers pooled mean CCN activation error from 1.98% to 1.38% and frozen-fraction error from 1.72% to 0.56%. In a matched integration benchmark, GPU integration time per trajectory, amortized over a batch, is more than three orders of magnitude lower than for both single-core CPU references. These coagulation results provide a proof of concept for advancing aerosol microphysics efficiently in a compact learned state while retaining information about aerosol mixing state. They establish a foundation for future large-scale atmospheric models to represent aerosol microphysics with greater mixing-state detail within practical computational budgets.