用极限学习机提升病理图像分类效率与精度
Extreme Learning Machines for Attention-based Multiple Instance Learning in Whole-Slide Image Classification
- 将极限学习机与注意力MIL结合,降低训练参数量
- 高维特征空间使AUC提升超10%,模型更稳定
- 适合需要快速训练的单细胞诊断场景
全切片图像分类是计算病理学的核心挑战。基于注意力的多实例学习(MIL)虽有效,但其架构对生物医学图像的影响尚不明确。本文对比多种MIL方法,提出一种利用高维特征空间的深度MIL新方法,并开发融合极限学习机与注意力MIL的新算法,以提升敏感性并降低训练复杂度。应用于外周血中循环稀有细胞(如网织红细胞)检测任务,结果表明非线性对分类稳定性至关重要:移除非线性使平均AUC下降超4%。采用高维特征空间后,平均AUC提升超10%,模型鲁棒性显著增强。极限学习机将训练参数减少5倍,同时保持平均AUC仅比深度MIL低1.5%。未来可探索经典计算框架与量子算法融合。本研究有助于推动精准医学中的高效单细胞诊断。
原文摘要 · Abstract (English)
Whole-slide image classification represents a key challenge in computational pathology and medicine. Attention-based multiple instance learning (MIL) has emerged as an effective approach for this problem. However, the effect of attention mechanism architecture on model performance is not well-documented for biomedical imagery. In this work, we compare different methods and implementations of MIL, including deep learning variants. We introduce a new method using higher-dimensional feature spaces for deep MIL. We also develop a novel algorithm for whole-slide image classification where extreme machine learning is combined with attention-based MIL to improve sensitivity and reduce training complexity. We apply our algorithms to the problem of detecting circulating rare cells (CRCs), such as erythroblasts, in peripheral blood. Our results indicate that nonlinearities play a key role in the classification, as removing them leads to a sharp decrease in stability in addition to a decrease in average area under the curve (AUC) of over 4%. We also demonstrate a considerable increase in robustness of the model with improvements of over 10% in average AUC when higher-dimensional feature spaces are leveraged. In addition, we show that extreme learning machines can offer clear improvements in terms of training efficiency by reducing the number of trained parameters by a factor of 5 whilst still maintaining the average AUC to within 1.5% of the deep MIL model. Finally, we discuss options of enriching the classical computing framework with quantum algorithms in the future. This work can thus help pave the way towards more accurate and efficient single-cell diagnostics, one of the building blocks of precision medicine.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。