提出可训练的染色归一化模型,提升病理图像跨域分析性能。
Adaptive Stain Normalization for Cross-Domain Medical Histology
- 基于贝叶-兰伯特定律,通过算法展开NMF提取结构不变特征
- 在多个数据集上实现优于主流方法的跨域检测与分类准确率
- 无需模板图像,避免伪影,适合集成到各类下游任务中
深度学习已推动数字病理自动化分析发展,但染色协议与成像条件差异导致显著颜色变异,引发领域偏移问题,降低模型在不同数据上的表现。现有归一化方法常引入伪影或需手动选择模板图像。为此,我们提出一种可训练的染色归一化模型,可嵌入任意主干网络用于目标检测与分类任务。该模型基于成像过程的物理规律(贝叶-兰伯特定律),通过非负矩阵分解(NMF)的算法展开,提取原始病理图像中的染色无关结构信息作为后续处理输入。实验在公开病理数据集及内部疟疾血涂片数据集上验证,所提方法在跨域目标检测与分类任务中均优于多项先进归一化方法。代码已开源:https://github.com/xutianyue/BeerLaNet。
原文摘要 · Abstract (English)
Deep learning advances have revolutionized automated digital pathology analysis. However, differences in staining protocols and imaging conditions can introduce significant color variability. In deep learning, such color inconsistency often reduces performance when deploying models on data acquired under different conditions from the training data, a challenge known as domain shift. Many existing methods attempt to address this problem via color normalization but suffer from several notable drawbacks such as introducing artifacts or requiring careful choice of a template image for stain mapping. To address these limitations, we propose a trainable color normalization model that can be integrated with any backbone network for downstream tasks such as object detection and classification. Based on the physics of the imaging process per the Beer-Lambert law, our model architecture is derived via algorithmic unrolling of a nonnegative matrix factorization (NMF) model to extract stain-invariant structural information from the original pathology images, which serves as input for further processing. Experimentally, we evaluate the method on publicly available pathology datasets and an internally curated collection of malaria blood smears for cross-domain object detection and classification, where our method outperforms many state-of-the-art stain normalization methods. Our code is available at https://github.com/xutianyue/BeerLaNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。