Page 25 - 《爆炸与冲击》2026年第5期
P. 25

第 46 卷         袁基宸,等: 基于GNN/KAN的高应变速率金属材料本构关系的表征方法                              第 5 期

               息传递机制在图结构数据中传播消息,即每个数据节点向其邻居节点发送其温度、应变速率、应变等特
               征属性消息。每个节点将自身特征向量、所有相邻节点的特征向量和节点之间的边特征进行信息汇总,
               再以此作为输入通过节点更新函数对每个节点进行状态更新。更新后,所有节点的温度、应变速率、应
               变等特征组成图结构数据的图特征,通过图特征可以对测试数据进行应力预测。




                                                            V
                                                     (V, E)                             (V, E)

                                              Node raw data
                        High-dimensional raw data              Hidden layers


                                      Data to be              E
                                      predicted                                   Passing



                                                         Termination conditions
                                             Graph feature    are met
                                               vectors         Read
                                                                                  Updating
                                                         Optimize parameters
                                                Loss
                             One-dimensional
                               predict data

                                                     图 3    GNN  模型结构
                                                Fig. 3    Structure of the GNN model

                1.4    KAN  模型
                   基于   Kolmogorov-Arnold networks 基本原理构建     KAN  模型,模型结构如图         4  所示,其中    d×k 为输入
               数据的维度,y 为权重值,N         为模型中     KAN  线形层(KAN-Linear 层)的层数。

                                                                                     KAN-Linear layer

                                                            y (base)
                                                              Base
                                                             weights
                                                                         y (output)
                         Input
                         (d×k)    Spline mesh               y (spline)
                                       Adaptive mesh updates  B-spline weights







                                                         Output
                                                     (One-dimensional)       N×KAN linear layer

                                                     图 4    KAN  模型结构
                                                Fig. 4    Structure of the KAN model

                   模型的主要结构为多层           KAN-Linear 层组成的神经网络,它在输入特征与输出之间构建了一个灵活
               的非线性映射,每一层由基础权重(base weight)和样条权重(spline weight)组成。首先,将需要训练的原



                                                         051421-5
   20   21   22   23   24   25   26   27   28   29   30