arXiv:2412.03084eess.IVcs.CV2024-12

用混合深度学习模型提升肝癌分级准确率,解决人工判读效率低问题。

Hybrid deep learning-based strategy for the hepatocellular carcinoma cancer grade classification of H&E stained liver histopathology images

  • 结合预训练CNN与自定义全连接分类器,通过迁移学习提取病理图像特征。
  • 在TCGA数据集上达到100%准确率,在KMC数据集上达96.7%准确率。
  • 适合医学影像分析、病理科辅助诊断研究者参考。

肝细胞癌(HCC)是常见肝癌类型,早期诊断面临挑战,主要因人为主观评估苏木精-伊红染色全切片图像耗时且结果易变。为实现精准检测,本文提出一种混合深度学习架构:利用预训练卷积神经网络(CNN)模型进行特征提取,并结合由全连接层构成的分类器。模型在公开的癌症基因组图谱肝癌数据库(TCGA-LIHC,n=491)上开发,并在印度卡斯特尔巴·甘地医学院(KMC)数据库上验证。预处理包括图像块提取、颜色归一化和增强,共生成3920个图像块用于TCGA数据集。采用五折交叉验证训练包含预训练特征提取器和定制人工神经网络分类器的混合模型。比较了8种主流模型作为特征提取器的表现。基于ResNet50的混合模型在TCGA数据库上实现敏感性、特异性、F1分数、准确率和AUC均为100.00%;在KMC数据库中,EfficientNetb3表现最优,对应指标分别为96.97%、98.85%、96.71%、96.71%和0.99。相比独立预训练模型,该方法在两数据库上分别提升准确率2%和4%。

原文摘要 · Abstract (English)

Hepatocellular carcinoma (HCC) is a common type of liver cancer whose early-stage diagnosis is a common challenge, mainly due to the manual assessment of hematoxylin and eosin-stained whole slide images, which is a time-consuming process and may lead to variability in decision-making. For accurate detection of HCC, we propose a hybrid deep learning-based architecture that uses transfer learning to extract the features from pre-trained convolutional neural network (CNN) models and a classifier made up of a sequence of fully connected layers. This study uses a publicly available The Cancer Genome Atlas Hepatocellular Carcinoma (TCGA-LIHC)database (n=491) for model development and database of Kasturba Gandhi Medical College (KMC), India for validation. The pre-processing step involves patch extraction, colour normalization, and augmentation that results in 3920 patches for the TCGA dataset. The developed hybrid deep neural network consisting of a CNN-based pre-trained feature extractor and a customized artificial neural network-based classifier is trained using five-fold cross-validation. For this study, eight different state-of-the-art models are trained and tested as feature extractors for the proposed hybrid model. The proposed hybrid model with ResNet50-based feature extractor provided the sensitivity, specificity, F1-score, accuracy, and AUC of 100.00%, 100.00%, 100.00%, 100.00%, and 1.00, respectively on the TCGA database. On the KMC database, EfficientNetb3 resulted in the optimal choice of the feature extractor giving sensitivity, specificity, F1-score, accuracy, and AUC of 96.97, 98.85, 96.71, 96.71, and 0.99, respectively. The proposed hybrid models showed improvement in accuracy of 2% and 4% over the pre-trained models in TCGA-LIHC and KMC databases.

肝癌诊断深度学习病理图像医学影像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。