提出新框架提升癌症生存预测多模态融合效果
Decouple, Reorganize, and Fuse: A Multimodal Framework for Cancer Survival Prediction
- 先解耦再随机重组特征,动态增强跨模态信息交互
- 在肝癌和TCGA四个数据集上显著优于现有方法
- 适合做医学多模态分析、生存预测的研究者参考
癌症生存分析通常整合多种医学模态信息进行生存时间预测。现有方法主要提取各模态的独立特征,并采用拼接、注意力或基于专家混合(MoE)的融合方式。然而,这些方法仍面临两大挑战:一是固定融合策略(如拼接和注意力)导致模型过度依赖预设特征组合,限制了特征的动态融合;二是MoE方法中各专家处理独立解耦特征,限制了特征间的交互。为此,本文提出一种新的解耦-重组-融合框架(DeReF),在解耦与动态MoE融合模块间引入随机特征重组策略。其优势包括:一是增加特征组合的多样性与粒度,提升后续专家网络的泛化能力;二是克服信息封闭问题,帮助专家网络更好捕捉解耦特征间的关联。此外,在模态解耦模块中引入区域交叉注意力网络,提升解耦特征表示质量。在自建肝癌(LC)数据集及三个广泛使用的TCGA公开数据集上的大量实验验证了所提方法的有效性。代码已开源:https://github.com/ZJUMAI/DeReF。
原文摘要 · Abstract (English)
Cancer survival analysis commonly integrates information across diverse medical modalities to make survival-time predictions. Existing methods primarily focus on extracting different decoupled features of modalities and performing fusion operations such as concatenation, attention, and \revm{Mixture-of-Experts (MoE)-based fusion. However, these methods still face two key challenges: i) Fixed fusion schemes (concatenation and attention) can lead to model over-reliance on predefined feature combinations, limiting the dynamic fusion of decoupled features; ii) in MoE-based fusion methods, each expert network handles separate decoupled features, which limits information interaction among the decoupled features. To address these challenges, we propose a novel Decoupling-Reorganization-Fusion framework (DeReF), which devises a random feature reorganization strategy between modalities decoupling and dynamic MoE fusion modules.Its advantages are: i) it increases the diversity of feature combinations and granularity, enhancing the generalization ability of the subsequent expert networks; ii) it overcomes the problem of information closure and helps expert networks better capture information among decoupled features. Additionally, we incorporate a regional cross-attention network within the modality decoupling module to improve the representation quality of decoupled features. Extensive experimental results on our in-house Liver Cancer (LC) and three widely used TCGA public datasets confirm the effectiveness of our proposed method. Codes are available at https://github.com/ZJUMAI/DeReF.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。