CN109783910B - 一种利用生成对抗网络加速的结构优化设计方法 - Google Patents
一种利用生成对抗网络加速的结构优化设计方法 Download PDFInfo
- Publication number
- CN109783910B CN109783910B CN201811637907.3A CN201811637907A CN109783910B CN 109783910 B CN109783910 B CN 109783910B CN 201811637907 A CN201811637907 A CN 201811637907A CN 109783910 B CN109783910 B CN 109783910B
- Authority
- CN
- China
- Prior art keywords
- generator
- training
- discriminator
- network
- model
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Active
Links
Images
Landscapes
- Image Analysis (AREA)
Abstract
本发明公开了一种应用生成对抗网络来加速结构优化设计的方法。该加速方法能够解决传统SIMP算法在结构优化设计中计算复杂度高的问题,该方法包括模型的获取和使用两个部分,其中模型获取的主要流程为:1、预先使用SIMP法生成少量(100组)优化过程图作为训练集和测试集;2、利用数据增强技术对数据集进行扩充;3、使用编码器‑解码器构建生成器;4、使用深度卷积网络构建判别器;5、使用变形的pix2pix模型来训练生成器‑判别器并展示训练结果和保存最终训练模型。使用最终模型进行结构优化设计时,首先通过SIMP法计算少量迭代步骤,将迭代结果输入最终模型进行计算,从而实现最终优化结构的快速计算。
Description
技术领域
本发明属于人工智能领域,具体涉及一种应用生成对抗网络加速的结构优化设计技术,在保证生成结果正确的前提下,最大程度降低计算复杂度的方法。
背景技术
为了在给定的负载情况、约束条件和性能指标下设计出高效的材料结构分布,国内外的研究者应用拓扑优化方法来进行优化设计;拓扑优化的基本思想是将寻求结构的最优拓扑问题转化为在给定的设计区域内寻求最优材料的分布问题,目前拓扑优化方法主要有SIMP算法、ESO算法、水平集法等,这些方法的计算量取决于网格的规模,随着网格的不断增大,计算量成倍增加,使得获取最优设计结果需要花费很长的时间。
随着深度学习的不断发展,其在各个领域的作用也越来越大;深度学习是机器学习研究中的一个新的领域,其动机在于建立、模拟人脑进行分析学习的神经网络,它模仿人脑的机制来解释数据;通过使用神经网络模型,使得利用前期优化过程图生成最终结果图变为可能。
在利用传统的SIMP算法进行结构优化设计时,将优化的中间过程图输入到已经训练好的神经网络模型中,可在几秒内获取最终优化结果图,从而解决了传统拓扑优化算法计算复杂度高、耗时长的问题。生成对抗网络是一种深度学习模型,是近年来无监督学习中最具前景的方法之一;相比于其他生成式模型,生成对抗网络具有产生的样本更加清晰、无需利用马尔科夫链反复采样,无需再学习中进行推断等优点,因此这里的模型我们选用生成对抗网络。
发明内容
为克服上述拓扑优化算法的缺点,本发明提供了一种应用生成对抗网络加速的结构优化设计方法,大幅度降低计算复杂度,减少计算开销。
一种应用生成对抗网络加速的结构优化设计方法,包括以下步骤:
第一步、使用SIMP算法准备数据
在所有拓扑优化方法中选用传统的SIMP算法来生成少量优化过程图,并按照9:1的比例将生成的过程图数据分为训练集和测试集;
第二步、利用数据增强技术对数据集进行扩充
随机获取数据集中的数据,并进行以下操作:1)产生随机数A,若其大于0.5则进行上下翻转;2)产生随机数B,若其大于0.5则进行左右翻转;3)产生随机数C,若其大于0.5则进行顺时针旋转;
第三步、使用编码器-解码器构建生成器
自动编码器算法将神经网络的隐含层看成是一个编码器和解码器,使用编码器学习输入数据的隐含特征,解码器可使用学习到的新特征重构出原始输入数据;具体操作如下:取出数据集中的数据并进行第二步中的操作,将处理后得到的数据喂入编码器网络进行特征提取并完成图像压缩任务;将编码器网络输出的feature map再送入解码器网络,可将被压缩的图像进行放大,在此过程中,将编码器和解码器过程中具有相同尺寸的featuremap连接到解码器中,以便充分利用编码器网络中的结构信息,保持特征的不变性,达到更好的重建效果;
第四步、使用深度卷积网络构建判别器
将一张生成图片或者真实的最终结果图像作为判别器的输入;经过多层带有步长的卷积层网络进行特征提取后,最终得到和标签大小相同的、内参仅为0或1的标签数据;
第五步、使用变形的pix2pix模型进行训练
生成器网络和判别器网络分别定义完成后,需要将两者连接起来循环训练;先固定生成器训练判别器,使用均方误差(MSE)方法计算损失值,该损失值来自两部分,一部分为输入真实结果图得到的输出值与全1标签之间的均方误差,另一部分为输入生成结果图得到的输出值与全0标签之间的均方误差,向着判别器总损失下降的方向进行训练以达到训练判别器的目的;再固定判别器训练生成器,使用均方误差(MSE)方法计算损失值,该损失值的获取较判别器损失值的获取复杂,应先构建一个Model,该Model的输入为生成器的两个输入,输出为生成器的输出和判别器生成的标签,其中判别器的输入为生成器的输出;则生成器的损失值即为判别器生成的标签和全1标签之间的均方误差,向着生成器损失下降的方向进行训练以达到训练生成器的目的;多次循环进行固定生成器训练判别器、固定判别器训练生成器的操作,直到判别器的错误率为50%左右为止;
第六步、使用最终模型
对变形的pix2pix模型进行训练,生成器的图像生成能力和判别器的判别能力将会得到提高,每训练10次后进行训练结果输出和当前生成器模型的保存;当训练总过程结束后,首先通过传统的SIMP算法计算少量迭代步骤,将迭代结果输入到最终保存的生成器模型中进行处理,可迅速得到最终生成结果图,从而实现最终优化结构的快速计算;
本发明的有益效果体现在:
由于本方法使用了生成对抗网络,所以可以在拥有少量SIMP算法生成的优化过程图的情况下训练好模型;通过使用训练好的生成器模型,在保证生成结果正确的前提下,可大幅度降低计算复杂度,减少计算开销。
附图说明
图1为本发明的流程示意图。
图2为实施例中构建生成器网络的示意图。
图3为实施例中构建判别器网络的示意图。
图4为实施例中构建变形的pix2pix模型的示意图。
具体实施方式
下面结合附图和实施例对本发明做详细说明;
本发明提供了一种应用生成对抗网络加速的结构优化设计方法;如图1所示,一种利用生成对抗网络加速的结构优化设计方法,包括以下步骤:步骤一:使用SIMP算法准备数据;步骤二:利用数据增强技术对数据集进行扩充;步骤三:使用编码器-解码器构建生成器;步骤四:使用深度卷积网络构建判别器;步骤五:使用变形的pix2pix模型进行训练;步骤六:使用最终模型;本发明具有准确生成优化结构、大幅度降低计算复杂度、减少计算开销的优点。
生成对抗网络加速的结构优化设计方法的步骤如下:
第一、使用SIMP算法准备数据:在所有拓扑优化方法中选用传统的SIMP算法来生成少量(100组)优化过程图并保存为npz格式;按照9:1的比例将这100组过程图分为训练集(90组)和测试集(10组);
第二、利用数据增强技术对数据集进行扩充:产生一个取值范围为[0,99]的随机数A,从第一步准备的数据中取出第A组;对取出的数据进行以下操作:1)产生一个随机数B,取值范围为[0,1],若B>=0.5则对该组数据中的每一张图像进行上下翻转,否则不进行任何操作;2)产生一个随机数C,取值范围为[0,1],若C>=0则对该组数据中的每一张图像进行左右翻转,否则不进行任何操作;3)产生一个随机数D,取值范围为[0,1],若D>=0则对该组数据中的每一张图像进行顺时针旋转,否则不进行任何操作;将进行以上三步操作得到的数据组作为真正的输入以扩大数据集;
第三、使用编码器-解码器构建生成器:
如图2所示,先进行第二步操作以获取可作为输入的数据组,产生一个随机数A,取值范围为[20,99],将该数据组中的第A张图像作为输入a,将该数据组中的第A张图像与第A-1张图像的灰度差作为输入b;将输入a和输入b从最后一个维度进行拼接后喂入生成器网络的编码器部分,在使用卷积层进行特征提取的同时,还使用最大池化层(MaxPooling2D)进行图像压缩;将经过编码器处理后得到的feature map喂入解码器部分,在使用卷积层进行特征提取的同时,还使用上采样(UpSampling2D)进行图像放大,同时将编码器中与解码器过程中相同尺寸的feature map连接到一起,以便充分利用编码器网络中结构信息,达到更好的重建效果;
第四、使用深度卷积网络构建判别器:
如图3所示,将一张生成图像或者真实的最终结果图像作为判别器的输入;经过4次步长为2的卷积层网络进行特征提取,最终得到大小为5*5、内参为0或1的标签数据;
第五、使用变形的pix2pix模型进行训练:
如图4所示,生成器网络和判别器网络需要连接起来循环训练;先固定生成器训练判别器,使用均方误差(MSE)方法计算损失值,该损失值来自两部分,一部分如图4(a)所示,将真实结果图输入判别器网络,得到的输出标签与大小为5*5的全1标签之间的均方误差,另一部分如图4(b)所示,将生成结果图输入判别器网络,得到的输出标签与大小为5*5的全0标签之间的均方误差,向着判别器总损失下降的方向进行训练以达到训练判别器的目的;再固定住判别器训练生成器,使用均方误差(MSE)方法计算损失值,该损失值的获取较判别器损失值的获取复杂,应先构建一个Model如图4(b)所示,该Model的输入为输入图片A和输入图片B,输出为生成结果图和该生成结果图输入判别器后生成的标签,则生成器的损失值即为生成的标签和大小为5*5的全1标签之间的均方误差,向着生成器损失下降的方向进行训练以达到训练生成器的目的;多次循环进行固定生成器训练判别器、固定判别器训练生成器的操作,直到判别器的错误率为50%左右为止;
第六、使用最终模型:
对变形的pix2pix模型(如图4所示)进行训练时,生成器的图像生成能力和判别器的判别能力将会得到提高,每经过10次训练后进行一次训练结果的输出和当前生成器模型的保存;当200次训练过程结束后,首先通过传统的SIMP算法计算20次迭代步骤,将迭代结果输入到最终保存的生成器模型中进行计算,可在3秒内得到最终生成结果图,从而实现最终优化结构的快速计算。
Claims (2)
1.一种利用生成对抗网络加速的结构优化设计方法,其特征在于,包括以下步骤:
第一步、使用SIMP算法准备数据
在所有拓扑优化方法中选用传统的SIMP算法来生成少量优化过程图,并按照9:1的比例将生成的过程图数据分为训练集和测试集;
第二步、利用数据增强技术对数据集进行扩充
随机获取数据集中的数据,并进行以下操作:1)产生随机数A,若其大于0.5则进行上下翻转;2)产生随机数B,若其大于0.5则进行左右翻转;3)产生随机数C,若其大于0.5则进行顺时针旋转;
第三步、使用编码器-解码器构建生成器
自动编码器算法将神经网络的隐含层看成是一个编码器和解码器,使用编码器学习输入数据的隐含特征,解码器可使用学习到的新特征重构出原始输入数据;
第四步、使用深度卷积网络构建判别器
将一张生成图片或者真实的最终结果图像作为判别器的输入;经过多层带有步长的卷积层网络进行特征提取后,最终生成和标签大小相同的、内参仅为0或1的数据;
第五步、使用变形的pix2pix模型进行训练
生成器网络和判别器网络分别定义完成后,需要将两者连接起来循环训练;先固定生成器训练判别器,使用均方误差MSE方法计算损失值,该损失值来自两部分,一部分为输入真实结果图得到的输出值与全1标签之间的均方误差,另一部分为输入生成结果图得到的输出值与全0标签之间的均方误差,向着判别器总损失下降的方向进行训练以达到训练判别器的目的;再固定判别器训练生成器,使用均方误差MSE方法计算损失值,该损失值的获取较判别器损失值的获取复杂,应先构建一个Model,该Model的输入为生成器的两个输入,输出为生成器的输出和判别器生成的标签,其中判别器的输入为生成器的输出;则生成器的损失值即为判别器生成的标签和全1标签之间的均方误差,向着生成器损失下降的方向进行训练以达到训练生成器的目的;多次循环进行固定生成器训练判别器、固定判别器训练生成器的操作,直到判别器的错误率为50%左右为止;
第六步、使用最终模型
对变形的pix2pix模型进行训练,生成器的图像生成能力和判别器的判别能力将会得到提高,每训练10次进行训练结果输出和当前生成器模型的保存;当训练总过程结束后,首先通过传统的SIMP算法计算少量迭代步骤,将迭代结果输入到最终保存的生成器模型中进行处理,可迅速得到最终生成结果图,从而实现最终优化结构的快速计算。
2.根据权利要求1所述的一种利用生成对抗网络加速的结构优化设计方法,其特征在于,使用编码器-解码器构建生成器具体操作如下:取出数据集中的数据并进行第二步中的操作,将处理后得到的数据喂入编码器网络进行特征提取并完成图像压缩任务;将编码器网络输出的feature map再送入解码器网络,可将被压缩的图像进行放大,在此过程中,将编码器和解码器过程中具有相同尺寸的feature map连接到解码器中,以便充分利用编码器网络中的结构信息,保持特征的不变性,达到更好的重建效果。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201811637907.3A CN109783910B (zh) | 2018-12-29 | 2018-12-29 | 一种利用生成对抗网络加速的结构优化设计方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201811637907.3A CN109783910B (zh) | 2018-12-29 | 2018-12-29 | 一种利用生成对抗网络加速的结构优化设计方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN109783910A CN109783910A (zh) | 2019-05-21 |
CN109783910B true CN109783910B (zh) | 2020-08-28 |
Family
ID=66498985
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN201811637907.3A Active CN109783910B (zh) | 2018-12-29 | 2018-12-29 | 一种利用生成对抗网络加速的结构优化设计方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN109783910B (zh) |
Families Citing this family (9)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111985597B (zh) * | 2019-05-22 | 2023-10-24 | 华为技术有限公司 | 模型压缩方法及装置 |
CN110210422B (zh) * | 2019-06-05 | 2021-04-27 | 哈尔滨工业大学 | 一种基于光学图像辅助的舰船isar像识别方法 |
CN111008277B (zh) * | 2019-10-30 | 2020-11-03 | 创意信息技术股份有限公司 | 一种自动文本摘要方法 |
CN111105014B (zh) * | 2019-11-27 | 2024-05-28 | 联想(北京)有限公司 | 一种数据处理方法、设备和计算机可读存储介质 |
CN111241700B (zh) * | 2020-01-19 | 2022-12-30 | 中国科学院光电技术研究所 | 一种微波宽带超表面吸收器的智能设计方法 |
CN113076583B (zh) * | 2021-04-09 | 2024-04-02 | 西安交通大学 | 一种基于自注意力机制神经网络的悬臂梁结构设计方法 |
CN113205179B (zh) * | 2021-05-08 | 2023-04-07 | 湖南太观科技有限公司 | 一种用于介质激光加速的深度学习架构 |
CN113343427B (zh) * | 2021-05-11 | 2022-08-05 | 华中科技大学 | 一种基于卷积神经网络的结构拓扑构型预测方法 |
CN113343580A (zh) * | 2021-06-24 | 2021-09-03 | 大连理工大学 | 一种基于人工智能技术的实时拓扑优化生成设计方法 |
Citations (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108537271A (zh) * | 2018-04-04 | 2018-09-14 | 重庆大学 | 一种基于卷积去噪自编码机防御对抗样本攻击的方法 |
EP3406310A1 (en) * | 2017-05-23 | 2018-11-28 | Nokia Technologies Oy | Method and apparatuses for handling visual virtual reality content |
Family Cites Families (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108280305B (zh) * | 2018-01-30 | 2020-03-13 | 西安交通大学 | 基于深度学习的散热器件冷却通道快速拓扑优化设计方法 |
CN109063291B (zh) * | 2018-07-20 | 2021-07-13 | 西安交通大学 | 机电装备冷却通道结构智能化拓扑优化设计方法 |
-
2018
- 2018-12-29 CN CN201811637907.3A patent/CN109783910B/zh active Active
Patent Citations (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
EP3406310A1 (en) * | 2017-05-23 | 2018-11-28 | Nokia Technologies Oy | Method and apparatuses for handling visual virtual reality content |
CN108537271A (zh) * | 2018-04-04 | 2018-09-14 | 重庆大学 | 一种基于卷积去噪自编码机防御对抗样本攻击的方法 |
Also Published As
Publication number | Publication date |
---|---|
CN109783910A (zh) | 2019-05-21 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN109783910B (zh) | 一种利用生成对抗网络加速的结构优化设计方法 | |
CN111860982B (zh) | 一种基于vmd-fcm-gru的风电场短期风电功率预测方法 | |
CN112149316B (zh) | 基于改进的cnn模型的航空发动机剩余寿命预测方法 | |
CN107610194B (zh) | 基于多尺度融合cnn的磁共振图像超分辨率重建方法 | |
CN110212528B (zh) | 一种配电网量测数据缺失重构方法 | |
CN111932461B (zh) | 一种基于卷积神经网络的自学习图像超分辨率重建方法及系统 | |
CN111275171B (zh) | 一种基于参数共享的多尺度超分重建的小目标检测方法 | |
CN111861906B (zh) | 一种路面裂缝图像虚拟增广模型建立及图像虚拟增广方法 | |
CN109658347A (zh) | 一种同时生成多种图片风格的数据增强方法 | |
CN112686817A (zh) | 一种基于不确定性估计的图像补全方法 | |
CN113140023A (zh) | 一种基于空间注意力的文本到图像生成方法及系统 | |
CN116071352A (zh) | 一种电力安全工器具表面缺陷图像的生成方法 | |
CN115310209B (zh) | 基于vae的气动形状迁移优化方法及相关装置 | |
CN117237663A (zh) | 一种大感受野的点云修复方法 | |
CN117035049A (zh) | 智能体学习方法、装置、设备及存储介质 | |
CN114693823B (zh) | 一种基于空频双域并行重建的磁共振图像重建方法 | |
CN114494284B (zh) | 一种基于显式监督区域关系的场景解析模型及方法 | |
CN116152263A (zh) | 一种基于cm-mlp网络的医学图像分割方法 | |
CN114596488A (zh) | 基于密集特征融合网络的轻量级遥感目标检测方法 | |
Li et al. | U-net Network for Building Information Extraction of Remote-Sensing Imagery. | |
CN114630207A (zh) | 基于降噪自编码器的多传感节点感知数据收集方法 | |
CN111401141B (zh) | 一种基于骨架的3d手势估计方法 | |
CN114881843B (zh) | 一种基于深度学习的流体艺术控制方法 | |
CN118333861B (zh) | 一种遥感图像重建方法、系统、装置、介质 | |
CN118115769A (zh) | 一种基于Resnet50改进自编码器的深度卷积嵌入聚类方法、存储介质、终端设备 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |