Knowledge-guided machine learning for shape parameter and drag coefficient prediction of falling snowflakes
Fitzgerald, N., Hudson, C., Wshah, S., Heymsfield, A. J., Marshall, J. S.. (2025). Knowledge-guided machine learning for shape parameter and drag coefficient prediction of falling snowflakes. Artificial Intelligence for the Earth Systems, doi:https://doi.org/10.1175/aies-d-24-0059.1
| Title | Knowledge-guided machine learning for shape parameter and drag coefficient prediction of falling snowflakes |
|---|---|
| Genre | Article |
| Author(s) | N. Fitzgerald, C. Hudson, S. Wshah, Andrew J. Heymsfield, J. S. Marshall |
| Abstract | A novel knowledge-guided convolutional neural network (KGCNN) is presented in this work for the prediction of various three-dimensional shape parameters, drag coefficient, mass, and density of falling snowflakes. To compensate for the lack of extensive data on real snow, the neural network model is pretrained on images of synthetic snowflakes that are geometrically similar to real snowflakes. The model is then fine-tuned on real snowflake images from available datasets for effective drag coefficient prediction. Existing drag coefficient correlations are integrated into the final layer of the KGCNN to remove the burden of learning the relationship between the Reynolds number and drag coefficient from the model. The shape parameter outputs are also regularized by custom knowledge-guided loss functions to ensure physical interpretability and allow for simultaneous prediction of particle volume and density. Pretraining and custom loss functions were found to reduce normalized root-mean-square error (NRMSE) on mass by 11.8%. Integration of drag coefficient correlations reduced mass NRMSE by 30.1% over similar models directly predicting drag coefficient and outperformed existing physical correlations for snowflake drag coefficient. Predictions of shape parameters, volume, and density by the KGCNN were found to be consistent with experimental values. |
| Publication Title | Artificial Intelligence for the Earth Systems |
| Publication Date | Oct 1, 2025 |
| Publisher's Version of Record | https://doi.org/10.1175/aies-d-24-0059.1 |
| OpenSky Citable URL | https://n2t.net/ark:/85065/d72z1b2t |
| OpenSky Listing | View on OpenSky |
| MMM Affiliations | DPM |