中北大学学报（自然科学版）2024，Vol.45Issue(4)：428-438,11.DOI:10.3969/j.issn.1673-3193.2024.04.003
基于改进的深度学习模型分类乳腺癌分子亚型
Improved Deep Learning Models to Classify Breast Cancer Molecular Subtypes
摘要
Abstract
Due to the high heterogeneity of breast cancer,different molecular subtypes of breast cancer have great differences in the treatment and diagnosis.Therefore,it is of important research significance to improve the accuracy of the diagnosis of breast cancer subtypes,thus to further reduce the misdiagnosis rate of breast cancer and avoid overtreatment of breast cancer.The molecular subtypes of Luminal A and non-Luminal A of breast cancer were classified based on deep learning algorithm for MRI breast cancer medical images in TCIA database.In order to compare the superiority of deep learning algorithm in classi-fication of breast cancer molecular subtypes,gene expression data of TCGA database was used to classify breast cancer molecular subtypes.In the same breast cancer MRI image database,the classification of breast cancer molecular subtypes was studied based on traditional machine learning algorithm.Compara-tive studies between a variety of deep learning models were conducted in the processing of MRI images with deep learning algorithm.Fine-tuning VGG16 network and frozen convolution layer were imple-mented.An improved VGG16 network model was proposed,and a densenet network module was added,namely VGG16+densenet(4)model.The accuracy of the improved VGG16 network model was 0.96,and the AUC was 0.97.The classification accuracy of gene expression data was 0.73,and the AUC was 0.79.The classification accuracy and AUC of traditional machine learning reached 0.80 and 0.87 respec-tively.The experimental results show that the proposed VGG16+densenet(4)model improves the accu-racy of the molecular subtypes of breast cancer and has better classification effect.关键词
深度学习/乳腺癌分子亚型/传统机器学习/数据增强/基因表达数据Key words
deep learning/breast cancer molecular subtypes/traditional machine learning/data augmenta-tion/gene expression data分类
信息技术与安全科学引用本文复制引用
段美华,焦雄..基于改进的深度学习模型分类乳腺癌分子亚型[J].中北大学学报（自然科学版）,2024,45(4):428-438,11.基金项目
山西省自然科学基金资助项目(202203021221063) （202203021221063）
