Knowledge distillation via softmax regression representation learning
May 3, 2021
Speakers
Brais Martinez
Speaker · 0 followers
Adrian Bulat
Speaker · 0 followers
Georgios Tzimiropoulos
Speaker · 0 followers
About
This paper addresses the problem of model compression via knowledge distillation. We advocate for a method that optimizes the output feature of the penultimate layer of the student network and hence is directly related to representation learning. Previous distillation methods which typically impose direct feature matching between the student and the teacher do not take into account the classification problem at hand. On the contrary, our distillation method decouples representation learning and classification and utilizes the teacher's pre-trained classifier to train the student's penultimate layer feature. In particular, for the same input image, we wish the teacher's and student's feature to produce the same output when passed through the teacher's classifier which is achieved with a simple $L_2$ loss. Our method is extremely simple to implement and straightforward to train and is shown to consistently outperform previous state-of-the-art methods over a large set of experimental settings including different (a) network architectures, (b) teacher-student capacities, (c) datasets, and (d) domains.This paper addresses the problem of model compression via knowledge distillation. We advocate for a method that optimizes the output feature of the penultimate layer of the student network and hence is directly related to representation learning. Previous distillation methods which typically impose direct feature matching between the student and the teacher do not take into account the classification problem at hand. On the contrary, our distillation method decouples representation learning and…
Categories
AI & Data Science
Category · 10.8k presentations
About ICLR 2021
The International Conference on Learning Representations (ICLR) is the premier gathering of professionals dedicated to the advancement of the branch of artificial intelligence called representation learning, but generally referred to as deep learning. ICLR is globally renowned for presenting and publishing cutting-edge research on all aspects of deep learning used in the fields of artificial intelligence, statistics and data science, as well as important application areas such as machine vision, computational biology, speech recognition, text understanding, gaming, and robotics.
Like the format? Trust SlidesLive to capture your next event!
Professional recording and live streaming, delivered globally.
Sharing
Recommended Videos
Presentations on similar topic, category or speaker
Tradeoffs in Data Augmentation: An Empirical Study
Watch later
Learned perceptual representations
Watch later
Optimizing Neural Architectures For Vision
Watch later
Rapid Neural Architecture Search by Learning to Generate Graphs from Datasets
Watch later
Hayeon Lee, …
Heteroskedastic and Imbalanced Deep Learning with Adaptive Regularization
Watch later
Kaidi Cao, …
Learning Safe Multi-agent Control with Decentralized Neural Barrier Certificates
Watch later
Zengyi Qin, …