Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces a differentiable hybrid modeling framework that combines a JAX finite volume population balance solver with neural network components to learn constitutive laws and initial conditions from experimental data in chemical transport processes. By addressing the limitations of traditional models, which often rely on fixed assumptions, the framework enables more accurate and flexible modeling that reflects real-world conditions. The authors demonstrate its effectiveness not only for improved modeling but also for optimizing experimental settings, showcasing its broad applicability in chemical engineering.
Traditional models can misrepresent reality, but this new framework learns from data to optimize chemical transport processes more accurately than ever before.
Reliable transport models are essential when modelling and optimising many chemical engineering processes, yet, most models assume hand-picked constitutive laws which may not reflect reality, and often assume initial conditions are known exactly. Both restrictions can significantly bias model predictions and lead to systematic error when used in predictive and control settings. Black-box neural surrogate alternatives for modelling can better match real example data, but are confined to the task they were trained on and cannot be interrogated for physical consistency. Here we introduce a general-purpose differentiable hybrid modelling framework for transport processes, specifically for the case of population balance equations. Our framework integrates a JAX finite volume population balance solver with learnable neural network components which are trained to both discover constitutive laws and fit initial conditions from real experimental data, allowing us to better model real experimental transport systems. Furthermore, we use our framework for process optimisation, using its differentiability to allow us to direct optimising experimental settings for quantities of interest. This work highlights the huge potential of such differentiable hybrid modelling frameworks for learning and optimising any given chemical separation which involves mass, energy, and/or momentum transport.