Research on MedMNIST medical image classification and recognition method using improved Bayesian neural network

Hui Li; Shuqiu Tan

doi:10.1117/12.3034046

16 October 2024 Research on MedMNIST medical image classification and recognition method using improved Bayesian neural network

Hui Li, Shuqiu Tan

Author Affiliations +

Proceedings Volume 13291, Ninth International Symposium on Advances in Electrical, Electronics, and Computer Engineering (ISAEECE 2024); 132915K (2024) https://doi.org/10.1117/12.3034046
Event: Ninth International Symposium on Advances in Electrical, Electronics, and Computer Engineering (ISAEECE 2024), 2024, Changchun, China

Abstract

Rapid lesion identification in medical images is crucial for early warning and treatment of conditions. Aiming at the problem that images vary and it is challenging to establish a robust uncertainty estimation model, a MedMNIST medical image classification and recognition method based on Bayesian theory is proposed. This method uses ResNet18 as the baseline network, and some units use Bayesian convolutional layers and Bayesian fully connected layers to efficiently extract discriminative features, suppress background interference, and improve model training quality; the ResNet18_BNN framework is introduced to combine the weights in the model with The convolution kernel parameters are transformed from point estimation to probability distribution, and a comprehensive cost and risk loss function is designed for optimization; By comparing the performance of models such as ResNet50, MobileNetV2, MobileViT-XXS and ConvNeXt-T on the MedMNIST data set, quantitative and qualitative analysis, The generalization ability of the model was verified. Experimental results show that the improved model has an accuracy of 0.895 in some data sets, which is 0.052 higher than ResNet18, 0.038 higher than ResNet50, and 0.077, 0.061, and 0.063 higher than the lightweight SOTA detection algorithms MobileNet, MoBileViT, and ConvNeXt respectively. It shows that the algorithm has high accuracy, effectively improves the recognition accuracy, and provides ideas for model selection and performance optimization for subsequent work.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Hui Li and Shuqiu Tan "Research on MedMNIST medical image classification and recognition method using improved Bayesian neural network", Proc. SPIE 13291, Ninth International Symposium on Advances in Electrical, Electronics, and Computer Engineering (ISAEECE 2024), 132915K (16 October 2024); https://doi.org/10.1117/12.3034046

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

;

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
8 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Education and training

Neural networks

Data modeling

Medical imaging

Performance modeling

Image classification

Detection and tracking algorithms

Show All Keywords

Keywords/Phrases

Search In:

Publication Years