• Home
  • Search
  • Evaluating the performance of hyperparameters for unbiased and fair machine learning
  • Cite Icon1
  • https://doi.org/10.1117/12.3006909Copy DOI Icon

Evaluating the performance of hyperparameters for unbiased and fair machine learning

  • Apr 2, 2024
  • Vy Bui +4 more
Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Biased outcomes in machine learning models can arise due to various factors, including limited training data, imbalanced class distribution, suboptimal training methodologies, and overfitting. In training neural networks with Stochastic Gradient Descent (SGD) and backpropagation, the choice of hyperparameters like learning rate and momentum is crucial to influencing the model’s performance. A comprehensive grid search study was conducted using static hyperparameters with standard SGD and dynamic hyperparameters with the Adam optimizer. The investigation focused on a multifaceted analysis across different tasks —- classification, segmentation, and detection — and was applied to four image-based applications: digit classification using the MNIST dataset, tuberculosis detection from chest X-ray images, lung segmentation in chest X-ray images, and the detection of malaria parasites in blood smear images. In the first comparative study on the MNIST dataset, the SGD algorithm consistently outperformed the Adam algorithm as noise levels increased. SGD held a slight advantage in accuracy over Adam in a noise-free environment. This advantage became more apparent as noise was introduced and increased to moderate levels. At high noise levels, both algorithms experienced a significant decline in performance, yet SGD maintained a relatively better accuracy compared to Adam. This trend underscored SGD’s superior ability to generalize across varying noise conditions. For TB detection in the second application, the DenseNet121 architecture was used, and it was found that Adam showed better performance on the larger TBX11K dataset. However, SGD outperformed Adam on a smaller subset of TBX11K. In the third comparative study for lung segmentation using COVID-19 and NLM datasets, SGD slightly outperformed Adam based on the mean and Hausdorff distances. In the case of malaria parasite detection using the YOLOv8 architecture, SGD and Adam optimizers showed varying performances across different conditions. Initially, both achieved high accuracies on the full dataset, with Adam slightly outperforming SGD. However, with decreasing dataset size, SGD maintained more consistent performance, while Adam’s accuracy fluctuated significantly. In noise tests, both showed equal accuracy on clean data, but under noise conditions, SGD maintained higher accuracies, suggesting better generalization capabilities. Further experiments assessed how well SGD and Adam could cope with domain shifts, mainly when using data from different countries. SGD’s generalization performance was superior to Adam’s performance in these experiments. In summary, although Adam is arguably the more popular of the two optimization techniques, SGD held its own in the experiments and showed superior performance when it came to generalization or classifying in noisy conditions.

Similar Papers
  • Book Chapter

Study to Distinguish Covid-19 from Normal Cases Using Chest X-Ray Images with Convolution Neural Network

  • Nov 23, 2022
  • P Sathish Kumar +2
  • Conference Article
  • Citations1

On a relationship between integral compensation and stochastic gradient descent

  • Sep 01, 2017
  • Yusuke Fujimoto +2
  • Research Article
  • Citations2

Multi-label classification of chest X-ray images with pre-trained vision Transformer model

  • Jan 01, 2023
  • Journal of Image and Graphics
  • Xing Suxia +4
  • Conference Article
  • Citations19

A Multi-label Classification on Topics of Quranic Verses (English Translation) Using Backpropagation Neural Network with Stochastic Gradient Descent and Adam Optimizer

  • Jul 01, 2019
  • Nanang Saiful Huda +2
  • Research Article

Using subspaces of weight matrix for evaluating generative adversarial networks with Fréchet distance

  • Jul 08, 2021
  • Concurrency and Computation: Practice and Experience
  • Enes Eken
  • Conference Article
  • Citations2

Indonesian Clickbait Detection Using Improved Backpropagation Neural Network

  • Dec 16, 2021
  • Bellatasya Unrica Nadia +1
  • Conference Article
  • Citations5

A Novel Convolutional Neural Network Architecture to Diagnose COVID-19

  • May 13, 2021
  • S Thabasum Aara +3
  • Conference Article
  • Citations4

Comparing Adam and SGD optimizers to train AlexNet for classifying GPR C-scans featuring ancient structures

  • Dec 01, 2021
  • Merope Manataki +2
  • PDF
  • Research Article
  • Citations9

Implementation of CNN and ANN for Fashion-MNIST-Dataset using Different Optimizers

  • Dec 20, 2022
  • Indian Journal Of Science And Technology
  • Sumera Sumera +3
  • Research Article

An Innovative Technique for Medical Image Segmentation Using Convolutional Neural Networks Optimized Through Stochastic Gradient Descent

  • Jun 25, 2024
  • International Journal of Artificial Intelligence
  • Mohammad Taheri +2
  • Research Article

Phase diagram and eigenvalue dynamics of stochastic gradient descent in multilayer neural networks

  • Nov 18, 2025
  • Machine Learning: Science and Technology
  • Chanju Park +2
  • Research Article

Transfer Learning Analysis on Tuberculosis Classification Using MobileNetV2 Architecture Based on Chest X-Ray Images

  • Dec 06, 2025
  • Journal of Applied Informatics and Computing
  • Ali Samsul Latupono +1
  • Conference Article
  • Citations2

Examining Hyperparameters of Neural Networks Trained Using Local Search

  • Dec 01, 2019
  • Ahmed Aly +2
  • Conference Article
  • Citations2

Soft-Sign Stochastic Gradient Descent Algorithm for Wireless Federated Learning

  • Sep 27, 2021
  • Seunghoon Lee +4
  • Research Article
  • Citations2

Investigating Quantum Machine Learning Frameworks and Simulating Quantum Approaches

  • Oct 21, 2024
  • The Asian Bulletin of Big Data Management
  • Muhammad Jawad Khan +6
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.