An Effective Optimization Method for Machine Learning Based on ADAM

Applied Sciences

Sangmin Ji¹  Jaehyun Ahn¹  Dokkyun Yi² 

[1] Department of Mathematics, College of Natural Sciences, Chungnam National University, Daejeon 34134, Korea;Division of Creative Integrated General Studies, Daegu University College, Kyungsan 38453, Korea;

关键词: numerical optimization;    adam;    machine learning;    stochastic gradient methods;   

DOI  :  10.3390/app10031073

来源: DOAJ

【 摘 要 】

A machine is taught by finding the minimum value of the cost function which is induced by learning data. Unfortunately, as the amount of learning increases, the non-liner activation function in the artificial neural network (ANN), the complexity of the artificial intelligence structures, and the cost function’s non-convex complexity all increase. We know that a non-convex function has local minimums, and that the first derivative of the cost function is zero at a local minimum. Therefore, the methods based on a gradient descent optimization do not undergo further change when they fall to a local minimum because they are based on the first derivative of the cost function. This paper introduces a novel optimization method to make machine learning more efficient. In other words, we construct an effective optimization method for non-convex cost function. The proposed method solves the problem of falling into a local minimum by adding the cost function in the parameter update rule of the ADAM method. We prove the convergence of the sequences generated from the proposed method and the superiority of the proposed method by numerical comparison with gradient descent (GD, ADAM, and AdaMax).

【 授权许可】

Unknown