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.

DLUNet: Semi-supervised Learning Based Dual-Light UNet for Multi-organ Segmentation.

Multiorgan segmentation from partially labeled datasets with conditional nnU-Net

3D U$$^$$-Net: A 3D Universal U-Net for Multi-domain Medical Image Segmentation

MTSegNet: Semi-supervised Abdominal Organ Segmentation in CT

Densely Connected Deep U-Net For Abdominal Multi-Organ Segmentation

Multi-Dimensional Cascaded Net with Uncertain Probability Reduction for Abdominal Multi-Organ Segmentation in CT Sequences

Fully-automated multi-organ segmentation tool applicable to both non-contrast and post-contrast abdominal CT: deep learning algorithm developed using dual-energy CT images

Dcl-Net: Dual Contrastive Learning Network for Semi-Supervised Multi-Organ Segmentation

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

Double-branch U-Net for multi-scale organ segmentation

UNet 3+: A Full-Scale Connected UNet for Medical Image Segmentation

Semi-Supervised 3D Abdominal Multi-Organ Segmentation Via Deep Multi-Planar Co-Training

Multi-organ segmentation network for abdominal CT images based on spatial attention and deformable convolution

SUnet: A Multi-Organ Segmentation Network Based on Multiple Attention

Efficient Multi-Organ Segmentation from 3D Abdominal CT Images with Lightweight Network and Knowledge Distillation.

Iterative Semi-Supervised Learning for Abdominal Organs and Tumor Segmentation

Scribble-based 3D Multiple Abdominal Organ Segmentation via Triple-branch Multi-dilated Network with Pixel- and Class-wise Consistency

Semi-supervised Detection, Identification and Segmentation for Abdominal Organs

Semi-supervised Abdominal Multi-organ Segmentation Via Contour Aware Dual-Task Consistency

Semi-Supervised Multi-Organ Segmentation Via Deep Multi-Planar Co-Training

A Lightweight Network for Abdominal Multi-Organ Segmentation Based on Multi-Scale Context Fusion and Dual Self-Attention