Developing New Fully Connected Layers for Convolutional Neural Networks with Hyperparameter Optimization for Improved Multi-Label Image Classification

Katona, Tamás and Tóth, Gábor and Petró, Mátyás and Harangi, Balázs (2024) Developing New Fully Connected Layers for Convolutional Neural Networks with Hyperparameter Optimization for Improved Multi-Label Image Classification. Mathematics, 12 (6). p. 806. ISSN 2227-7390

[thumbnail of mathematics-12-00806.pdf] Text
mathematics-12-00806.pdf - Published Version

Download (3MB)

Abstract

Chest X-ray evaluation is challenging due to its high demand and the complexity of diagnoses. In this study, we propose an optimized deep learning model for the multi-label classification of chest X-ray images. We leverage pretrained convolutional neural networks (CNNs) such as VGG16, ResNet 50, and DenseNet 121, modifying their output layers and fine-tuning the models. We employ a novel optimization strategy using the Hyperband algorithm to efficiently search the hyperparameter space while adjusting the fully connected layers of the CNNs. The effectiveness of our approach is evaluated on the basis of the Area Under the Receiver Operating Characteristic Curve (AUC-ROC) metric. Our proposed methodology could assist in automated chest radiograph interpretation, offering a valuable tool that can be used by clinicians in the future.

Item Type: Article
Subjects: Science Repository > Multidisciplinary
Depositing User: Managing Editor
Date Deposited: 09 Mar 2024 04:59
Last Modified: 09 Mar 2024 04:59
URI: http://research.manuscritpub.com/id/eprint/4014

Actions (login required)

View Item
View Item