提出InherNet,让轻量网络更高效继承教师模型知识
Beyond Student: An Asymmetric Network for Neural Network Inheritance
- 用SVD对教师权重做非对称低秩分解,重建轻量表达网络
- 在同参数量下,性能超越传统学生网络,跨模态任务均有效
- 适合追求高效压缩与高精度的模型部署场景
知识蒸馏(KD)是模型压缩的重要技术,使轻量学生网络能受益于冗余教师网络。然而,固有的容量差距常限制学生网络性能。受预训练教师网络表现力启发,我们提出新问题:是否存在一种网络,不仅能继承教师结构,还能最大化其知识?为此,我们提出InherNet,通过在教师权重上进行非对称低秩分解,重建轻量但富有表现力的网络,且不显著改变架构。利用奇异值分解(SVD)初始化以保证主知识继承,InherNet在深度、宽度与压缩效率间取得良好平衡。在单模态与多模态任务上的实验表明,与同参数量学生网络相比,InherNet性能更高。研究揭示了超越传统蒸馏的高效模型压缩新方向。
原文摘要 · Abstract (English)
Knowledge Distillation (KD) has emerged as a powerful technique for model compression, enabling lightweight student networks to benefit from the performance of redundant teacher networks. However, the inherent capacity gap often limits the performance of student networks. Inspired by the expressiveness of pretrained teacher networks, a compelling research question arises: is there a type of network that can not only inherit the teacher's structure but also maximize the inheritance of its knowledge? Furthermore, how does the performance of such an inheriting network compare to that of student networks, all benefiting from the same teacher network? To further explore this question, we propose InherNet, a neural network inheritance method that performs asymmetric low-rank decomposition on the teacher's weights and reconstructs a lightweight yet expressive network without significant architectural disruption. By leveraging Singular Value Decomposition (SVD) for initialization to ensure the inheritance of principal knowledge, InherNet effectively balances depth, width, and compression efficiency. Experimental results across unimodal and multimodal tasks demonstrate that InherNet achieves higher performance compared to student networks of similar parameter sizes. Our findings reveal a promising direction for future research in efficient model compression beyond traditional distillation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。