Distilling Lightweight Domain Experts from Large ML Models by Identifying Relevant Subspaces
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2026
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866908755515408384 |
|---|---|
| author | Chormai, Pattarawat Hashemi, Ali Müller, Klaus-Robert Montavon, Grégoire |
| author_facet | Chormai, Pattarawat Hashemi, Ali Müller, Klaus-Robert Montavon, Grégoire |
| contents | Knowledge distillation involves transferring the predictive capabilities of large, high-performing AI models (teachers) to smaller models (students) that can operate in environments with limited computing power. In this paper, we address the scenario in which only a few classes and their associated intermediate concepts are relevant to distill. This scenario is common in practice, yet few existing distillation methods explicitly focus on the relevant subtask. To address this gap, we introduce 'SubDistill', a new distillation algorithm with improved numerical properties that only distills the relevant components of the teacher model at each layer. Experiments on CIFAR-100 and ImageNet with Convolutional and Transformer models demonstrate that SubDistill outperforms existing layer-wise distillation techniques on a representative set of subtasks. Our benchmark evaluations are complemented by Explainable AI analyses showing that our distilled student models more closely match the decision structure of the original teacher model. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2601_05913 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | Distilling Lightweight Domain Experts from Large ML Models by Identifying Relevant Subspaces Chormai, Pattarawat Hashemi, Ali Müller, Klaus-Robert Montavon, Grégoire Machine Learning Knowledge distillation involves transferring the predictive capabilities of large, high-performing AI models (teachers) to smaller models (students) that can operate in environments with limited computing power. In this paper, we address the scenario in which only a few classes and their associated intermediate concepts are relevant to distill. This scenario is common in practice, yet few existing distillation methods explicitly focus on the relevant subtask. To address this gap, we introduce 'SubDistill', a new distillation algorithm with improved numerical properties that only distills the relevant components of the teacher model at each layer. Experiments on CIFAR-100 and ImageNet with Convolutional and Transformer models demonstrate that SubDistill outperforms existing layer-wise distillation techniques on a representative set of subtasks. Our benchmark evaluations are complemented by Explainable AI analyses showing that our distilled student models more closely match the decision structure of the original teacher model. |
| title | Distilling Lightweight Domain Experts from Large ML Models by Identifying Relevant Subspaces |
| topic | Machine Learning |
| url | https://arxiv.org/abs/2601.05913 |