[发明专利]基于Swin-Transfomer的结构拓扑优化方法

说明书

本发明针对现有技术的局限性，提出了一种基于Swin‑Transfomer的结构拓扑优化方法，本发明提出了一种基于Swin‑Transfomer的神经网络架构，该架构从初始应力中学习，该模型利用结构的初始主应力矩阵代替结构的荷载情况和边界条件作为输入，可以基于相对较少的样本产生高精度的拓扑预测结果；还可以通过反向传播进行自我调整，以达到更好地预测优化结果。

技术领域

本发明涉及计算机深度学习的技术领域，具体地，涉及深度学习方法在结构拓扑优化的应用；更具体的，涉及一种基于Swin-Transfomer的结构拓扑优化方法。

背景技术

结构拓扑优化是为了在给定的设计域、载荷和边界条件下获得最佳的材料分布，以实现最佳的结构性能。拓扑优化可以方便工程师提出高效且新颖的概念设计方案；因此，它在工程领域引起了广泛的关注，并成为最近优化的重点。在优化问题中，涉及的计算通常涉及许多迭代过程。每个迭代过程通常需要对整个结构进行有限元计算。是一个计算昂贵、占用内存和耗时的过程。这些缺点限制了设计空间探索的机会，禁止了交互设计。

计算机辅助工程可以从几何深度学习革命中受益，类似于深度学习革命计算机视觉的方式。与传统的拓扑优化方法相比，深度学习方法无需进行重复迭代，能快速准确地优化复杂形状。现有的深度学习方法通常是使用卷积神经网络(CNN)构建模型，捕捉应力矩阵的信息，对结构进行拓扑优化。如公告日为2022.08.05的中国发明专利：一种基于卷积神经网络的结构拓扑构型预测方法中所示，包括：构建训练数据集，其中的每一个训练样本包含结构的最优拓扑构型和对应的多通道张量；结构预设体积分数、载荷位置和载荷方向中任一项发生变化时，至少一个通道会发生相应的变化；建立网络模型，其中，多个编码模块用于提取多通道张量的多个不同尺度的特征图谱；SE-ResNet模块用于获取特征图谱中各通道的注意力权重并融合至特征图谱；解码模块以前一模块及对应编码模块输出的特征图谱为输入，用于将特征图谱扩张到目标尺寸；利用训练数据集对网络模型进行训练，在训练结束后，得到结构最优拓扑构型预测模型。

但卷积神经网络(CNN)存在局限性，其无法建立相距较远像素点之间的联系，即无法获得全局感受野，另外，即使通过加深网络层数，相聚越远的像素点之间的可行信息传递路径越少，无法做到平均的全局感受野。

发明内容

针对现有技术的局限，本发明提出一种基于Swin-Transfomer的结构拓扑优化方法，本发明采用的技术方案是：

一种基于Swin-Transfomer架构的结构拓扑优化预测模型的建立方法，包括以下步骤：

S1，获取结构样本；对所述结构样本进行拓扑优化，获得初始应力矩阵以及样本优化结果；所述初始主应力矩阵为在前述拓扑优化过程的第一次迭代中通过有限元分析获得的主应力矩阵；

S2，构建基于Swin-Transfomer架构的模型；

S3，以所述初始主应力矩阵表示的图片作为训练数据，以所述样本优化结果作为训练标签，对所述步骤S2构建的模型进行训练，获得结构拓扑优化预测模型。

相较于现有技术，本发明提出了一种基于Swin-Transfomer的神经网络架构，该架构从初始应力中学习，该模型利用结构的初始主应力矩阵代替结构的荷载情况和边界条件作为输入，可以基于相对较少的样本产生高精度的拓扑预测结果；还可以通过反向传播进行自我调整，以达到更好地预测优化结果。

作为一种优选方案，在所述步骤S1中，通过最小化目标函数对所述结构样本进行拓扑优化：

其中，ρ为结构密度变量向量；U为全局位移，上标T表示转置；F为力向量；K为刚度矩阵；u_i为元素位移向量；k_i为元素刚度矩阵；N为元素总个数；θ为体积系数；V(ρ)代表密度变量ρ的有效体积；V₀是密度变量为1时的体积，表示设计域的大小；p是惩罚因子。