Auto-Transfer: Learning to Route Transferrable Representations

Murugesan, Keerthiram
Sadashivaiah, Vijay
Luss, Ronny
Shanmugam, Karthikeyan
Chen, Pin-Yu
Dhurandhar, Amit
No Thumbnail Available
Other Contributors
Issue Date
Feature routing , Transferable Representations
Terms of Use
Attribution 3.0 United States
Full Citation
Murugesan, K., Sadashivaiah, V., Luss, R., Shanmugam, K., Chen, P. Y., & Dhurandhar, A. (2021, September). Auto-Transfer: Learning to Route Transferable Representations. In International Conference on Learning Representations.
Knowledge transfer between heterogeneous source and target networks and tasks has received a lot of attention in recent times as large amounts of quality labeled data can be difficult to obtain in many applications. Existing approaches typically constrain the target deep neural network (DNN) feature representations to be close to the source DNNs feature representations, which can be limiting. We, in this paper, propose a novel adversarial multi-armed bandit approach that automatically learns to route source representations to appropriate target representations following which they are combined in meaningful ways to produce accurate target models. We see upwards of 5% accuracy improvements compared with the state-of-the-art knowledge transfer methods on four benchmark (target) image datasets CUB200, Stanford Dogs, MIT67, and Stanford40 where the source dataset is ImageNet. We qualitatively analyze the goodness of our transfer scheme by showing individual examples of the important features focused on by our target network at different layers compared with the (closest) competitors. We also observe that our improvement over other methods is higher for smaller target datasets making it an effective tool for small data applications that may benefit from transfer learning.
International Conference on Learning Representations