Maximum Principle Based Algorithms for Deep Learning

Tensorflow implementations of the E-MSA algorithm introduced in

Maximum Principle Based Algorithms for Deep Learning. Qianxiao Li, Long Chen, Cheng Tai, Weinan E. Journal of Machine Learning Research 18 165:1–165:29. 2018

URL: http://jmlr.org/papers/v18/17-653.html

Requirements

tensorflow, numpy, matplotlib, pyyaml

Examples

Run algorithm on test problems (configuration file in config.yml)

    python -u main_toy.py | tee test.log
    python -u main_mnist.py | tee mnist.log

Plot training curves

    python plot_logs.py --logdir test.log

License

This project is licensed under the MIT license - see the LICENSE.md file for details