论文标题
3D超声波中具有里程碑意义的检测的带有图形的区域提案网络,并具有图形损失
Region Proposal Network with Graph Prior and IoU-Balance Loss for Landmark Detection in 3D Ultrasound
论文作者
论文摘要
3D超声(US)可以促进用于胎儿生长监测的详细产前检查。为了分析3D US量,准确识别评估器官的解剖标志是至关重要的。典型的深度学习方法通常会直接回归坐标或涉及热图匹配。但是,这些方法难以应对具有较大尺寸以及胎儿高度变化的位置和方向的数量。在这项工作中,我们利用一个对象检测框架来检测3D胎儿面部美国体积中的地标。通过严格的标准回归以里程碑为中心的边界框(B-box)的多个参数,提出的模型能够查明目标地标的确切位置。具体而言,该模型使用3D区域建议网络(RPN)生成3D候选区域,然后使用几个3D分类分支来选择最佳候选人。它还采取了损失,以改善有利于学习过程的分支机构之间的沟通。此外,它在正规化训练之前利用基于距离的图并有助于减少假阳性预测。在3D US数据集上评估了拟议框架的性能,以检测五个关键的胎儿面部标志。结果表明,所提出的方法优于疗效和效率的一些最新方法。
3D ultrasound (US) can facilitate detailed prenatal examinations for fetal growth monitoring. To analyze a 3D US volume, it is fundamental to identify anatomical landmarks of the evaluated organs accurately. Typical deep learning methods usually regress the coordinates directly or involve heatmap-matching. However, these methods struggle to deal with volumes with large sizes and the highly-varying positions and orientations of fetuses. In this work, we exploit an object detection framework to detect landmarks in 3D fetal facial US volumes. By regressing multiple parameters of the landmark-centered bounding box (B-box) with a strict criteria, the proposed model is able to pinpoint the exact location of the targeted landmarks. Specifically, the model uses a 3D region proposal network (RPN) to generate 3D candidate regions, followed by several 3D classification branches to select the best candidate. It also adopts an IoU-balance loss to improve communications between branches that benefits the learning process. Furthermore, it leverages a distance-based graph prior to regularize the training and helps to reduce false positive predictions. The performance of the proposed framework is evaluated on a 3D US dataset to detect five key fetal facial landmarks. Results showed the proposed method outperforms some of the state-of-the-art methods in efficacy and efficiency.