arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2307.03427eess.IVcs.CVcs.LG

用于头颈部癌症生存预测的合并-发散混合Transformer网络

Merging-Diverging Hybrid Transformer Networks for Survival Prediction in Head and Neck Cancer

School of Computer Science, The University of Sydney(悉尼大学计算机科学学院);Institute of Translational Medicine, Shanghai Jiao Tong University(上海交通大学转化医学研究院);Royal Prince Alfred Hospital(皇家阿尔弗雷德王子医院);Med-X Research Institute, Shanghai Jiao Tong University(上海交通大学Med-X研究院)

Mingyuan Meng, Lei Bi, Michael Fulham, Dagan Feng, Jinman Kim

AI总结:

针对现有深度生存模型在多模态图像利用和区域特异性信息提取上的不足,提出合并-发散学习框架XSurv,通过HPCA模块融合PET-CT特征、RAG模块筛选病变区域特征,在HECKTOR 2022数据集上实现了优于现有方法的头颈部癌症生存预测性能。

摘要:

生存预测对癌症患者至关重要,它能为治疗规划提供早期预后信息。近年来,基于深度学习和医学图像的深度生存模型在生存预测方面已展现出可观的性能。然而,现有深度生存模型在利用多模态图像(如PET-CT)以及提取区域特异性信息(如原发肿瘤(PT)和转移性淋巴结(MLN)区域中的预后信息)方面仍存在不足。鉴于此,我们提出一种用于多模态图像生存预测的合并-发散学习框架。该框架包含一个用于融合多模态信息的合并编码器,以及一个用于提取区域特异性信息的发散解码器。在合并编码器中,我们提出了混合并行交叉注意力(Hybrid Parallel Cross-Attention, HPCA)模块,通过并行卷积层和交叉注意力Transformer有效融合多模态特征。在发散解码器中,我们提出了区域特异性注意力门(Region-specific Attention Gate, RAG)模块,以筛选出与病变区域相关的特征。我们通过设计一种X形合并-发散混合Transformer网络(命名为XSurv),在头颈部(H&N)癌症的PET-CT图像生存预测任务上验证了该框架。我们的XSurv融合了PET和CT图像中的互补信息,并提取了PT和MLN区域的区域特异性预后信息。在头颈肿瘤分割与结局预测挑战赛(HECKTOR 2022)的公开数据集上开展的大量实验表明,我们的XSurv性能优于当前最先进的生存预测方法。

英文摘要:

Survival prediction is crucial for cancer patients as it provides early prognostic information for treatment planning. Recently, deep survival models based on deep learning and medical images have shown promising performance for survival prediction. However, existing deep survival models are not well developed in utilizing multi-modality images (e.g., PET-CT) and in extracting region-specific information (e.g., the prognostic information in Primary Tumor (PT) and Metastatic Lymph Node (MLN) regions). In view of this, we propose a merging-diverging learning framework for survival prediction from multi-modality images. This framework has a merging encoder to fuse multi-modality information and a diverging decoder to extract region-specific information. In the merging encoder, we propose a Hybrid Parallel Cross-Attention (HPCA) block to effectively fuse multi-modality features via parallel convolutional layers and cross-attention transformers. In the diverging decoder, we propose a Region-specific Attention Gate (RAG) block to screen out the features related to lesion regions. Our framework is demonstrated on survival prediction from PET-CT images in Head and Neck (H&N) cancer, by designing an X-shape merging-diverging hybrid transformer network (named XSurv). Our XSurv combines the complementary information in PET and CT images and extracts the region-specific prognostic information in PT and MLN regions. Extensive experiments on the public dataset of HEad and neCK TumOR segmentation and outcome prediction challenge (HECKTOR 2022) demonstrate that our XSurv outperforms state-of-the-art survival prediction methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑