Conference paper Open Access

ensmallen: a flexible C++ library for efficient function optimization

Bhardwaj, Shikhar; Curtin, Ryan R.; Edel, Marcus; Mentekidis, Yannis; Sanderson, Conrad

We present ensmallen, a fast and flexible C++ library for mathematical optimization of arbitrary user-supplied functions, which can be applied to many machine learning problems. Several types of optimizations are supported, including differentiable, separable, constrained, and categorical objective functions. The library provides many pre-built optimizers (including numerous variants of SGD and Quasi-Newton optimizers) as well as a flexible framework for implementing new optimizers and objective functions. Implementation of a new optimizer requires only one method and a new objective function requires typically one or two C++ functions. This can aid in the quick implementation and prototyping of new machine learning algorithms. Due to the use of C++ template metaprogramming, ensmallen is able to support compiler optimizations that provide fast runtimes. Empirical comparisons show that ensmallen is able to outperform other optimization frameworks (like Julia and SciPy), sometimes by large margins. The library is distributed under the BSD license and is ready for use in production environments.

 

Published in: Workshop on Systems for ML and Open Source Software at NIPS / NeurIPS, 2018. http://learningsys.org/nips18/
Files (331.6 kB)
Name Size
ensmallen_2018.pdf
md5:a64623f6c3d8b8acbea87b52d24950d7
331.6 kB Download
127
59
views
downloads
All versions This version
Views 127127
Downloads 5959
Data volume 19.6 MB19.6 MB
Unique views 112112
Unique downloads 5252

Share

Cite as