arXiv · 2604.06732
Extraction of linearized models from pre-trained networks via knowledge distillation
Abstract
Recent developments in hardware, such as photonic integrated circuits and optical devices, are driving demand for research on constructing machine learning architectures tailored for linear operations. Hence, it is valuable to explore methods for constructing learning machines with only linear operations after simple nonlinear preprocessing. In this study, we propose a framework to extract a linearized model from a pre-trained neural network for classification tasks by integrating Koopman operator theory with knowledge distillation. Numerical demonstrations on the MNIST and the Fashion-MNIST datasets reveal that the proposed model consistently outperforms the conventional least-squares-based Koopman approximation in both classification accuracy and numerical stability.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Fumito Kimura, Jun Ohkubo. 2026-04-08. Extraction of linearized models from pre-trained networks via knowledge distillation. https://arxiv.org/abs/2604.06732
Cite the original work for its findings. Save a collection to share your selection of sources.