关键词: 机器学习, 恶性肿瘤, 辅助诊断, 特征选择, 特征降维, 分类器, 层次聚类, 神经网络

Abstract:

In the field of tumor diagnosis, the artificial intelligence-assisted diagnosis system can accurately distinguish and diagnose tumor attributes and malignant tumor stages, thereby prolonging the survival time of patients. In this study, taking breast tumor as a case, in view of the over-fitting problem caused by the excessive amount of data in the feature extraction process, a supervised learning artificial intelligence-assisted diagnosis model is proposed. When extracting features, hierarchical clustering analysis is introduced to perform effective feature reduction, and the classified feature data is used as the feature input of the artificial neural network model to achieve effective training of the classifier. The experimental results show that compared with other algorithm, the accuracy and AUC value of the proposed algorithm are improved, indicating that the model can not only solve the over-fitting problem caused by the description of massive feature regions, but also enhance the artificial intelligence-assisted diagnosis, thereby completing the mammography target breast tumor high-precision distinction.

Key words: machine learning, malignant tumor, auxiliary diagnosis, feature selection, feature dimension reduction, classifier, hierarchical clustering, neural network