Continual and Multi-Task Architecture Search

Ramakanth Pasunuru, Mohit Bansal


Abstract
Architecture search is the process of automatically learning the neural model or cell structure that best suits the given task. Recently, this approach has shown promising performance improvements (on language modeling and image classification) with reasonable training speed, using a weight sharing strategy called Efficient Neural Architecture Search (ENAS). In our work, we first introduce a novel continual architecture search (CAS) approach, so as to continually evolve the model parameters during the sequential training of several tasks, without losing performance on previously learned tasks (via block-sparsity and orthogonality constraints), thus enabling life-long learning. Next, we explore a multi-task architecture search (MAS) approach over ENAS for finding a unified, single cell structure that performs well across multiple tasks (via joint controller rewards), and hence allows more generalizable transfer of the cell structure knowledge to an unseen new task. We empirically show the effectiveness of our sequential continual learning and parallel multi-task learning based architecture search approaches on diverse sentence-pair classification tasks (GLUE) and multimodal-generation based video captioning tasks. Further, we present several ablations and analyses on the learned cell structures.
Anthology ID:
P19-1185
Volume:
Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics
Month:
July
Year:
2019
Address:
Florence, Italy
Editors:
Anna Korhonen, David Traum, Lluís Màrquez
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
1911–1922
Language:
URL:
https://aclanthology.org/P19-1185/
DOI:
10.18653/v1/P19-1185
Bibkey:
Cite (ACL):
Ramakanth Pasunuru and Mohit Bansal. 2019. Continual and Multi-Task Architecture Search. In Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics, pages 1911–1922, Florence, Italy. Association for Computational Linguistics.
Cite (Informal):
Continual and Multi-Task Architecture Search (Pasunuru & Bansal, ACL 2019)
Copy Citation:
PDF:
https://aclanthology.org/P19-1185.pdf
Video:
 https://aclanthology.org/P19-1185.mp4
Code
 ramakanth-pasunuru/CAS-MAS
Data
DiDeMoGLUEMSR-VTTMSVDQNLI