Library of deep-learning image segmentation and outcomes model-implementations

Abstract

An open-source library of implementations for deep-learning-based image segmentation and outcomes models based on radiotherapy and radiomics is presented. As oncology treatment planning becomes increasingly driven by automation, such a library of model implementations is crucial to (i) validate existing models on datasets collected at different institutions, (ii) automate segmentation, (iii) create ensembles for improving performance and (iv) incorporate validated models in the clinical workflow. Inclusion of deep-learning-based image segmentation and outcomes models in the same library provides a fully automated and reproduceable pipeline to estimate prognosis. The library was developed with the Computational Environment for Radiological Research (CERR) software platform. Centralizing model implementations in CERR builds upon its rich set of radiotherapy and radiomics tools and caters to the world-wide user base. CERR provides well-validated feature extraction pipelines for radiotherapy dosimetry and radiomics with fine control over the calculation settings, allowing users to select appropriate parameters used in model derivation. Models for automatic image segmentation are distributed via containers, allowing them to be deployed with a variety of scientific computing architectures. The library includes implementations of popular DVH-based models outlined in the Quantitative Analysis of Normal Tissue Effects in the Clinic effort and recently published literature. Radiomics models include features from the Image Biomarker Standardization Initiative and application-specific features found to be relevant across multiple sites and image modalities. The library is distributed as a module within CERR at https://www.github.com/cerr/CERR under the GNU-GPL copyleft with additional restrictions on clinical and commercial use and provision to dual license in future.

Keywords: Deep-learning; Image segmentation; Library; Model implementations; Normal tissue complication; Radiomics; Radiotherapy outcomes; Tumor control.