Agent with Warm Start and Adaptive Dynamic Termination for Plane
Localization in 3D Ultrasound
- URL: http://arxiv.org/abs/2103.14502v1
- Date: Fri, 26 Mar 2021 14:57:26 GMT
- Title: Agent with Warm Start and Adaptive Dynamic Termination for Plane
Localization in 3D Ultrasound
- Authors: Xin Yang, Haoran Dou, Ruobing Huang, Wufeng Xue, Yuhao Huang, Jikuan
Qian, Yuanji Zhang, Huanjia Luo, Huizhi Guo, Tianfu Wang, Yi Xiong, Dong Ni
- Abstract summary: This study enhances our previous RL framework with a newly designed adaptive dynamic termination to enable an early stop for the agent searching.
Our approach achieves localization error of 2.52mm/10.26 degrees, 2.48mm/10.39 degrees, 2.02mm/10.48 degrees, 2.00mm/14.57 degrees, 2.61mm/9.71 degrees, 3.09mm/9.58 degrees, 1.49mm/7.54 degrees for the transcerebellar, transventricular, transthalamic planes in fetal brain, abdominal plane in fetal abdomen, and mid-sagittal, transverse and coronal planes in uterus.
- Score: 14.256624552635786
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Accurate standard plane (SP) localization is the fundamental step for
prenatal ultrasound (US) diagnosis. Typically, dozens of US SPs are collected
to determine the clinical diagnosis. 2D US has to perform scanning for each SP,
which is time-consuming and operator-dependent. While 3D US containing multiple
SPs in one shot has the inherent advantages of less user-dependency and more
efficiency. Automatically locating SP in 3D US is very challenging due to the
huge search space and large fetal posture variations. Our previous study
proposed a deep reinforcement learning (RL) framework with an alignment module
and active termination to localize SPs in 3D US automatically. However,
termination of agent search in RL is important and affects the practical
deployment. In this study, we enhance our previous RL framework with a newly
designed adaptive dynamic termination to enable an early stop for the agent
searching, saving at most 67% inference time, thus boosting the accuracy and
efficiency of the RL framework at the same time. Besides, we validate the
effectiveness and generalizability of our algorithm extensively on our in-house
multi-organ datasets containing 433 fetal brain volumes, 519 fetal abdomen
volumes, and 683 uterus volumes. Our approach achieves localization error of
2.52mm/10.26 degrees, 2.48mm/10.39 degrees, 2.02mm/10.48 degrees, 2.00mm/14.57
degrees, 2.61mm/9.71 degrees, 3.09mm/9.58 degrees, 1.49mm/7.54 degrees for the
transcerebellar, transventricular, transthalamic planes in fetal brain,
abdominal plane in fetal abdomen, and mid-sagittal, transverse and coronal
planes in uterus, respectively. Experimental results show that our method is
general and has the potential to improve the efficiency and standardization of
US scanning.
Related papers
- Ultrasound Plane Pose Regression: Assessing Generalized Pose Coordinates
in the Fetal Brain [9.465965149145559]
We aim to build a US plane localization system for 3D visualization, training, and guidance without integrating additional sensors.
This work builds on our previous work, which predicts the six-dimensional (6D) pose of arbitrarily oriented US planes slicing the fetal brain.
We investigate the impact of registration quality in the training and testing data and its subsequent effect on trained models.
arXiv Detail & Related papers (2023-01-19T21:16:36Z) - Localizing Scan Targets from Human Pose for Autonomous Lung Ultrasound
Imaging [61.60067283680348]
With the advent of COVID-19 global pandemic, there is a need to fully automate ultrasound imaging.
We propose a vision-based, data driven method that incorporates learning-based computer vision techniques.
Our method attains an accuracy level of 15.52 (9.47) mm for probe positioning and 4.32 (3.69)deg for probe orientation, with a success rate above 80% under an error threshold of 25mm for all scan targets.
arXiv Detail & Related papers (2022-12-15T14:34:12Z) - Agent with Tangent-based Formulation and Anatomical Perception for
Standard Plane Localization in 3D Ultrasound [56.7645826576439]
We introduce a novel reinforcement learning framework for automatic SP localization in 3D US.
First, we formulate SP localization in 3D US as a tangent-point-based problem in RL to restructure the action 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.
arXiv Detail & Related papers (2022-07-01T14:53:27Z) - Federated Learning Enables Big Data for Rare Cancer Boundary Detection [98.5549882883963]
We present findings from the largest Federated ML study to-date, involving data from 71 healthcare institutions across 6 continents.
We generate an automatic tumor boundary detector for the rare disease of glioblastoma.
We demonstrate a 33% improvement over a publicly trained model to delineate the surgically targetable tumor, and 23% improvement over the tumor's entire extent.
arXiv Detail & Related papers (2022-04-22T17:27:00Z) - Searching Collaborative Agents for Multi-plane Localization in 3D
Ultrasound [15.573821037925143]
3D ultrasound can contain multiple standard planes (SPs) in one shot.
Manual SP localization in 3D US is challenging because of the low image quality, huge search space and large anatomical variability.
We propose a novel multi-agent reinforcement learning framework to simultaneously localize multiple SPs in 3D US.
arXiv Detail & Related papers (2021-05-22T02:48:23Z) - Searching Collaborative Agents for Multi-plane Localization in 3D
Ultrasound [59.97366727654676]
3D ultrasound (US) is widely used due to its rich diagnostic information, portability and low cost.
Standard plane (SP) localization in US volume not only improves efficiency and reduces user-dependence, but also boosts 3D US interpretation.
We propose a novel Multi-Agent Reinforcement Learning framework to localize multiple uterine SPs in 3D US simultaneously.
arXiv Detail & Related papers (2020-07-30T07:23:55Z) - Hybrid Attention for Automatic Segmentation of Whole Fetal Head in
Prenatal Ultrasound Volumes [52.53375964591765]
We propose the first fully-automated solution to segment the whole fetal head in US volumes.
The segmentation task is firstly formulated as an end-to-end volumetric mapping under an encoder-decoder deep architecture.
We then combine the segmentor with a proposed hybrid attention scheme (HAS) to select discriminative features and suppress the non-informative volumetric features.
arXiv Detail & Related papers (2020-04-28T14:43:05Z) - Agent with Warm Start and Active Termination for Plane Localization in
3D Ultrasound [56.14006424500334]
Standard plane localization is crucial for ultrasound (US) diagnosis.
In prenatal US, dozens of standard planes are manually acquired with a 2D probe.
We propose a novel reinforcement learning framework to automatically localize fetal brain standard planes in 3D US.
arXiv Detail & Related papers (2019-10-10T02:21:52Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.