arXiv:2509.08499cs.LG2025-09

对比10种优化器,找出心脏病预测最佳训练方案

Heart Disease Prediction: A Comparative Study of Optimisers Performance in Deep Neural Networks

  • 统一训练流程,横向比较10种优化器性能
  • RMSProp综合表现最优,精度0.765,召回0.827,AUC 0.841
  • 适合追求效率与平衡的医疗预测场景

优化算法是深度学习模型训练中的关键因素,但优化器的选择尚缺乏系统评估。本文在Kaggle心脏病数据集上,对10种优化器在简单多层感知机模型上的表现进行对比,采用一致训练范式,评估收敛速度、稳定性及分类指标(如AUC、精确率、召回率)。结果表明,Adagrad和Adadelta更稳定但收敛较慢。在各项指标中,RMSProp表现最优:精确率0.765,召回率0.827,AUC达0.841,且训练更快,虽非最稳定。建议在计算资源充足时,通过全面评估选择优化器,以提升模型训练的科学性与性能。

原文摘要 · Abstract (English)

Optimization has been an important factor and topic of interest in training deep learning models, yet less attention has been given to how we select the optimizers we use to train these models. Hence, there is a need to dive deeper into how we select the optimizers we use for training and the metrics that determine this selection. In this work, we compare the performance of 10 different optimizers in training a simple Multi-layer Perceptron model using a heart disease dataset from Kaggle. We set up a consistent training paradigm and evaluate the optimizers based on metrics such as convergence speed and stability. We also include some other Machine Learning Evaluation metrics such as AUC, Precision, and Recall, which are central metrics to classification problems. Our results show that there are trade-offs between convergence speed and stability, as optimizers like Adagrad and Adadelta, which are more stable, took longer time to converge. Across all our metrics, we chose RMSProp to be the most effective optimizer for this heart disease prediction task because it offered a balanced performance across key metrics. It achieved a precision of 0.765, a recall of 0.827, and an AUC of 0.841, along with faster training time. However, it was not the most stable. We recommend that, in less compute-constrained environments, this method of choosing optimizers through a thorough evaluation should be adopted to increase the scientific nature and performance in training deep learning models.

心脏病预测优化器对比深度学习分类评估

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。