用深度域适应让癌症病理图像跨类型识别,提升模型泛化能力
Transfer Learning from One Cancer to Another via Deep Learning Domain Adaptation
- 将ResNet50改为域对抗网络,实现不同癌种间知识迁移
- 在未标注肺腺癌数据上达95.56%准确率,显著优于原始模型
- 模型关注核密集区等生物有意义区域,具备临床可解释性
监督深度学习模型在训练分布内表现优异,但难以跨分布泛化。在癌症组织病理学中,卷积神经网络(CNN)对训练数据中的癌种分类准确,却无法有效识别未见的相似癌种。尽管不同器官的腺癌具有相似形态特征,仍需直接处理域偏移以实现稳健性能。域适应可通过有标签的源癌种数据,帮助无标签目标癌种的分类,缓解医学图像标注稀缺问题。本研究评估了肺、结肠、乳腺和肾腺癌间的跨域分类表现。单一腺癌训练的ResNet50在自身域上准确率超98%,但在其他域表现极差;集成多个监督模型也无法突破此瓶颈。相反,将ResNet50转化为域对抗神经网络(DANN),在无标签肺腺癌数据上达到95.56%准确率。此外,染色归一化的影响因目标域而异:对肺腺癌,准确率从95.56%降至66.60%;对乳腺和结肠癌,分别从49.22%升至81.29%、78.48%升至83.36%。最后,使用积分梯度分析显示,DANN始终关注密集核区等生物学意义区域,表明其学习到可临床应用的特征。
原文摘要 · Abstract (English)
Supervised deep learning models often achieve excellent performance within their training distribution but struggle to generalize beyond it. In cancer histopathology, for example, a convolutional neural network (CNN) may classify cancer severity accurately for cancer types represented in its training data, yet fail on related but unseen types. Although adenocarcinomas from different organs share morphological features that might support limited cross-domain generalization, addressing domain shift directly is necessary for robust performance. Domain adaptation offers a way to transfer knowledge from labeled data in one cancer type to unlabeled data in another, helping mitigate the scarcity of annotated medical images. This work evaluates cross-domain classification performance among lung, colon, breast, and kidney adenocarcinomas. A ResNet50 trained on any single adenocarcinoma achieves over 98% accuracy on its own domain but shows minimal generalization to others. Ensembling multiple supervised models does not resolve this limitation. In contrast, converting the ResNet50 into a domain adversarial neural network (DANN) substantially improves performance on unlabeled target domains. A DANN trained on labeled breast and colon data and adapted to unlabeled lung data reaches 95.56% accuracy. We also examine the impact of stain normalization on domain adaptation. Its effects vary by target domain: for lung, accuracy drops from 95.56% to 66.60%, while for breast and colon targets, stain normalization boosts accuracy from 49.22% to 81.29% and from 78.48% to 83.36%, respectively. Finally, using Integrated Gradients reveals that DANNs consistently attribute importance to biologically meaningful regions such as densely packed nuclei, indicating that the model learns clinically relevant features and can apply them to unlabeled cancer types.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。