Abstract

Automatic landmark detection in knee x-rays requires an optimizer to improve its performance while shortening training time. The aim of this research is to find a suitable optimizer for the YOLOv8m model for the task. We focus the comparison on six optimizers, including SGD, Adam, AdaMax, AdamW, NAdam, and RAdam. This study used 400 normal Kellgren-Lawrence (KL-0) right knee x-ray images from the Public Osteoarthritis Initiative (OAI) dataset with 5.600 landmark spatial features. Each knee x-ray image in the dataset is labeled with 14 landmarks on the femur and tibia bones. We resized the knee x-ray image as input to a size of 640×640 pixels, then using the YOLOv8m model with batch size 16 and epoch 150, we carried out training and testing with the SGD, Adam, AdaMax, AdamW, NAdam, and RAdam optimization algorithms one by one. We found that the optimizer average precision (mAP) was 0.696 with the Adam optimizer outperforming the others, while the precision of 0.716 and recall of 0.685 with the AdaMax optimizer outperformed the others, and the fastest time was 0.385 hours with the AdamW optimizer.

Original languageEnglish
Title of host publicationICITDA 2023 - Proceedings of the 2023 8th International Conference on Information Technology and Digital Applications
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9798350344691
DOIs
Publication statusPublished - 2023
Event8th International Conference on Information Technology and Digital Applications, ICITDA 2023 - Yogyakarta, Indonesia
Duration: 17 Nov 202318 Nov 2023

Publication series

NameICITDA 2023 - Proceedings of the 2023 8th International Conference on Information Technology and Digital Applications

Conference

Conference8th International Conference on Information Technology and Digital Applications, ICITDA 2023
Country/TerritoryIndonesia
CityYogyakarta
Period17/11/2318/11/23

Keywords

  • Hyperparameter
  • Knee Landmark
  • Optimizer

Fingerprint

Dive into the research topics of 'A Comparison of Optimizer Algorithms in YOLOv8 for Automatic Detection of Knee Landmarks'. Together they form a unique fingerprint.

Cite this