arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.21395cs.CV

动量记忆用于计算病理学中的知识蒸馏

Momentum Memory for Knowledge Distillation in Computational Pathology

Wake Forest University School of Medicine(威克森林大学医学院)

Yongxin Guo, Hao Lu, Onur C. Koyun, Zhengjie Zhu, Muhammet Fatih Demir, Metin Nafi Gurcan

AI总结:

本文提出MoMKD框架,通过动量更新的记忆机制整合基因组和病理学信息,提升跨模态知识蒸馏性能,实验表明其在病理学任务中表现优异。

摘要:

多模态学习整合基因组学和组织病理学在癌症诊断中展现出强大潜力,但其临床应用受限于配对的组织病理学-基因组学数据有限。知识蒸馏(KD)通过将基因组监督转移到病理学模型中,使仅使用病理学进行准确推断成为可能。然而,现有KD方法依赖于批量本地对齐,由于批量内比较有限,导致不稳定并最终降低性能。为了解决这些限制,我们提出动量记忆知识蒸馏(MoMKD),一种由动量更新的记忆驱动的跨模态蒸馏框架。该记忆在批量间聚合基因组和病理学信息,有效扩大了每个小批量可用的监督上下文。此外,我们解耦基因组和病理学分支的梯度,防止基因组信号在训练过程中主导病理学特征学习,并在推理时消除模态差距问题。在TCGA-BRCA基准(HER2、PR和ODX分类任务)和一个独立的内部测试数据集上的广泛实验表明,MoMKD在病理学单独推断下一致优于最先进的MIL和多模态KD基线,表现出强大的性能和泛化能力。总体而言,MoMKD建立了计算病理学的稳健且可推广的知识蒸馏范式。

英文摘要:

Multimodal learning that integrates genomics and histopathology has shown strong potential in cancer diagnosis, yet its clinical translation is hindered by the limited availability of paired histology-genomics data. Knowledge distillation (KD) offers a practical solution by transferring genomic supervision into histopathology models, enabling accurate inference using histology alone. However, existing KD methods rely on batch-local alignment, which introduces instability due to limited within-batch comparisons and ultimately degrades performance. To address these limitations, we propose Momentum Memory Knowledge Distillation (MoMKD), a cross-modal distillation framework driven by a momentum-updated memory. This memory aggregates genomic and histopathology information across batches, effectively enlarging the supervisory context available to each mini-batch. Furthermore, we decouple the gradients of the genomics and histology branches, preventing genomic signals from dominating histology feature learning during training and eliminating the modality-gap issue at inference time. Extensive experiments on the TCGA-BRCA benchmark (HER2, PR, and ODX classification tasks) and an independent in-house testing dataset demonstrate that MoMKD consistently outperforms state-of-the-art MIL and multimodal KD baselines, delivering strong performance and generalization under histology-only inference. Overall, MoMKD establishes a robust and generalizable knowledge distillation paradigm for computational pathology.

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

↑