Motion Informed Needle Segmentation in Ultrasound Images

Raghavv Goel,Cecilia Morales,Manpreet Singh,Artur Dubrawski,John Galeotti,Howie Choset
2024-05-04
Abstract:Segmenting a moving needle in ultrasound images is challenging due to the presence of artifacts, noise, and needle occlusion. This task becomes even more demanding in scenarios where data availability is limited. In this paper, we present a novel approach for needle segmentation for 2D ultrasound that combines classical Kalman Filter (KF) techniques with data-driven learning, incorporating both needle features and needle motion. Our method offers three key contributions. First, we propose a compatible framework that seamlessly integrates into commonly used encoder-decoder style architectures. Second, we demonstrate superior performance compared to recent state-of-the-art needle segmentation models using our novel convolutional neural network (CNN) based KF-inspired block, achieving a 15\% reduction in pixel-wise needle tip error and an 8\% reduction in length error. Third, to our knowledge we are the first to implement a learnable filter to incorporate non-linear needle motion for improving needle segmentation.
Image and Video Processing,Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
The paper addresses the challenge of accurately segmenting moving needle tips in ultrasound images, which is difficult due to artifacts, noise, and needle occlusions in the images. The paper proposes a method that combines classical Kalman filter (KF) techniques and data-driven learning to improve 2D ultrasound needle segmentation by utilizing needle features and motion information. This approach handles non-linear needle motion using a learnable filter, resulting in improved segmentation performance and reduced pixel-level tip error and length error.