Dual-Channel Prototype Network for Few-Shot Pathology Image Classification

被引:15
作者
Quan, Hao [1 ]
Li, Xinjia [2 ]
Hu, Dayu
Nan, Tianhang
Cui, Xiaoyu [1 ]
机构
[1] Northeastern Univ, Coll Med & Biol Informat Engn, Shenyang 110819, Peoples R China
[2] Northeastern Univ, Coll Med & Biol Informat Engn, Shenyang 410073, Peoples R China
关键词
Task analysis; Pathology; Prototypes; Adaptation models; Metalearning; Transformers; Measurement; Few-shot learning; Self-supervised learning; Transformer; Pathology images; Classification;
D O I
10.1109/JBHI.2024.3386197
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In the field of pathology, the scarcity of certain diseases and the difficulty of annotating images hinder the development of large, high-quality datasets, which in turn affects the advancement of deep learning-assisted diagnostics. Few-shot learning has demonstrated unique advantages in modeling tasks with limited data, yet explorations of this method in the field of pathology remain in the early stages. To address this issue, we present a dual-channel prototype network (DCPN), a novel few-shot learning approach for efficiently classifying pathology images with limited data. The DCPN leverages self-supervised learning to extend the pyramid vision transformer (PVT) to few-shot classification tasks and combines it with a convolutional neural network to construct a dual-channel network for extracting multi-scale, high-precision pathological features, thereby substantially enhancing the generalizability of prototype representations. Additionally, we design a soft voting classifier based on multi-scale features to further augment the discriminative power of the model in complex pathology image classification tasks. We constructed three few-shot classification tasks with varying degrees of domain shift using three publicly available pathological datasets-CRCTP, NCTCRC, and LC25000-to emulate real-world clinical scenarios. The results demonstrated that the DCPN outperformed the prototypical network across all metrics, achieving the highest accuracies in same-domain tasks-70.86% for 1-shot, 82.57% for 5-shot, and 85.2% for 10-shot setups-corresponding to improvements of 5.51%, 5.72%, and 6.81%, respectively, over the prototypical network. Notably, in the same-domain 10-shot setting, the accuracy of the DCPN (85.2%) surpassed that of the PVT-based supervised learning model (85.15%), confirming its potential to diagnose rare diseases within few-shot learning frameworks.
引用
收藏
页码:4132 / 4144
页数:13
相关论文
共 71 条
[1]  
Borkowski AA, 2019, Arxiv, DOI [arXiv:1912.12142, 10.48550/arXiv.1912.12142]
[2]   Principal component analysis [J].
Abdi, Herve ;
Williams, Lynne J. .
WILEY INTERDISCIPLINARY REVIEWS-COMPUTATIONAL STATISTICS, 2010, 2 (04) :433-459
[3]  
Bertinetto L., 2019, P INT C LEARN REPR S
[4]  
Bromley J., 1993, International Journal of Pattern Recognition and Artificial Intelligence, V7, P669, DOI 10.1142/S0218001493000339
[5]   Few-Shot Learning for Medical Image Classification [J].
Cai, Aihua ;
Hu, Wenxin ;
Zheng, Jun .
ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING, ICANN 2020, PT I, 2020, 12396 :441-452
[6]   Computer-aided diagnosis in the era of deep learning [J].
Chan, Heang-Ping ;
Hadjiiski, Lubomir M. ;
Samala, Ravi K. .
MEDICAL PHYSICS, 2020, 47 (05) :E218-E227
[7]   Disentangle Then Calibrate With Gradient Guidance: A Unified Framework for Common and Rare Disease Diagnosis [J].
Chen, Yuanyuan ;
Guo, Xiaoqing ;
Xia, Yong ;
Yuan, Yixuan .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2024, 43 (05) :1816-1827
[8]   Dynamic feature splicing for few-shot rare disease diagnosis [J].
Chen, Yuanyuan ;
Guo, Xiaoqing ;
Pan, Yongsheng ;
Xia, Yong ;
Yuan, Yixuan .
MEDICAL IMAGE ANALYSIS, 2023, 90
[9]   Artificial intelligence and computational pathology [J].
Cui, Miao ;
Zhang, David Y. .
LABORATORY INVESTIGATION, 2021, 101 (04) :412-422
[10]   Multi-Prototype Few-shot Learning in Histopathology [J].
Deuschel, Jessica ;
Firmbach, Daniel ;
Geppert, Carol, I ;
Eckstein, Markus ;
Hartmann, Arndt ;
Bruns, Volker ;
Kuritcyn, Petr ;
Dexl, Jakob ;
Hartmann, David ;
Perrin, Dominik ;
Wittenberg, Thomas ;
Benz, Michaela .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW 2021), 2021, :620-628