Learning Flexible Body Collision Dynamics with Hierarchical Contact Mesh Transformer

Youn-Yeol Yu,Jeongwhan Choi,Woojin Cho,Kookjin Lee,Nayong Kim,Kiseok Chang,Chang-Seung Woo,Ilho Kim,Seok-Woo Lee,Joon-Young Yang,Sooyoung Yoon,Noseong Park
2024-03-26
Abstract:Recently, many mesh-based graph neural network (GNN) models have been proposed for modeling complex high-dimensional physical systems. Remarkable achievements have been made in significantly reducing the solving time compared to traditional numerical solvers. These methods are typically designed to i) reduce the computational cost in solving physical dynamics and/or ii) propose techniques to enhance the solution accuracy in fluid and rigid body dynamics. However, it remains under-explored whether they are effective in addressing the challenges of flexible body dynamics, where instantaneous collisions occur within a very short timeframe. In this paper, we present Hierarchical Contact Mesh Transformer (HCMT), which uses hierarchical mesh structures and can learn long-range dependencies (occurred by collisions) among spatially distant positions of a body -- two close positions in a higher-level mesh correspond to two distant positions in a lower-level mesh. HCMT enables long-range interactions, and the hierarchical mesh structure quickly propagates collision effects to faraway positions. To this end, it consists of a contact mesh Transformer and a hierarchical mesh Transformer (CMT and HMT, respectively). Lastly, we propose a flexible body dynamics dataset, consisting of trajectories that reflect experimental settings frequently used in the display industry for product designs. We also compare the performance of several baselines using well-known benchmark datasets. Our results show that HCMT provides significant performance improvements over existing methods. Our code is available at <a class="link-external link-https" href="https://github.com/yuyudeep/hcmt" rel="external noopener nofollow">this https URL</a>.
Machine Learning,Artificial Intelligence,Computational Engineering, Finance, and Science
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to effectively simulate the dynamic behavior of instantaneous collisions in flexible body dynamics. Specifically, although the existing grid - based Graph Neural Network (GNN) models have made significant progress in fluid and rigid body dynamics, they still have deficiencies when dealing with flexible body collision problems. The characteristics of flexible body collisions are a high degree of nonlinearity and the occurrence of collisions in a very short time, which makes it difficult for traditional GNN models to quickly propagate the collision effect to distant positions. To solve these problems, the authors propose the Hierarchical Contact Mesh Transformer (HCMT), whose main objectives are: 1. **Effectively propagate the collision effect**: By introducing a hierarchical mesh structure, HCMT can learn long - range dependencies between spatially distant positions, thereby more effectively propagating the collision effect. 2. **Improve computational efficiency**: By reducing the number of nodes, HCMT can reduce the computational complexity, thereby speeding up the training and inference speed. 3. **Provide a new benchmark dataset**: In order to evaluate the model performance, the authors created a new dataset named Impact Plate, which reflects the experimental settings commonly used in the product design of the display industry. ### Specific problem description The collision problems in flexible body dynamics have the following characteristics: - **Strong nonlinearity**: The dynamic equations of flexible bodies usually contain complex nonlinear terms, such as the mass matrix \(M(x)\), the damping matrix \(D(x)\), and the stiffness matrix \(T(x)\), which change with time and spatial position. - **Instantaneous collision**: The time of collision occurrence is very short, and the model needs to be able to quickly capture and propagate the collision effect. - **Long - range dependence**: The collision effect not only affects the area near the contact point but also quickly spreads to distant positions. ### Limitations of existing methods The existing grid - based GNN models mainly have the following limitations: - **Local message passing**: GNN models usually adopt a local message - passing mechanism and it is difficult to quickly propagate long - range dependencies. - **High computational complexity**: Direct application of Transformer for global self - attention calculation will lead to a computational complexity of \(O(N^2)\), where \(N\) is the number of nodes, which is not feasible in large - scale systems. ### HCMT's solutions HCMT solves the above problems in the following ways: - **Hierarchical mesh structure**: By reducing the number of nodes through hierarchical pooling operations, the computational complexity is reduced. - **Two - branch self - attention mechanism**: Deal with contact edges and mesh edges respectively to capture collision dynamics and long - range interactions. - **New benchmark dataset**: Provide a dataset more suitable for evaluating the performance of flexible body collision simulation. In summary, this paper aims to fill the gap of existing methods in flexible body collision simulation by proposing HCMT and provide an efficient and accurate solution.