Using semantic entities to improve the distillation of transformers