20th AIAI 2024, 27 - 30 June 2024, Corfu, Greece

MTA-Net: a Multi-task Assisted Network for Whole-body Lymphoma Segmentation

Zhaohai Liang, Jiayi Wu, Siyi Chai, Yingkai Wang, Chengdong Li, Cong Shen, Jingmin Xin

Abstract:

  Lymphoma treatment planning and prognosis assessment require accurate segmentation of lymphoma lesions. Positron emission tomography (PET) /computed tomography (CT) is widely used for lymphoma segmentation. Many methods do automatic segmentation of lymphoma based on PET/CT. However, a significant challenge that limits the effectiveness of the segmentation method is the large and imbalance variation in size of whole-body lymphoma lesions.For example, a small percentage of images contain large lesions, while most images contain only small lesions or even no lesions, which results in inaccurate segmentation. In this paper, we propose a Multi-task Assisted Network (MTA-Net) for whole-body lymphoma segmentation. First, we design a novel Multi-task Cross-scale Transformer (MCT) block, which combines the pixels regression task and the whole image classification task at multiple scales. Second, we design a Classification Dynamic Convolution (CDC) whose parameters are additionally controlled by the classification task to assist the segmentation task. In our private whole-body lymphoma dataset, experiments show that MTA-Net achieves the best result among state-of-the-art methods on Dice, HD (Hausdorff Distance), Recall, and Precision.  

*** Title, author list and abstract as seen in the Camera-Ready version of the paper that was provided to Conference Committee. Small changes that may have occurred during processing by Springer may not appear in this window.