机构:[1]School of Mechanical and Electrical Engineering, University of Electronic Science and Technology of China, Chengdu, China[2]SenseTime Research, Shanghai, China
Deep learning networks have shown promising performance for object localization in medical images, but require large amount of annotated data for supervised training. To address this problem, we propose: 1) A novel contrastive learning method which embeds the anatomical structure by predicting the Relative Position Regression (RPR) between any two patches from the same volume; 2) An one-shot framework for organ and landmark localization in volumetric medical images. Our main idea comes from that tissues and organs from different human bodies own similar relative position and context. Therefore, we could predict the relative positions of their non-local patches, thus locate the target organ. Our one-shot localization framework is composed of three parts: 1) A deep network trained to project the input patch into a 3D latent vector, representing its anatomical position; 2) A coarse-to-fine framework contains two projection networks, providing more accurate localization of the target; 3) Based on the coarse-to-fine model, we transfer the organ bounding-box (B-box) detection to locating six extreme points along x, y and z directions in the query volume. Experiments on multi-organ localization from head-and-neck (HaN) and abdominal CT volumes showed that our method acquired competitive performance in real time, which is more accurate and 10(5) times faster than template matching methods with the same setting for one-shot localization in 3D medical images. Code is available at https://github.com/HiLab-git/RPR-Loc.
基金:
National Natural Science Foundations of ChinaNational Natural Science Foundation of China (NSFC) [61901084, 81771921]; key research and development project of Sichuan province, China [20ZDYF2817]
语种:
外文
被引次数:
WOS:
第一作者:
第一作者机构:[1]School of Mechanical and Electrical Engineering, University of Electronic Science and Technology of China, Chengdu, China
共同第一作者:
通讯作者:
推荐引用方式(GB/T 7714):
Wenhui Lei,Wei Xu,Ran Gu,et al.Contrastive Learning of Relative Position Regression for One-Shot Object Localization in 3D Medical Images[J].MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2021, PT II.2021,12902:155-165.doi:10.1007/978-3-030-87196-3_15.
APA:
Wenhui Lei,Wei Xu,Ran Gu,Hao Fu,Shaoting Zhang&Guotai Wang.(2021).Contrastive Learning of Relative Position Regression for One-Shot Object Localization in 3D Medical Images.MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2021, PT II,12902,
MLA:
Wenhui Lei,et al."Contrastive Learning of Relative Position Regression for One-Shot Object Localization in 3D Medical Images".MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2021, PT II 12902.(2021):155-165