skip to main content
10.1145/3663976.3663992acmotherconferencesArticle/Chapter ViewAbstractPublication PagescvipprConference Proceedingsconference-collections
research-article

Stacking algorithm based on naive Bayes

Published: 27 June 2024 Publication History

Abstract

With the advent of the data era, medical image data plays an increasingly important role in diagnosing and treating diseases. This study combined deep learning, ensemble learning, and mathematical theory to construct a Stacking algorithm model based on the Naive Bayes method to achieve more accurate classification of medical images. In the Stacking algorithm, the results of the primary classifier are treated equally, and the performance of the primary classifier is ignored. The poor performance of the primary classifier will affect the final classification result. To solve this problem, a Stacking algorithm based on the Naive Bayes method is proposed in this paper. Firstly, the performance of the primary classifier is evaluated, and the output of the primary classifier is selected reasonably through the evaluation results. Secondly, the VGGNet, InceptionNet, and ResNet convolutional networks were used as primary classifiers to construct a complete Stacking algorithm based on the Naive Bayes method. Next, the algorithm was tested and validated using COVID-19 and NoCOVID-19 lung CT image data. Finally, it is compared with a single primary classifier and the traditional Stacking algorithm. The experimental results show that the Stacking algorithm based on the Naive Bayes method is suitable for the binary classification task of medical images and has better classification results than a single primary classifier and the traditional Stacking algorithm.

References

[1]
Wang Yue, Wang Weidong, Zhao Lei, Research on multi-classification system for automatic recognition of gastroscopy images based on transfer learning. China Medical Equipment,2021, 36(03):81-83.
[2]
Yang Weiting, Li Baoxiang, Zuo Wenbin. Image processing method based on machine vision [J]. Information Technology and Informatization,2021,(07):143-145.
[3]
Sun Fuquan, Cong Chenglong, Zhang Kun, Classification of breast cancer pathological medical images based on multi-model CNN [J]. Small Microcomputer System,2020,41(04):732-735.
[4]
Alex Krizhevsky,Ilya Sutskever,etal. ImageNet Classification with Deep Convolutional Neural Networks[J].NIPS,2012:3-7.
[5]
Karen Simonyan, Andrew Zisserman. Very Deep Convolutional Networks for Large-Scale Image Recognition[J]. 2015:2-6.
[6]
Alhudhaif Adi, Polat Kemal, Karaman Onur. Determination of COVID-19 pneumonia based on generalized convolutional neural network model from chest X-ray images[J]. Expert Systems With Applications,2021,180:3-6.
[7]
FU Yu. Research on image fusion based on deep learning [D]. Jiangnan University,2021.
[8]
Varun Gulshan, Lily Peng, Marc Coram,et al. Development and Validation of a Deep Learning Algorithm for Detection of Diabetic Retinopathy in Retinal Fundus Photographs[J]. 2016,316(22):1-8.
[9]
Wang Yao, Li Wei, Wu Kehe, Cui Wenchao. Application of GBDT and LR Fusion Model in Encrypted Traffic Identification [J]. Computer & Modernization,2020(03):93-98.
[10]
He K, Zhang X, Ren S, Deep Residual Learning for Image Recognition[J]. IEEE, 2016:1- 8.
[11]
Esteva A, Kuprel B, Novoa R A, Dermatologist-level classification of skin cancer with deep neural Networks[J].Nature, 2017, 542:115-118.
[12]
Longshuai Sheng, Li Ce, Li Xin. Mammography classification based on attention mechanism [J]. Computer Engineering and Application,2020,56(08):166-170.
[13]
Lopez A R, Burdick J, Skin lesion classification from dermoscopic images using deep learning techniques[C].Proceedings of 2017 13th IASTED International Conference on Biomedical Engineering (BioMed), 2017: 49-54.
[14]
Xu X W, Jiang X G, Ma C L, A deep learning system to screen novel coronavirus disease 2019 pneu-monia[J].Engineering, 2020, 6(10):1122-1129.
[15]
Liu Y, Chen X, Wang Z, Deep learning for pixel-level image fusion: Recent advances and future prospects[J].Information Fusion, 2018, 42:158-173.
[16]
Srivastava R K, Greff K,et al. Training Very Deep Networks[J]. Computer Science, 2015.3-6.
[17]
Zhang Xu, Wu Peiwang, Qiao Feng. Data mining and bioinformatics analysis of tuberculosis gene based on three Bayesian methods [J]. Journal of Engineering Mathematics,2021,38(05):601-609.
[18]
Sergey Ioffe, Christian Szegedy. Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift [J]. CoRR, 2015, abs / 1502.03167.3-4.
[19]
Jabra M B, Koubaa A, Benjdira B, COVID-19 Diagnosis in Chest X-rays Using Deep Learning and Majority Voting[J]. Applied Sciences, 2021, 11 (6) : 2884-2884.
[20]
Zhou Zhihua. Machine Learning [M]. Beijing: Tsinghua University Press,2016.171-181
[21]
Szegedy C, Vanhoucke V, Ioffe S, Rethinking the Inception Architecture for Computer Vision[J]. IEEE, 2016:2818-2826.
[22]
Qiu Chen. Research on Bayesian learning algorithm and its application [D]. China University of Geosciences,2020.

Recommendations

Comments

Information & Contributors

Information

Published In

cover image ACM Other conferences
CVIPPR '24: Proceedings of the 2024 2nd Asia Conference on Computer Vision, Image Processing and Pattern Recognition
April 2024
373 pages
ISBN:9798400716607
DOI:10.1145/3663976
Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected].

Publisher

Association for Computing Machinery

New York, NY, United States

Publication History

Published: 27 June 2024

Permissions

Request permissions for this article.

Check for updates

Author Tags

  1. Convolutional neural network
  2. Medical image
  3. Naive Bayes method
  4. Stacking algorithm

Qualifiers

  • Research-article
  • Research
  • Refereed limited

Funding Sources

  • Key R&D Program of the Scientific Research Department
  • Beijing Natural Science Foundation
  • National Natural Science Foundation of China

Conference

CVIPPR 2024

Acceptance Rates

Overall Acceptance Rate 14 of 38 submissions, 37%

Contributors

Other Metrics

Bibliometrics & Citations

Bibliometrics

Article Metrics

  • 0
    Total Citations
  • 26
    Total Downloads
  • Downloads (Last 12 months)26
  • Downloads (Last 6 weeks)8
Reflects downloads up to 09 Jan 2025

Other Metrics

Citations

View Options

Login options

View options

PDF

View or Download as a PDF file.

PDF

eReader

View online with eReader.

eReader

HTML Format

View this article in HTML Format.

HTML Format

Media

Figures

Other

Tables

Share

Share

Share this Publication link

Share on social media