通过全局-局部关联学习,提升直肠癌淋巴结转移预测准确率
WeGA: Weakly-Supervised Global-Local Affinity Learning Framework for Lymph Node Metastasis Prediction in Rectal Cancer
- 采用双分支结构融合全局上下文与局部节点特征
- 在三个测试中心均达AUC 0.75以上,最高0.822
- 适合医学影像智能诊断研究者与临床辅助决策系统开发者
直肠癌淋巴结转移(LNM)的准确评估对治疗方案制定至关重要,但当前基于MRI的评估精度不足,导致临床决策欠佳。自动化系统发展受限于缺乏节点级标注,且现有方法将淋巴结视为孤立个体,忽略其空间与上下文关联。为此,本文提出WeGA框架,通过三项创新解决该问题:1)基于DINOv2的双分支架构,分别提取全局上下文与局部节点细节;2)全局-局部关联提取器,通过跨注意力融合实现多尺度特征对齐;3)区域关联损失,强制分类图与解剖区域保持结构一致性。在1个内部和2个外部测试中心的实验表明,WeGA性能超越现有方法,分别取得AUC 0.750、0.822、0.802。该模型有效建模单个淋巴结与其整体上下文的关系,提供更准确、泛化性更强的转移预测,有望提升直肠癌诊疗精准度。
原文摘要 · Abstract (English)
Accurate lymph node metastasis (LNM) assessment in rectal cancer is essential for treatment planning, yet current MRI-based evaluation shows unsatisfactory accuracy, leading to suboptimal clinical decisions. Developing automated systems also faces significant obstacles, primarily the lack of node-level annotations. Previous methods treat lymph nodes as isolated entities rather than as an interconnected system, overlooking valuable spatial and contextual information. To solve this problem, we present WeGA, a novel weakly-supervised global-local affinity learning framework that addresses these challenges through three key innovations: 1) a dual-branch architecture with DINOv2 backbone for global context and residual encoder for local node details; 2) a global-local affinity extractor that aligns features across scales through cross-attention fusion; and 3) a regional affinity loss that enforces structural coherence between classification maps and anatomical regions. Experiments across one internal and two external test centers demonstrate that WeGA outperforms existing methods, achieving AUCs of 0.750, 0.822, and 0.802 respectively. By effectively modeling the relationships between individual lymph nodes and their collective context, WeGA provides a more accurate and generalizable approach for lymph node metastasis prediction, potentially enhancing diagnostic precision and treatment selection for rectal cancer patients.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。