HNT-AI: An Automatic Segmentation Framework for Head and Neck Primary Tumors and Lymph Nodes in FDG- PET/CT Images

Zohaib Salahuddin*, Yi Chen, Xian Zhong, Nastaran Mohammadian Rad, Henry C. Woodruff, Philippe Lambin

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference article in proceedingAcademicpeer-review

33 Downloads (Pure)

Abstract

Head and neck cancer is one of the most prevalent cancers in the world. Automatic delineation of primary tumors and lymph nodes is important for cancer diagnosis and treatment. In this paper, we develop a deep learning-based model for automatic tumor segmentation, HNT-AI, using PET/CT images provided by the MICCAI 2022 Head and Neck Tumor (HECKTOR) segmentation Challenge. We investigate the effect of residual blocks, squeeze-and-excitation normalization, and grid-attention gates on the performance of 3D-UNET. We project the predicted masks on the z-axis and apply k-means clustering to reduce the number of false positive predictions. Our proposed HNT-AI segmentation framework achieves an aggregated dice score of 0.774 and 0.759 for primary tumors and lymph nodes, respectively, on the unseen external test set. Qualitative analysis of the predicted segmentation masks shows that the predicted segmentation mask tends to follow the high standardized uptake value (SUV) area on the PET scans more closely than the ground truth masks. The largest tumor volume, the larget lymph node volume, and the total number of lymph nodes derived from the segmentation proved to be potential biomarkers for recurrence-free survival with a C-index of 0.627 on the test set.
Original languageEnglish
Title of host publicationHead and Neck Tumor Segmentation and Outcome Prediction
Subtitle of host publication3rd Challenge, HECKTOR 2022, Held in Conjunction with MICCAI 2022, Proceedings
EditorsVincent Andrearczyk, Valentin Oreiller, Adrien Depeursinge, Mathieu Hatt
PublisherSpringer Verlag
Pages212-220
Number of pages9
Volume13626 LNCS
Edition3
ISBN (Electronic)9783031274206
ISBN (Print)9783031274190
DOIs
Publication statusPublished - 1 Jan 2023
Event3rd 3D Head and Neck Tumor Segmentation in PET/CT Challenge, HECKTOR 2022, held in Conjunction with the 25th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2022 - Singapore, Singapore
Duration: 22 Sept 202222 Sept 2022
Conference number: 3

Publication series

SeriesLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume13626 LNCS
ISSN0302-9743

Conference

Conference3rd 3D Head and Neck Tumor Segmentation in PET/CT Challenge, HECKTOR 2022, held in Conjunction with the 25th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2022
Country/TerritorySingapore
CitySingapore
Period22/09/2222/09/22

Keywords

  • 3D UNet
  • Grid-attention
  • Residual networks
  • Segmentation biomarkers
  • Squeeze-and-excitation

Cite this