Tsendsuren Munkhdalai, Xingdi Yuan, Soroush Mehri, Adam Trischler
We describe a mechanism by which artificial neural networks can learn rapid adaptation - the ability to adapt on the fly, with little data, to new tasks - that we call conditionally shifted neurons. We apply this mechanism in the framework of metalearning, where the aim is to replicate some of the flexibility of human learning in machines. Conditionally shifted neurons modify their activation values with task-specific shifts retrieved from a memory module, which is populated rapidly based on limited task experience. On metalearning benchmarks from the vision and language domains, models augmented with conditionally shifted neurons achieve state-of-the-art results.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Image Classification | OMNIGLOT - 1-Shot, 5-way | Accuracy | 98.42 | adaCNN (DF) |
| Image Classification | OMNIGLOT - 5-Shot, 5-way | Accuracy | 99.37 | adaCNN (DF) |
| Image Classification | Mini-Imagenet 5-way (5-shot) | Accuracy | 71.94 | adaResNet (DF) |
| Image Classification | Mini-Imagenet 5-way (1-shot) | Accuracy | 56.88 | adaResNet (DF) |
| Few-Shot Image Classification | OMNIGLOT - 1-Shot, 5-way | Accuracy | 98.42 | adaCNN (DF) |
| Few-Shot Image Classification | OMNIGLOT - 5-Shot, 5-way | Accuracy | 99.37 | adaCNN (DF) |
| Few-Shot Image Classification | Mini-Imagenet 5-way (5-shot) | Accuracy | 71.94 | adaResNet (DF) |
| Few-Shot Image Classification | Mini-Imagenet 5-way (1-shot) | Accuracy | 56.88 | adaResNet (DF) |