FocusNetv2: Imbalanced large and small organ segmentation with adversarial shape constraint for head and neck CT images

Yunhe Gao,Rui Huang,Yiwei Yang,Jie Zhang,Kainan Shao,Changjuan Tao,Yuanyuan Chen,Dimitris N Metaxas,Hongsheng Li,Ming Chen,Dimitris N. Metaxas
DOI: https://doi.org/10.1016/j.media.2020.101831
IF: 10.9
2021-01-01
Medical Image Analysis
Abstract:<p>Radiotherapy is a treatment where radiation is used to eliminate cancer cells. The delineation of organs-at-risk (OARs) is a vital step in radiotherapy treatment planning to avoid damage to healthy organs. For nasopharyngeal cancer, more than 20 OARs are needed to be precisely segmented in advance. The challenge of this task lies in complex anatomical structure, low-contrast organ contours, and the extremely imbalanced size between large and small organs. Common segmentation methods that treat them equally would generally lead to inaccurate small-organ labeling. We propose a novel two-stage deep neural network, FocusNetv2, to solve this challenging problem by automatically locating, ROI-pooling, and segmenting small organs with specifically designed small-organ localization and segmentation sub-networks while maintaining the accuracy of large organ segmentation. In addition to our original FocusNet, we employ a novel adversarial shape constraint on small organs to ensure the consistency between estimated small-organ shapes and organ shape prior knowledge. Our proposed framework is extensively tested on both self-collected dataset of 1,164 CT scans and the <em>MICCAI Head and Neck Auto Segmentation Challenge 2015</em> dataset, which shows superior performance compared with state-of-the-art head and neck OAR segmentation methods.</p>
engineering, biomedical,computer science, interdisciplinary applications, artificial intelligence,radiology, nuclear medicine & medical imaging
What problem does this paper attempt to address?
The paper attempts to address the issue of imbalanced segmentation of organs of varying sizes in head and neck CT images. Specifically, radiotherapy is an important means of cancer treatment, but to prevent radiation damage to healthy organs, it is necessary to accurately delineate organs at risk (OARs) in the treatment plan. For nasopharyngeal carcinoma, it is required to accurately segment more than 20 OARs before treatment, which faces challenges such as complex anatomical structures, low-contrast organ contours, and extremely imbalanced sizes between large and small organs. Existing segmentation methods often result in inaccurate annotation of small organs. To address these issues, the authors propose a new two-stage deep neural network framework—FocusNetv2. This framework can automatically locate, ROI pool, and segment small organs while maintaining the accuracy of large organ segmentation. Additionally, the authors introduce a new adversarial shape constraint to ensure that the predicted shapes of small organs are consistent with medical prior knowledge. Through this approach, FocusNetv2 demonstrates superior performance over existing methods on both large-scale hospital datasets and open-source datasets.