Emergent behavior and neural dynamics in artificial agents tracking turbulent plumes

TL;DR

This study uses deep reinforcement learning to train artificial agents that develop insect-like behaviors for tracking turbulent plumes, revealing neural dynamics and strategies that could inform biological understanding.

Contribution

It demonstrates how RNN-based agents can learn robust plume tracking behaviors and neural representations, offering new insights into the neural computations underlying this complex task.

Findings

Agents exhibit insect-like tracking behaviors

Neural representations encode task-relevant variables

Plume shape following is key in changing wind conditions

Abstract

Tracking a turbulent plume to locate its source is a complex control problem because it requires multi-sensory integration and must be robust to intermittent odors, changing wind direction, and variable plume statistics. This task is routinely performed by flying insects, often over long distances, in pursuit of food or mates. Several aspects of this remarkable behavior have been studied in detail in many experimental studies. Here, we take a complementary in silico approach, using artificial agents trained with reinforcement learning to develop an integrated understanding of the behaviors and neural computations that support plume tracking. Specifically, we use deep reinforcement learning (DRL) to train recurrent neural network (RNN) agents to locate the source of simulated turbulent plumes. Interestingly, the agents' emergent behaviors resemble those of flying insects, and the RNNs learn to represent task-relevant variables, such as head direction and time since last odor encounter. Our analyses suggest an intriguing experimentally testable hypothesis for tracking plumes in changing wind direction -- that agents follow local plume shape rather than the current wind direction. While reflexive short-memory behaviors are sufficient for tracking plumes in constant wind, longer timescales of memory are essential for tracking plumes that switch direction. At the level of neural dynamics, the RNNs' population activity is low-dimensional and organized into distinct dynamical structures, with some correspondence to behavioral modules. Our in silico approach provides key intuitions for turbulent plume tracking strategies and motivates future targeted experimental and theoretical developments.