Skip to content

Adaptive Momentum Coefficient for Neural Network Optimization

License

Notifications You must be signed in to change notification settings

zanarashidi/AMoC

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AMoC (Adaptive Momentum Coefficient for Neural Network Optimization)

About

The code accompanying our ECML-PKDD 2020 paper Adaptive Momentum Coefficient for Neural Network Optimization.

Summary

Adaptive Momentum Coefficient (AMoC) utilizes the inner product of the gradient and the previous update to the parameters, to effectively control the amount of weight put on the momentum term based on the change of direction in the optimization path. It is easy to implement and its computational overhead over momentum methods is negligible. Extensive empirical results on both convex and neural network objectives show that AMoC performs well in practise and compares favourably with other first and second-order optimization algorithms.

Contents

The repository contains the implementation of the AMoC optimizer along with the Deep Autoencoder experiments on the MNIST dataset included in the paper.

Hyperparameters

Variables and hyperparameters including learning rate (ε), beta (β) and momentum (μ) can be modified in the scripts.

Requirements

The requirements are Python 3.8, Pytorch 1.4.0 along with CUDA 10.1 and CuDNN 7.6.3.

About

Adaptive Momentum Coefficient for Neural Network Optimization

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages