Neurophysiologically Realistic Environment for Comparing Adaptive Deep Brain Stimulation Algorithms in Parkinson Disease
Ekaterina Kuzmina, Dmitrii Kriukov, Mikhail Lebedev, Dmitry V. Dylov
TL;DR
This work addresses the challenge of validating adaptive DBS (aDBS) policies for Parkinson disease by introducing a neurophysiologically realistic benchmark built on Kuramoto oscillator dynamics. The environment encodes bandwidth, spatial, and temporal features, including beta-band activity, electrode-drift, neural plasticity, and directional stimulation, to train and evaluate reinforcement learning controllers in a controlled, configurable setting. It demonstrates how online RL methods, particularly SAC, can learn strategies that suppress pathological beta oscillations while managing energy expenditure across progressively complex environments, highlighting the value of a standardized pretraining platform for ML-driven neurostimulation. The framework aims to improve generalizability and stability of aDBS strategies and to accelerate their translation by providing an open, scalable benchmark for ML researchers and clinicians alike.
Abstract
Adaptive deep brain stimulation (aDBS) has emerged as a promising treatment for Parkinson disease (PD). In aDBS, a surgically placed electrode sends dynamically altered stimuli to the brain based on neurophysiological feedback: an invasive gadget that limits the amount of data one could collect for optimizing the control offline. As a consequence, a plethora of synthetic models of PD and those of the control algorithms have been proposed. Herein, we introduce the first neurophysiologically realistic benchmark for comparing said models. Specifically, our methodology covers not only conventional basal ganglia circuit dynamics and pathological oscillations, but also captures 15 previously dismissed physiological attributes, such as signal instabilities and noise, neural drift, electrode conductance changes and individual variability - all modeled as spatially distributed and temporally registered features via beta-band activity in the brain and a feedback. Furthermore, we purposely built our framework as a structured environment for training and evaluating deep reinforcement learning (RL) algorithms, opening new possibilities for optimizing aDBS control strategies and inviting the machine learning community to contribute to the emerging field of intelligent neurostimulation interfaces.
