Model-Based Simulators and Reinforcement Learning for Magnetic Surgical Robots

Oct 2021 - Sept 2022

Relevant Publication

  • Y. Barnoy, O. Erin, S. Raval, W. Pryor, L. Mair, I. Weinberg, Y. Diaz-Mercado, A. Krieger, G. Hager, “Control of Magnetic Surgical Robots With Model-Based Simulators and Reinforcement Learning”, in IEEE Transactions on Medical Robotics and Bionics, vol. 4, no. 4, pp. 945-956, 2022. [Paper]

Summary

Magnetically manipulated medical robots are a promising alternative to current robotic platforms, allowing for miniaturization and tetherless actuation. Controlling such systems autonomously may enable safe, accurate operation. However, classical control methods require rigorous models of magnetic fields, robot dynamics, and robot environments, which can be difficult to generate. Model-free reinforcement learning (RL) offers an alternative that can bypass these requirements. We apply RL to a robotic magnetic needle manipulation system. Reinforcement learning algorithms often require long runtimes, making them impractical for many surgical robotics applications, most of which require careful, constant monitoring. Our approach first constructs a model-based simulation (MBS) on guided real-world exploration, learning the dynamics of the environment. After intensive MBS environment training, we transfer the learned behavior from the MBS environment to the real world. Our MBS method applies RL roughly 200 times faster than doing so in the real world, and achieves a 6 mm (25% of tool length) root mean-square (RMS) error for a square reference trajectory. In comparison, pure simulation-based approaches fail to transfer, producing a 31 mm RMS error. These results demonstrate that MBS environments are a good solution for domains where running model-free RL is impractical, especially if an accurate simulation is not available.

Video Explanation for Project