Heuristic optimisation of multi-task dynamic architecture neural network (DAN2)
Published version
Peer-reviewed
Repository URI
Repository DOI
Change log
Authors
Abstract
jats:titleAbstract</jats:title>jats:pThis article proposes a novel method to optimise the Dynamic Architecture Neural Network (DAN2) adapted for a multi-task learning problem. The multi-task learning neural network adopts a multi-head and serial architecture with DAN2 layers acting as the basic subroutine. Adopting a dynamic architecture, the layers are added consecutively starting from a minimal initial structure. The optimisation method adopts an iterative heuristic scheme that sequentially optimises the shared layers and the task-specific layers until the solver converges to a small tolerance. Application of the method has demonstrated the applicability of the algorithm to simulated datasets. Comparable results to Artificial Neural Networks (ANNs) have been obtained in terms of accuracy and speed.</jats:p>
Description
Funder: China Scholarship Council
Funder: China Scholarship Council; doi: http://dx.doi.org/10.13039/501100004543
Funder: BASF Corporation; doi: http://dx.doi.org/10.13039/100007487
Keywords
Journal Title
Conference Name
Journal ISSN
1433-3058