Duality Diagram Similarity: a generic framework for initialization selection in task transfer learning
This addresses the challenge of model selection in transfer learning for practitioners, though it is incremental as it builds on existing methods with a new similarity measure.
The paper tackles the problem of selecting a pre-trained model initialization for transfer learning by proposing Duality Diagram Similarity (DDS), a framework that efficiently predicts transfer performance rankings with high correlation (0.86) and outperforms state-of-the-art methods by 10% on the Taskonomy benchmark.
In this paper, we tackle an open research question in transfer learning, which is selecting a model initialization to achieve high performance on a new task, given several pre-trained models. We propose a new highly efficient and accurate approach based on duality diagram similarity (DDS) between deep neural networks (DNNs). DDS is a generic framework to represent and compare data of different feature dimensions. We validate our approach on the Taskonomy dataset by measuring the correspondence between actual transfer learning performance rankings on 17 taskonomy tasks and predicted rankings. Computing DDS based ranking for $17\times17$ transfers requires less than 2 minutes and shows a high correlation ($0.86$) with actual transfer learning rankings, outperforming state-of-the-art methods by a large margin ($10\%$) on the Taskonomy benchmark. We also demonstrate the robustness of our model selection approach to a new task, namely Pascal VOC semantic segmentation. Additionally, we show that our method can be applied to select the best layer locations within a DNN for transfer learning on 2D, 3D and semantic tasks on NYUv2 and Pascal VOC datasets.