PumpKin: A machine-learning pipeline for automatically tracking localized kinematics in freely moving C. elegans

One of the many goals of neuroscience is to understand how the brain encodes and transforms sensory information into behavior. These animal behaviors can be studied at the level of multi-limb poses or through the focused analysis of individual body parts. Techniques for tracking animal pose, such as DeepLabCut and SLEAP, enable detailed studies of large-scale multi-limb behaviors but show reduced accuracy when used for single-keypoint tracking, where insufficient spatial context leads to increased drift and instability in tracking (Arent I, Schmidt FP, Botsch M et al. Marker-less motion capture of insect locomotion with deep neural networks pre-trained on synthetic videos. Frontiers in Behavioral Neuroscience. Vol. 15. 2021. Tang G, Han Y, Sun X, et al. Anti-drift pose tracker (ADPT), a transformer-based network for robust animal pose estimation cross-species. eLife. Vol. 13. 2025). More general techniques, such as Faster Region-based Convolutional Neural Network (Faster R-CNN) and You Only Look Once (YOLO), have also been used to track location-based behaviors such as center-of-mass position and velocity. However, behaviors localized to a single body structure, such as the pharygeal pumping (i.e., feeding) in the microscopic roundworm Caenorhabditis elegans (C. elegans), are particularly sensitive to noise from moving non-target body parts. This limitation cannot be resolved by simply adding more training data, as doing so often leads to overfitting rather than improved robustness, and instead requires additional processing beyond existing object tracking packages. To address these challenges, we present a fast, automated method that reliably measures pumping in freely moving C. elegans by combining a state-of-the-art object detector (Faster R-CNN) with a tunable noise filter in a technique we call PumpKin. To validate its performance, we demonstrate both its speed (average of 0.4 seconds/frame) and its robust estimation capabilities through application to eight different experimental conditions that encompass both satiety and genetically-driven changes to feeding. PumpKin accurately estimates average pumping rates under eight different experimental conditions, which are positively correlated with the estimates of two expert annotators. Furthermore, PumpKin provides reliable estimates of the instantaneous pumping rate dynamics, achieving an average overlap that exceeds the human-human agreement measured via leave-one-out analysis. Applying PumpKin to conditions differing in satiety revealed a shared basal pumping rate of 0.5 Hz across all worm groups recorded off food, regardless of genetic background or satiety state. Together, these findings highlight PumpKin's ability to accurately isolate and estimate the motion of a single body part during locomotion. Although we present results specific to C. elegans, we anticipate that PumpKin will generalize to behaviors localized to a single body structure in other systems.