Abstract:Background and Objective: Accurately and reliably defining organs at risk (OARs) and tumors are the cornerstone of radiation therapy (RT) treatment planning for lung cancer. Almost all segmentation networks based on deep learning techniques rely on fully annotated data with strong supervision. However, existing public imaging datasets encountered in the RT domain frequently include singly labelled tumors or partially labelled organs because annotating full OARs and tumors in CT images is both rigorous and tedious. To utilize labelled data from different sources, we proposed a dual-path semi-supervised conditional nnU-Net for OARs and tumor segmentation that is trained on a union of partially labelled datasets.Methods: The framework employs the nnU-Net as the base model and introduces a conditioning strategy by incorporating auxiliary information as an additional input layer into the decoder. The conditional nnU-Net efficiently leverages prior conditional information to classify the target class at the pixelwise level. Specifically, we employ the uncertainty-aware mean teacher (UA-MT) framework to assist in OARs segmentation, which can effectively leverage unlabelled data (images from a tumor labelled dataset) by encouraging consistent predictions of the same input under different perturbations. Furthermore, we individually design different combinations of loss functions to optimize the segmentation of OARs (Dice loss and cross-entropy loss) and tumors (Dice loss and focal loss) in a dual path.Results: The proposed method is evaluated on two publicly available datasets of the spinal cord, left and right lung, heart, esophagus, and lung tumor, in which satisfactory segmentation performance has been achieved in term of both the region-based Dice similarity coefficient (DSC) and the boundary-based Hausdorff distance (HD).Conclusions: The proposed semi-supervised conditional nnU-Net breaks down the barriers between nonoverlapping labelled datasets and further alleviates the problem of "data hunger" and "data waste" in multi-class segmentation. The method has the potential to help radiologists with RT treatment planning in clinical practice.

Automatically transferring supervised targets method for segmenting lung lesion regions with CT imaging

3D Segmentation of Pulmonary Nodules Based on Multi-View and Semi-Supervised

Automatic segmentation of organs at risk and tumors in CT images of lung cancer from partially labelled datasets with a semi-supervised conditional nnU-Net

Multi-task Semi-supervised Learning for Pulmonary Lobe Segmentation

Automatic Lung Segmentation in Low-Dose CT Image with Contrastive Attention Module

CHDNet: A lightweight weakly supervised segmentation network for lung CT image

Deep Co-supervision and Attention Fusion Strategy for Automatic COVID-19 Lung Infection Segmentation on CT Images

Automatic Lung Segmentation in Chest X-Ray Images Using Self-Supervised Learning

Robust and accurate pulmonary nodule detection with self-supervised feature learning on domain adaptation

A general approach for automatic segmentation of pneumonia, pulmonary nodule, and tuberculosis in CT images

Learning COVID-19 Pneumonia Lesion Segmentation from Imperfect Annotations Via Divergence-Aware Selective Training

Automatic Pulmonary Lobe Segmentation Using Deep Learning

A Dual-Branch Framework With Prior Knowledge for Precise Segmentation of Lung Nodules in Challenging CT Scans

Lightweight Techniques to Improve Generalization and Robustness of U-Net Based Networks for Pulmonary Lobe Segmentation

Detection-Guided Deep Learning-Based Model with Spatial Regularization for Lung Nodule Segmentation

Multilevel Attention Unet Segmentation Algorithm for Lung Cancer Based on CT Images

Uncertainty-Aware Deep Learning with Cross-Task Supervision for PHE Segmentation on CT Images

Accurate Lung Nodules Segmentation with Detailed Representation Transfer and Soft Mask Supervision

[Non-local attention and multi-task learning based lung segmentation in chest X-ray]

Self and Mixed Supervision to Improve Training Labels for Multi-Class Medical Image Segmentation

Weakly Supervised Segmentation Framework with Uncertainty: A Study on Pneumothorax Segmentation in Chest X-ray.