English

Agent with Tangent-based Formulation and Anatomical Perception for Standard Plane Localization in 3D Ultrasound

Computer Vision and Pattern Recognition 2022-07-04 v1 Artificial Intelligence Machine Learning Image and Video Processing

Abstract

Standard plane (SP) localization is essential in routine clinical ultrasound (US) diagnosis. Compared to 2D US, 3D US can acquire multiple view planes in one scan and provide complete anatomy with the addition of coronal plane. However, manually navigating SPs in 3D US is laborious and biased due to the orientation variability and huge search space. In this study, we introduce a novel reinforcement learning (RL) framework for automatic SP localization in 3D US. Our contribution is three-fold. First, we formulate SP localization in 3D US as a tangent-point-based problem in RL to restructure the action space and significantly reduce the search space. Second, we design an auxiliary task learning strategy to enhance the model's ability to recognize subtle differences crossing Non-SPs and SPs in plane search. Finally, we propose a spatial-anatomical reward to effectively guide learning trajectories by exploiting spatial and anatomical information simultaneously. We explore the efficacy of our approach on localizing four SPs on uterus and fetal brain datasets. The experiments indicate that our approach achieves a high localization accuracy as well as robust performance.

Keywords

Cite

@article{arxiv.2207.00475,
  title  = {Agent with Tangent-based Formulation and Anatomical Perception for Standard Plane Localization in 3D Ultrasound},
  author = {Yuxin Zou and Haoran Dou and Yuhao Huang and Xin Yang and Jikuan Qian and Chaojiong Zhen and Xiaodan Ji and Nishant Ravikumar and Guoqiang Chen and Weijun Huang and Alejandro F. Frangi and Dong Ni},
  journal= {arXiv preprint arXiv:2207.00475},
  year   = {2022}
}

Comments

Accepted by MICCAI 2022