超参数
卷积神经网络
计算机科学
人工智能
机器学习
估计
人工神经网络
姿势
模式识别(心理学)
工程类
系统工程
作者
Endah Suryawati Ningrum,Eko Mulyanto Yuniarno,Mauridhi Hery Purnomo
标识
DOI:10.1109/isitia59021.2023.10221125
摘要
The infant stage is crucial for human development, with fidgeting playing a key role in the development of balance and coordination. Recent studies have developed machine learning algorithms that utilize body posture estimation to detect fidgety movements in babies. However, research on optimal hyperparameters for infant posture estimation is still limited. Without a reference to the optimal configuration, research on infant-based pose estimation could be prolonged and deviate from its main goal of detecting infant growth through movement.This paper employs a computer vision approach to enhance the accuracy of predicting fidgety movements in babies. Evaluating the hyperparameters of the Convolutional Neural Network (CNN) model for Baby Pose Estimation can significantly improve its performance. The synthetic and real infant pose (SyRIP) dataset, along with the high-resolution net (HRnet) and distribution-aware coordinate representation of keypoin (DARKPose) models, is utilized for the infant pose estimation dataset. The hyperparameter values were exploited to identify the most optimal results in this research. Among the 37 scenarios, the following hyperparameter combinations yielded the best results: Batch Size combinations of 2 and 4, train epochs of 15 and 150, lambda value of 0.0001, learning rate of 0.00005, learning rate factor of 0.1, learning rate steps of 60 and 120, weight decay of 0.00005, gamma of 0.95, and momentum of 0.9. Increasing the epochs and pre-epochs has proven to enhance the model's performance. Lambda values show a positive correlation with model performance. Conversely, values such as Learning Rate and its factor, steps, gamma, momentum, and weight decay demonstrate a negative correlation.
科研通智能强力驱动
Strongly Powered by AbleSci AI