Multiple Teacher Distillation for Robust and Greener Models

Artur Ilichev, Nikita Sorokin, Huawei Noah’s Ark lab, Moscow, Russia, Valentin Andreevich Malykh, Huawei Noah’s Ark lab, Moscow, Russia, Irina Piontkovskaya, Huawei Noah’s Ark lab, Moscow, Russia · 2021

The language models nowadays are in the center of natural language processing progress.These models are mostly of significant size.There are successful attempts to reduce them, but at least some of these attempts rely on randomness.We propose a novel distillation procedure leveraging on multiple teachers usage which alleviates random seed dependency and makes the models more robust.We show that this procedure applied to TinyBERT and Dis-tilBERT models improves their worst case results up to 2% while keeping almost the same best-case ones.The latter fact keeps true with a constraint on computational time, which is important to lessen the carbon footprint.In addition, we present the results of an application of the proposed procedure to a computer vision model ResNet, which shows that the statement keeps true in this totally different domain.

Read the paper · More papers on PaperTik