基于AdamW优化算法的胃肠道图像分类应用

邵一川; 张鹏程; 武志军; 孙海静; 赵骞; 张乐

doi:10.16103/j.cnki.21-1583/n.2025.05.007

沈阳大学学报（自然科学版） ›› 2025, Vol. 37 ›› Issue (05) : 399 -406. DOI: 10.16103/j.cnki.21-1583/n.2025.05.007

基于AdamW优化算法的胃肠道图像分类应用

邵一川, 张鹏程, 武志军, 孙海静, 赵骞, 张乐

作者信息 +

Author information +

文章历史 +

摘要

为提高胃肠道图像分类的准确性和稳定性，提出一种基于AdamW的改进优化算法AdamW＿snip。该算法在AdamW的更新过程中引入Hessian矩阵估计与Nesterov动量机制，以进一步提升医学图像分类的性能。为验证AdamW＿snip的优化效果，开展了多组对比实验，与Adam、SGD、AdamW、AdaGrad和RMSProp等优化算法进行比较，并统一采用ResNet18网络架构。实验结果表明，在胃肠道图像测试集上，AdamW＿snip优化算法取得了最高的分类准确率(79.50%)和最低的损失值(0.665 1)。