FIESTA : Fast IdEntification of State-of-The-Art models using adaptive bandit algorithms

Moss, Henry B. and Moore, Andrew and Leslie, David S. and Rayson, Paul (2019) FIESTA : Fast IdEntification of State-of-The-Art models using adaptive bandit algorithms. In: Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics :. Association for Computational Linguistics, ITA, pp. 2920-2930.

[thumbnail of 1906.12230v1]
Text (1906.12230v1)
1906.12230v1.pdf - Accepted Version

Download (398kB)


We present FIESTA, a model selection approach that significantly reduces the computational resources required to reliably identify state-of-the-art performance from large collections of candidate models. Despite being known to produce unreliable comparisons, it is still common practice to compare model evaluations based on single choices of random seeds. We show that reliable model selection also requires evaluations based on multiple train-test splits (contrary to common practice in many shared tasks). Using bandit theory from the statistics literature, we are able to adaptively determine appropriate numbers of data splits and random seeds used to evaluate each model, focusing computational resources on the evaluation of promising models whilst avoiding wasting evaluations on models with lower performance. Furthermore, our user-friendly Python implementation produces confidence guarantees of correctly selecting the optimal model. We evaluate our algorithms by selecting between 8 target-dependent sentiment analysis methods using dramatically fewer model evaluations than current model selection approaches.

Item Type:
Contribution in Book/Report/Proceedings
?? machine learningmodel selectionmulti-armed banditnlp ??
ID Code:
Deposited By:
Deposited On:
16 Jul 2019 14:00
Last Modified:
16 Jul 2024 04:39