1,046 research outputs found
Towards Oracle Knowledge Distillation with Neural Architecture Search
We present a novel framework of knowledge distillation that is capable of
learning powerful and efficient student models from ensemble teacher networks.
Our approach addresses the inherent model capacity issue between teacher and
student and aims to maximize benefit from teacher models during distillation by
reducing their capacity gap. Specifically, we employ a neural architecture
search technique to augment useful structures and operations, where the
searched network is appropriate for knowledge distillation towards student
models and free from sacrificing its performance by fixing the network
capacity. We also introduce an oracle knowledge distillation loss to facilitate
model search and distillation using an ensemble-based teacher model, where a
student network is learned to imitate oracle performance of the teacher. We
perform extensive experiments on the image classification datasets---CIFAR-100
and TinyImageNet---using various networks. We also show that searching for a
new student model is effective in both accuracy and memory size and that the
searched models often outperform their teacher models thanks to neural
architecture search with oracle knowledge distillation.Comment: accepted by AAAI-2
Classifying Options for Deep Reinforcement Learning
In this paper we combine one method for hierarchical reinforcement learning -
the options framework - with deep Q-networks (DQNs) through the use of
different "option heads" on the policy network, and a supervisory network for
choosing between the different options. We utilise our setup to investigate the
effects of architectural constraints in subtasks with positive and negative
transfer, across a range of network capacities. We empirically show that our
augmented DQN has lower sample complexity when simultaneously learning subtasks
with negative transfer, without degrading performance when learning subtasks
with positive transfer.Comment: IJCAI 2016 Workshop on Deep Reinforcement Learning: Frontiers and
Challenge
- …