Automated Segmentation of Metastatic Lymph Nodes in FDG PET/MRI for Lymphoma Cancer Patients Using Multi-Scale Swin Transformers

Automated segmentation of tumors and designing a computer-assisted diagnosis system requires co-learning of imaging features from the complementary PET/MRI. We aim to develop an automated method for the segmentation of cancer-affected lymph nodes on Positron Emission Tomography/Magnetic Resonance Imaging (PET/MRI) using a modified Swin Transformer model (ST) integrated with a novel Multi-Scale Feature Fusion & Reorganization Module (MSFFRM). Our results with Hodgkin lymphoma (HL) and Non-Hodgkin lymphoma subtype Diffuse Large B-Cell Lymphoma (DLBCL) datasets show that our proposed model ST-MSFFRM achieved better performance in lesion segmentation in comparison to other state-of-the-art segmentation methods.