Assessing the presence of potentially malignant lymph nodes aids in estimating cancer progression, and identifying surrounding benign lymph nodes can assist in determining potential metastatic pathways for cancer. For quantitative analysis, automatic segmentation of lymph nodes is crucial. However, due to the labor-intensive and time-consuming manual annotation process required for a large number of lymph nodes, it is more practical to annotate only a subset of the lymph node instances to reduce annotation costs. In this study, we propose a pre-trained Dual-Branch network with Dynamically Mixed Pseudo label (DBDMP) to learn from partial instance annotations for lymph nodes segmentation. To obtain reliable pseudo labels for lymph nodes that are not annotated, we employ a dual-decoder network to generate different outputs that are then dynamically mixed. We integrate the original weak partial annotations with the mixed pseudo labels to supervise the network. To further leverage the extensive amount of unannotated voxels, we apply a self-supervised pre-training strategy to enhance the model’s feature extraction capability. Experiments on the mediastinal Lymph Node Quantification (LNQ) dataset demonstrate that our method, compared to directly learning from partial instance annotations, significantly improves the Dice Similarity Coefficient (DSC) from 11.04% to 54.10% and reduces the Average Symmetric Surface Distance (ASSD) from 20.83 mm to 8.72 mm. The code is available at https://github.com/WltyBY/LNQ2023_training_code
Lymph Nodes Segmentation · Label-efficient Learning · Pseudo Labels
@article{melba:2024:017:wang,
title = "Weakly Supervised Lymph Nodes Segmentation Based on Partial Instance Annotations with Pre-trained Dual-branch Network and Pseudo Label Learning",
author = "Wang, Litingyu and Qu, Yijie and Luo, Xiangde and Liao, Wenjun and Zhang, Shichuan and Wang, Guotai",
journal = "Machine Learning for Biomedical Imaging",
volume = "2",
issue = "MICCAI 2023 LNQ challenge special issue",
year = "2024",
pages = "1030--1047",
issn = "2766-905X",
doi = "https://doi.org/10.59275/j.melba.2024-489g",
url = "https://melba-journal.org/2024:017"
}
TY - JOUR
AU - Wang, Litingyu
AU - Qu, Yijie
AU - Luo, Xiangde
AU - Liao, Wenjun
AU - Zhang, Shichuan
AU - Wang, Guotai
PY - 2024
TI - Weakly Supervised Lymph Nodes Segmentation Based on Partial Instance Annotations with Pre-trained Dual-branch Network and Pseudo Label Learning
T2 - Machine Learning for Biomedical Imaging
VL - 2
IS - MICCAI 2023 LNQ challenge special issue
SP - 1030
EP - 1047
SN - 2766-905X
DO - https://doi.org/10.59275/j.melba.2024-489g
UR - https://melba-journal.org/2024:017
ER -