A Continuous Optimisation Benchmark Suite from Neural Network Regression
arXiv:2109.05606 · doi:10.1007/978-3-031-14714-2_13
Abstract
Designing optimisation algorithms that perform well in general requires experimentation on a range of diverse problems. Training neural networks is an optimisation task that has gained prominence with the recent successes of deep learning. Although evolutionary algorithms have been used for training neural networks, gradient descent variants are by far the most common choice with their trusted good performance on large-scale machine learning tasks. With this paper we contribute CORNN (Continuous Optimisation of Regression tasks using Neural Networks), a large suite for benchmarking the performance of any continuous black-box algorithm on neural network training problems. Using a range of regression problems and neural network architectures, problem instances with different dimensions and levels of difficulty can be created. We demonstrate the use of the CORNN Suite by comparing the performance of three evolutionary and swarm-based algorithms on over 300 problem instances, showing evidence of performance complementarity between the algorithms. As a baseline, the performance of the best population-based algorithm is benchmarked against a gradient-based approach. The CORNN suite is shared as a public web repository to facilitate easy integration with existing benchmarking platforms.
This preprint has not undergone peer review or any post-submission improvements or corrections. The Version of Record of this contribution is published in Parallel Problem Solving from Nature - PPSN XVII, Lecture Notes in Computer Science, Vol 13398, and is available online at https://doi.org/10.1007/978-3-031-14714-2_13
References in corpus (5)
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- The Loss Surfaces of Multilayer Networks
- Identifying and attacking the saddle point problem in high-dimensional non-convex optimization
- What Matters In On-Policy Reinforcement Learning? A Large-Scale Empirical Study
- Towards Realistic Optimization Benchmarks: A Questionnaire on the Properties of Real-World Problems