Weakly Supervised Lymph Nodes Segmentation Based on Partial Instance Annotations with Pre-trained Dual-branch Network and Pseudo Label Learning

Litingyu Wang1, Yijie Qu1, Xiangde Luo1,2, Wenjun Liao1,3, Shichuan Zhang1,3, Guotai Wang1,2Orcid
1: University of Electronic Science and Technology of China, Chengdu, China, 2: Shanghai AI Laboratory, Shanghai, China, 3: Department of Radiation Oncology, Sichuan Cancer Hospital & Institute, Sichuan Cancer Center, Chengdu, China
Publication date: 2024/08/05
https://doi.org/10.59275/j.melba.2024-489g
PDF · Code · arXiv

Abstract

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

Keywords

Lymph Nodes Segmentation · Label-efficient Learning · Pseudo Labels

Bibtex @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" }
RISTY - 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 -

2024:017 cover