Contour interpolation by deep learning approach.

Abstract

PurposeContour interpolation is an important tool for expediting manual segmentation of anatomical structures. The process allows users to manually contour on discontinuous slices and then automatically fill in the gaps, therefore saving time and efforts. The most used conventional shape-based interpolation (SBI) algorithm, which operates on shape information, often performs suboptimally near the superior and inferior borders of organs and for the gastrointestinal structures. In this study, we present a generic deep learning solution to improve the robustness and accuracy for contour interpolation, especially for these historically difficult cases.ApproachA generic deep contour interpolation model was developed and trained using 16,796 publicly available cases from 5 different data libraries, covering 15 organs. The network inputs were a 128×128×5 image patch and the two-dimensional contour masks for the top and bottom slices of the patch. The outputs were the organ masks for the three middle slices. The performance was evaluated on both dice scores and distance-to-agreement (DTA) values.ResultsThe deep contour interpolation model achieved a dice score of 0.95±0.05 and a mean DTA value of 1.09±2.30 mm , averaged on 3167 testing cases of all 15 organs. In a comparison, the results by the conventional SBI method were 0.94±0.08 and 1.50±3.63 mm , respectively. For the difficult cases, the dice score and DTA value were 0.91±0.09 and 1.68±2.28 mm by the deep interpolator, compared with 0.86±0.13 and 3.43±5.89 mm by SBI. The t-test results confirmed that the performance improvements were statistically significant ( p<0.05 ) for all cases in dice scores and for small organs and difficult cases in DTA values. Ablation studies were also performed.ConclusionsA deep learning method was developed to enhance the process of contour interpolation. It could be useful for expediting the tasks of manual segmentation of organs and structures in the medical images.