arXiv:2508.19752cs.LG2025-08

用3D扩散模型快速生成大规模真实颗粒系统,初始化速度提升百倍。

Fast 3D Diffusion for Scalable Granular Media Synthesis

  • 分两阶段生成:先无条件扩散生成独立体素块,再用3D修复拼接无缝融合
  • 可生成超20万颗粒的铁路道砟和月壤系统,模拟时间从数天缩至数小时
  • 输出兼容现有DEM流程,支持凸与非凸颗粒,适合大尺度颗粒模拟场景

颗粒介质的离散元方法(DEM)模拟计算成本高,尤其在初始阶段因大位移和高动能而耗时。本文提出一种基于3D扩散模型的生成新范式,可直接合成任意规模、力学上合理的颗粒结构。采用两阶段流程:首先用无条件扩散模型生成独立的3D体素网格;其次利用源自2D的掩码输入与重绘策略,将这些网格无缝拼接。修复模型通过学习邻近区域上下文,生成平滑融合的新区域。两个模型均在小尺度DEM仿真生成的二值化3D占据网格数据库上训练,输出体素数线性扩展。原需数日的模拟现可在数小时内完成,实际支持超过20万颗粒的仿真。该流程完全兼容现有DEM工作流,将扩散生成的体素网格后处理为符合DEM要求的粒子网格。生成结果在关键粒度指标上与原始DEM一致,适用于多种颗粒介质应用,可生成凸形与非凸形颗粒。在铁路道砟与月壤两个实例中展示,重新定义了颗粒系统初始化方式,实现传统DEM无法企及的大规模生成。

原文摘要 · Abstract (English)

Discrete Element Method (DEM) simulations of granular media are computationally intensive, particularly during initialization phases dominated by large displacements and kinetic energy. This paper presents a novel generative pipeline based on 3D diffusion models that directly synthesizes arbitrarily large granular assemblies in mechanically realistic configurations. The approach employs a two-stage pipeline. First, an unconditional diffusion model generates independent 3D voxel grids representing granular media; second, a 3D inpainting model, adapted from 2D techniques using masked inputs and repainting strategies, seamlessly stitches these grids together. The inpainting model uses the outputs of the unconditional diffusion model to learn from the context of adjacent generations and creates new regions that blend smoothly into the context region. Both models are trained on binarized 3D occupancy grids derived from a database of small-scale DEM simulations, scaling linearly with the number of output voxels. Simulations that spanned over days can now run in hours, practically enabling simulations containing more than 200k ballast particles. The pipeline remains fully compatible with existing DEM workflows as it post-processes the diffusion generated voxel grids into DEM compatible particle meshes. Being mechanically consistent on key granulometry metrics with the original DEM simulations, the pipeline is also compatible with many other applications in the field of granular media, with capability of generating both convex and non-convex particles. Showcased on two examples (railway ballast and lunar regolith), the pipeline reimagines the way initialization of granular media simulations is performed, enabling scales of generation previously unattainable with traditional DEM simulations.

3D扩散颗粒模拟高效生成DEM加速

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。