Takanori Hashimoto, Nobuyuki Matsui, Naotake Kamiura, Teijiro Isokawa
Journal of Advanced Computational Intelligence and Intelligent Informatics, 27(4) 537-542, Jul 20, 2023 Peer-reviewed
In this study, we investigate model structures for neural ODEs to improve the data efficiency in learning the dynamics of control systems. We introduce two model structures and compare them with a typical baseline structure. The first structure considers the relationship between the coordinates and velocities of the control system, while the second structure adds linearity with respect to the control term to the first structure. Both of these structures can be easily implemented without requiring additional computation. In numerical experiments, we evaluate these structure on simulated simple pendulum and CartPole systems and show that incorporating these characteristics into the model structure leads to accurate learning with a smaller amount of training data compared to the baseline structure.