Deep Reinforcement Learning for Data Association in Cell Tracking

Front Bioeng Biotechnol. 2020 Apr 9:8:298. doi: 10.3389/fbioe.2020.00298. eCollection 2020.

Abstract

Accurate target detection and association are vital for the development of reliable target tracking, especially for cell tracking based on microscopy images due to the similarity of cells. We propose a deep reinforcement learning method to associate the detected targets between frames. According to the dynamic model of each target, the cost matrix is produced by conjointly considering various features of targets and then used as the input of a neural network. The proposed neural network is trained using reinforcement learning to predict a distribution over the association solution. Furthermore, we design a residual convolutional neural network that results in more efficient learning. We validate our method on two applications: the multiple target tracking simulation and the ISBI cell tracking. The results demonstrate that our approach based on reinforcement learning techniques could effectively track targets following different motion patterns and show competitive results.

Keywords: cell tracking; data association; deep learning; deep reinforcement learning; linear assignment problem; residual CNN.