About the Publication

Automated Optimization-Based Deep Learning Models for Image Classification Tasks
AfriAI Lab

Applying deep learning models requires design and optimization when solving multifaceted artificial intelligence tasks. Optimization relies on human expertise and is achieved only with great exertion. The current literature concentrates on automating design; optimization needs more attention. Similarly, most existing optimization libraries focus on other machine learning tasks rather than image classification. For this reason, an automated optimization scheme of deep learning models for image classification tasks is proposed in this paper. A sequential-model-based optimization algorithm was used to implement the proposed method. Four deep learning models, a transformer-based model, and standard datasets for image classification challenges were employed in the experiments. Through empirical evaluations, this paper demonstrates that the proposed scheme improves the performance of deep learning models. Specifically, for a Virtual Geometry Group (VGG-16), accuracy was heightened from 0.937 to 0.983, signifying a 73% relative error rate drop within an hour of automated optimization. Similarly, training-related parameter values are proposed to improve the performance of deep learning models. The scheme can be extended to automate the optimization of transformer-based models. The insights from this study may assist efforts to provide full access to the building and optimization of DL models, even for amateurs.

Automated Optimization-Based Deep Learning Models for Image Classification Tasks

Title: Automated Optimization-Based Deep Learning Models for Image Classification Tasks

Authors: Daudi Mashauri Migayo, Shubi Kaijage, Stephen Swetala, and Devotha G. Nyambo.

Year: 2023

Publisher/Journal: MDPI

Status: Published

More Details

Publication Added: Aug 30, 2025

Mobile App
AfriAI Mobile App
Release v1.0 • 75.3 MB

Download the official Android APK to view live timetable schedules, receive updates, engage in live session chats, and access resources offline.

Offline Agenda Real-Time Feeds