preprints.org > computer science and mathematics > artificial intelligence and machine learning > doi: 10.20944/preprints202407.1111.v1 (registering DOI)

Preprint Article Version 1 This version is not peer-reviewed

Version 1 : Received: 12 July 2024 / Approved: 13 July 2024 / Online: 15 July 2024 (10:12:16 CEST)

This study investigates the application of deep residual networks for predicting the dynamics of interacting three-dimensional rigid bodies. We present a framework combining a 3D physics simulator implemented in C++ with a deep learning model constructed using PyTorch. The simulator generates training data encompassing linear and angular motion, elastic collisions, fluid friction, gravitational effects, and damping. Our deep residual network, consisting of an input layer, multiple residual blocks, and an output layer, is designed to handle the complexities of 3D dynamics. We evaluate the network's performance using a dataset of 10,000 simulated scenarios, each involving 3-5 interacting rigid bodies. The model achieves a mean squared error of 0.015 for position predictions and 0.022 for orientation predictions, representing a 25% improvement over baseline methods. Our results demonstrate the network's ability to capture intricate physical interactions, with particular success in predicting elastic collisions and rotational dynamics. This work significantly contributes to physics-informed machine learning by showcasing the immense potential of deep residual networks in modeling complex 3D physical systems. We discuss our approach's limitations and propose future directions for improving generalization to more diverse object shapes and materials.

Deep Residual Networks, 3D Physics Simulator, Rigid Body Dynamics, Elastic Collisions, Fluid Friction, Gravitational Effects, Damping, Torch, Machine Learning, Computational Physics

Computer Science and Mathematics, Artificial Intelligence and Machine Learning

* All users must log in before leaving a comment