用混合模型从复合材料图像直接预测应力应变场,速度比传统方法快上百倍。
Hybrid Unet-Transformer Model for Generating Stress and Strain Fields from Composite Geometrics

- 结合UNet与Transformer,从几何图像直接生成多种力学场分布。
- 周期性结构预测准确率高达R2=0.9991,多数子数据集误差低于0.05。
- 模型自动关注关键力学区域,无需标注即可学习物理规律,适合材料设计者。
精确预测分层复合材料微结构中的应力与应变场对物理引导的材料设计至关重要,但传统有限元方法(FEM)计算成本高,单次评估需数分钟至数日。本文提出一种混合UNet-Transformer架构,可直接从复合材料微结构图像预测复杂力学场分布,作为FEM的高效替代方案,覆盖十类不同应力/应变场类型,涵盖正方、六边形、三角形等多相复合结构配置、多种边界条件及高分辨率几何。结果表明,该模型在多数子数据集上表现优异,周期性结构预测中最高达R2=0.9991,SSIM=0.9936,MAE=0.0050(边界条件子集与三角形结构子集)。在八个评估子数据集中的六个,归一化[0,1]像素尺度下MAE低于0.05。通过Grad-CAM和Grad-CAM++分析编码器注意力,证实模型自发形成具有物理意义的内部表征,能聚焦于相界、杆件连接处、压头接触区等关键区域,无须显式结构监督。在稀疏软相包含的不规则正方网格结构上性能下降,如S11正应力子集的R2=0.7735,SSIM=0.7126,与平滑损失图像转换模型难以捕捉应力突变的已知局限一致。
原文摘要 · Abstract (English)
Accurate prediction of stress and strain fields in hierarchical composite microstructures is critical for physics-informed material design, yet conventional finite element method (FEM) simulations are computationally prohibitive at scale, requiring minutes to days per evaluation. In this work, we propose a hybrid UNet-Transformer architecture that predicts complex mechanical field distributions directly from composite microstructure geometry images, serving as an efficient surrogate for FEM across ten distinct stress and strain field types spanning diverse two-phase composite configurations including square, hexagonal, and triangular tessellations, multiple boundary conditions, and high-resolution geometries. Results demonstrate that the proposed architecture achieves strong predictive performance across the majority of subdatasets, with peak accuracy on periodic tessellation geometries reaching R2=0.9991, SSIM=0.9936, and MAE=0.0050 on the boundary condition subdataset and the triangular tessellation subdataset respectively. Across six of the eight evaluated subdatasets, MAE remains below 0.05 on the normalized [0,1] pixel scale. Encoder attention analysis via Grad-CAM and Grad-CAM++ confirms that the model develops physically meaningful internal representations, localizing attention at mechanically critical regions including phase boundaries, ligament junctions, and indenter contact zones without explicit structural supervision. Performance degrades on irregular square-grid geometries with sparse soft-phase inclusions, with the S11 normal stress subdataset yielding R2=0.7735 and SSIM=0.7126, consistent with the known limitation of smooth-loss image translation models in reproducing sharp stress discontinuities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。