Skip to main content

A Meta-Reinforcement Learning Approach to Optimize Parameters and Hyper-parameters Simultaneously.

Ali, A.R., Budka, M. and Gabrys, B., 2019. A Meta-Reinforcement Learning Approach to Optimize Parameters and Hyper-parameters Simultaneously. In: 16th Pacific Rim International Conference on Artificial Intelligence, 26-30 August 2019, Fiji.

Full text available as:

[img]
Preview
PDF
_Abbas__Meta_RL (1).pdf - Accepted Version
Available under License Creative Commons Attribution Non-commercial.

667kB

Official URL: https://www.pricai.org/2019/10-main-page/11-the-16...

Abstract

In the last few years, we have witnessed a resurgence of interest in neural networks. The state-of-the-art deep neural network architectures are however challenging to design from scratch and requiring computationally costly empirical evaluations. Hence, there has been a lot of research effort dedicated to effective utilisation and adaptation of previously proposed architectures either by using transfer learning or by modifying the original architecture. The ultimate goal of designing a network architecture is to achieve the best possible accuracy for a given task or group of related tasks. Although there have been some efforts to automate network architecture design process, most of the existing solutions are still very computationally intensive. This work presents a framework to automatically find a good set of hyper-parameters resulting in reasonably good accuracy, which at the same time is less computationally expensive than the existing approaches. The idea presented here is to frame the hyper-parameter selection and tuning within the reinforcement learning regime. Thus, the parameters of a meta-learner, RNN, and hyper-parameters of the target network are tuned simultaneously. Our meta-learner is being updated using policy network and simultaneously generates a tuple of hyper-parameters which are utilized by another network. The network is trained on a given task for a number of steps and produces validation accuracy whose delta is used as reward. The reward along with the state of the network, comprising statistics of network’s final layer outcome and training loss, are fed back to the meta-learner which in turn generates a tuned tuple of hyper-parameters for the next time step. Therefore, the effectiveness of a recommended tuple can be tested very quickly rather than waiting for the network to converge. This approach produces accuracy close to the state-of-the-art approach and is found to be comparatively less computationally intensive.

Item Type:Conference or Workshop Item (Paper)
Uncontrolled Keywords:Convolutional Neural Networks; Meta-Learning; Reinforcement Learning; Policy Gradients; Hyper-parameter optimization
Group:Faculty of Science & Technology
ID Code:32529
Deposited By: Symplectic RT2
Deposited On:15 Jul 2019 10:53
Last Modified:14 Mar 2022 14:17

Downloads

Downloads per month over past year

More statistics for this item...
Repository Staff Only -