arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1908.11406cs.LGcs.AI

Learning to Transfer Learn: Reinforcement Learning-Based Selection for Adaptive Transfer Learning

Google Cloud AI(谷歌云AI);University of Technology Sydney(悉尼科技大学)

Linchao Zhu, Sercan O. Arik, Yi Yang, Tomas Pfister

英文摘要:

We propose a novel adaptive transfer learning framework, learning to transfer learn (L2TL), to improve performance on a target dataset by careful extraction of the related information from a source dataset. Our framework considers cooperative optimization of shared weights between models for source and target tasks, and adjusts the constituent loss weights adaptively. The adaptation of the weights is based on a reinforcement learning (RL) selection policy, guided with a performance metric on the target validation set. We demonstrate that L2TL outperforms fine-tuning baselines and other adaptive transfer learning methods on eight datasets. In the regimes of small-scale target datasets and significant label mismatch between source and target datasets, L2TL shows particularly large benefits.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑