Discovering neural policies to drive behaviour by integrating deep reinforcement learning agents with biological neural networks

Chenguang Li,Gabriel Kreiman,Sharad Ramanathan
DOI: https://doi.org/10.1038/s42256-024-00854-2
IF: 23.8
2024-06-15
Nature Machine Intelligence
Abstract:Deep reinforcement learning (RL) has been successful in a variety of domains but has not yet been directly used to learn biological tasks by interacting with a living nervous system. As proof of principle, we show how to create such a hybrid system trained on a target-finding task. Using optogenetics, we interfaced the nervous system of the nematode Caenorhabditis elegans with a deep RL agent. Agents adapted to strikingly different sites of neural integration and learned site-specific activations to guide animals towards a target, including in cases where agents interfaced with sets of neurons with previously uncharacterized responses to optogenetic modulation. Agents were analysed by plotting their learned policies to understand how different sets of neurons were used to guide movement. Further, the animal and agent generalized to new environments using the same learned policies in food-search tasks, showing that the system achieved cooperative computation rather than the agent acting as a controller for a soft robot. Our system demonstrates that deep RL is a viable tool both for learning how neural circuits can produce goal-directed behaviour and for improving biologically relevant behaviour in a flexible way.
computer science, artificial intelligence, interdisciplinary applications
What problem does this paper attempt to address?