Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

An Evolutionary Approach to Dynamic Introduction of Tasks in Large-scale Multitask Learning Systems

About

Multitask learning assumes that models capable of learning from multiple tasks can achieve better quality and efficiency via knowledge transfer, a key feature of human learning. Though, state of the art ML models rely on high customization for each task and leverage size and data scale rather than scaling the number of tasks. Also, continual learning, that adds the temporal aspect to multitask, is often focused to the study of common pitfalls such as catastrophic forgetting instead of being studied at a large scale as a critical component to build the next generation artificial intelligence.We propose an evolutionary method capable of generating large scale multitask models that support the dynamic addition of new tasks. The generated multitask models are sparsely activated and integrates a task-based routing that guarantees bounded compute cost and fewer added parameters per task as the model expands.The proposed method relies on a knowledge compartmentalization technique to achieve immunity against catastrophic forgetting and other common pitfalls such as gradient interference and negative transfer. We demonstrate empirically that the proposed method can jointly solve and achieve competitive results on 69public image classification tasks, for example improving the state of the art on a competitive benchmark such as cifar10 by achieving a 15% relative error reduction compared to the best model trained on public data.

Andrea Gesmundo, Jeff Dean• 2022

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (test)--
3518
Image ClassificationCIFAR-10 (test)--
3381
Image ClassificationVTAB 1k (test)--
121
Image ClassificationVTAB-1K 1.0 (test)--
102
Handwritten Character ClassificationEMNIST-Letters (test)
Accuracy93.68
15
Handwritten Character ClassificationEMNIST-Digits (test)
Accuracy99.82
12
Image ClassificationK-MNIST (test)
Accuracy98.68
12
ClassificationOmniglot (test)
Top-1 Acc99.84
11
Multitask Character ClassificationMultitask Character Classification Benchmark (MCCB) (test)
Max Accuracy92.98
4
Visual Domain ClassificationVisual Domain Decathlon Benchmark (VDDB) (test)
Max Test Accuracy83.58
4
Showing 10 of 14 rows

Other info

Code

Follow for update