Multi-organ localization with cascaded global-to-local regression and shape prior

Abstract

We propose a method for fast, accurate and robust localization of several organs in medical images. We generalize the global-to-local cascade of regression random forest to multiple organs. A first regressor encodes the global relationships between organs, learning simultaneously all organs parameters. Then subsequent regressors refine the localization of each organ locally and independently for improved accuracy. By combining the regression vote distribution and the organ shape prior (through probabilistic atlas representation) we compute confidence maps that are organ-dedicated probability maps. They are used within the cascade itself, to better select the test voxels for the second set of regressors, and to provide richer information than the classical bounding boxes result thanks to the shape prior. We propose an extensive study of the different learning and testing parameters, showing both their robustness to reasonable perturbations and their influence on the final algorithm accuracy. Finally we demonstrate the robustness and accuracy of our approach by evaluating the localization of six abdominal organs (liver, two kidneys, spleen, gallbladder and stomach) on a large and diverse database of 130 CT volumes. Moreover, the comparison of our results with two existing methods shows significant improvements brought by our approach and our deep understanding and optimization of the parameters.

Keywords: 3D CT; Abdominal organs; Multi-organ localization; Random forest; Regression.