arXiv · 2605.20074
Towards Distillation Guarantees under Algorithmic Alignment for Combinatorial Optimization
Abstract
Distillation transfers knowledge from a large model trained on broad data to a smaller, more efficient model suitable for deployment. In structured prediction settings, prior knowledge about the task can guide the choice of a target architecture that is algorithmically aligned with the underlying problem. Building on recent learning-theoretic analyses of decision-tree (DT) distillation (Boix-Adsera, 2024), we study when distillation succeeds for combinatorial optimization tasks. We focus on the case where the target model is a graph neural network whose architecture is aligned with a dynamic programming (DP) algorithm for the task. Assuming that the source model is sufficiently rich, formalized through the linear representation hypothesis (LRH) (Elhage et al., 2022; Park et al., 2024), we show that the distillation problem can be solved efficiently in the complexity parameters of the DP transition function, represented as a DT. Our results provide a rigorous sufficient condition for successful distillation in the flavour of algorithmic alignment.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Thien Le, Melanie Weber. 2026-05-19. Towards Distillation Guarantees under Algorithmic Alignment for Combinatorial Optimization. https://arxiv.org/abs/2605.20074
Cite the original work for its findings. Save a collection to share your selection of sources.