Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Meta-Learning Initializations for Image Segmentation
(Submitted on 13 Dec 2019 (v1), last revised 7 May 2020 (this version, v4))
Abstract: We extend first-order model agnostic meta-learning algorithms (including FOMAML and Reptile) to image segmentation, present a novel neural network architecture built for fast learning which we call EfficientLab, and leverage a formal definition of the test error of meta-learning algorithms to decrease error on out of distribution tasks. We show state of the art results on the FSS-1000 dataset by meta-training EfficientLab with FOMAML and using Bayesian optimization to infer the optimal test-time adaptation routine hyperparameters. We also construct a small benchmark dataset, FP-k, for the empirical study of how meta-learning systems perform in both few- and many-shot settings. On the FP-k dataset, we show that meta-learned initializations provide value for canonical few-shot image segmentation but their performance is quickly matched by conventional transfer learning with performance being equal beyond 10 labeled examples. Our code, meta-learned model, and the FP-k dataset are available at this https URL .
Submission history
From: Sean Hendryx [view email][v1] Fri, 13 Dec 2019 01:58:36 GMT (1191kb,D)
[v2] Sun, 12 Apr 2020 06:44:55 GMT (4140kb,D)
[v3] Sun, 19 Apr 2020 20:52:49 GMT (4140kb,D)
[v4] Thu, 7 May 2020 23:33:07 GMT (4140kb,D)
Link back to: arXiv, form interface, contact.