LNQ Challenge 2023: Learning Mediastinal Lymph Node Segmentation with a Probabilistic Lymph Node Atlas

Sofija Engelson,Jan Ehrhardt,Timo Kepp,Joshua Niemeijer,Heinz Handels
DOI: https://doi.org/10.59275/j.melba.2024-009
2024-06-06
Abstract:The evaluation of lymph node metastases plays a crucial role in achieving precise cancer staging, influencing subsequent decisions regarding treatment options. Lymph node detection poses challenges due to the presence of unclear boundaries and the diverse range of sizes and morphological characteristics, making it a resource-intensive process. As part of the LNQ 2023 MICCAI challenge, we propose the use of anatomical priors as a tool to address the challenges that persist in mediastinal lymph node segmentation in combination with the partial annotation of the challenge training data. The model ensemble using all suggested modifications yields a Dice score of 0.6033 and segments 57% of the ground truth lymph nodes, compared to 27% when training on CT only. Segmentation accuracy is improved significantly by incorporating a probabilistic lymph node atlas in loss weighting and post-processing. The largest performance gains are achieved by oversampling fully annotated data to account for the partial annotation of the challenge training data, as well as adding additional data augmentation to address the high heterogeneity of the CT images and lymph node appearance. Our code is available at <a class="link-external link-https" href="https://github.com/MICAI-IMI-UzL/LNQ2023" rel="external noopener nofollow">this https URL</a>.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
This paper focuses on the automatic segmentation of mediastinal lymph nodes in medical imaging, which is an important step in cancer staging and has an impact on subsequent treatment decisions. The researchers participated in the LNQ (MICCAI Challenge) in 2023 and proposed a deep learning approach that combines a probabilistic lymph node atlas to address the lymph node segmentation challenge with partially annotated data. They integrated multiple spatial anatomical priors into the deep learning model, used the probabilistic lymph node atlas for loss weighting and post-processing, and improved the segmentation accuracy. Key techniques mentioned in the paper include: 1. Using a probabilistic lymph node atlas for loss weighting and post-processing to handle class imbalance and partially annotated data. 2. Compensating for the lack of training data with partially annotated data by oversampling fully annotated data. 3. Introducing additional data augmentation to handle the high heterogeneity of CT images and the diversity of lymph node appearance. 4. Using an architecture based on nnU-Net with improvements, including adding residual connections in the encoder. The experimental results showed that compared to training using only CT images, the proposed method improved the Dice score and increased the number of detectable lymph nodes by almost two times. However, the researchers also pointed out that in real-world challenges, the model performance submitted was lower than the best model due to the lack of comprehensive hyperparameter search on the validation set and differences in features between the training and test sets. In summary, this paper aims to improve the accuracy of mediastinal lymph node automatic segmentation by leveraging anatomical prior knowledge and deep learning techniques to assist in cancer diagnosis and treatment planning.