ATTL: An Automated Targeted Transfer Learning with Deep Neural Networks
Sayyed Farid Ahamed, Priyanka Aggarwal, Sachin S. Shetty, Erin Lanus, Laura Freeman · 2021 IEEE Global Communications Conference (GLOBECOM) · 2021
Success of machine learning algorithms hinges on access to labeled dataset. Obtaining a labeled dataset is an expensive, challenging and time-consuming process, leading to the development of transfer learning (TL) methodology. TL incorporates gained knowledge from a previously trained source model into specific yet similar task models with limited data domain coverage. In this paper, we propose an automated targeted transfer learning (ATTL) method to resolve the transferability between source and target with minimal data requirements. The ATTL method decides how much target data is essential for model training, along with selected source data, to obtain the skateholder's specified performance metrics. The ATTL framework optimizes the system to select minimal target data based on two approaches: combinatorial coverage and adaptive selection methodology, along with specific source data for fine-tuning given a pre-trained source model. We evaluated the ATTL method on the Kaggle's ‘planes in satellite imagery’ dataset and the results identified that acquiring a small number of intentionally well-chosen samples from the target environment can achieve model performance of 97% in comparison to the baseline transfer learning accuracy of 92%.