Equivariant Neural Prediction of the Stokes Resistance Tensors for Arbitrary Microparticle Shapes
Abstract
The Stokes-flow hydrodynamics of an irregular microparticle is encoded by its grand resistance matrix, a 6x6 tensor whose translational and rotational blocks (A and C) govern settling, diffusion, and orientational transport. Empirical drag correlations compress these tensors to a single scalar, discarding drag's orientation dependence and the rotational response. We present an SO(3)-equivariant neural network that predicts the full symmetric positive-definite A and C blocks from a particle's spherical-harmonic surface representation, equivariant by construction to floating-point precision. Trained on 1.1x10^5 random shapes from near-spherical to very rough, with a sealed test set of 18,000, it achieves 1.4% and 2.7% mean relative error on the two blocks while evaluating each shape 4-5 orders of magnitude faster than the regularised-Stokeslet solver that generated its labels. A spectral-convergence study confirms the representation is faithful: truncating a shape at spherical-harmonic degree 15 changes its resistance by a median of 0.14% (translation) and 0.38% (rotation), while the training shapes, generated band-limited at degree 15, carry no truncation error. Across 1.16x10^6 orientation-sampled settling, rotation and diffusion events, the surrogate reveals lateral drift up to 11 deg, rotational misalignment up to 46 deg, and shape-induced diffusion spreads of 30% (translational) and 2.4x (rotational), all identically zero under any scalar or spheroid reduction. Even the orientation-averaged scalar friction the correlations target, accurate to 1.7-2.8% (median; 2.2-3.2% mean), carries no tensor orientation, whereas the surrogate reproduces that scalar to ~1% while supplying the full anisotropic tensors. A fast, equivariant tensor surrogate can replace both solver and scalar approximation in atmospheric dust transport, microplastic fate and colloidal Brownian dynamics.
Related
- The Stokes resistance of an arbitrary particle: a classification of hydrodynamic symmetries
- Haar-Bayesian Pure-State Prediction under Relative-Entropy Loss: Arbitrary-Effect Reduction and Global Optimality
- Twisted shock waves enable ballistic microparticle manipulation
- Orthogonal Inline Microscopy: Continuous Determination of Microparticle Size, Shape, and Concentration via contrast-gated Flow Imaging
- Proposal for matter-wave interferometry with a rare-earth-doped microparticle
BibTeX
@misc{pradeep2026equivariant,
title = {{Equivariant Neural Prediction of the Stokes Resistance Tensors for Arbitrary Microparticle Shapes}},
author = {Sanjay Pradeep and David Dandy and Candace S. J. Tsai and Jeff D. Eldredge},
howpublished = {arXiv},
year = {2026},
month = sep,
doi = {10.48550/arxiv.2609.23105},
eprint = {2609.23105},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2609.23105},
}
RIS
TY - UNPB
TI - Equivariant Neural Prediction of the Stokes Resistance Tensors for Arbitrary Microparticle Shapes
AU - Sanjay Pradeep
AU - David Dandy
AU - Candace S. J. Tsai
AU - Jeff D. Eldredge
PY - 2026
DA - 2026/09/19
PB - arXiv
DO - 10.48550/arxiv.2609.23105
UR - https://arxiv.org/abs/2609.23105
AB - The Stokes-flow hydrodynamics of an irregular microparticle is encoded by its grand resistance matrix, a 6x6 tensor whose translational and rotational blocks (A and C) govern settling, diffusion, and orientational transport. Empirical drag correlations compress these tensors to a single scalar, discarding drag's orientation dependence and the rotational response. We present an SO(3)-equivariant neural network that predicts the full symmetric positive-definite A and C blocks from a particle's spherical-harmonic surface representation, equivariant by construction to floating-point precision. Trained on 1.1x10^5 random shapes from near-spherical to very rough, with a sealed test set of 18,000, it achieves 1.4% and 2.7% mean relative error on the two blocks while evaluating each shape 4-5 orders of magnitude faster than the regularised-Stokeslet solver that generated its labels. A spectral-convergence study confirms the representation is faithful: truncating a shape at spherical-harmonic degree 15 changes its resistance by a median of 0.14% (translation) and 0.38% (rotation), while the training shapes, generated band-limited at degree 15, carry no truncation error. Across 1.16x10^6 orientation-sampled settling, rotation and diffusion events, the surrogate reveals lateral drift up to 11 deg, rotational misalignment up to 46 deg, and shape-induced diffusion spreads of 30% (translational) and 2.4x (rotational), all identically zero under any scalar or spheroid reduction. Even the orientation-averaged scalar friction the correlations target, accurate to 1.7-2.8% (median; 2.2-3.2% mean), carries no tensor orientation, whereas the surrogate reproduces that scalar to ~1% while supplying the full anisotropic tensors. A fast, equivariant tensor surrogate can replace both solver and scalar approximation in atmospheric dust transport, microplastic fate and colloidal Brownian dynamics.
ER -