用可计算的描述长度目标训练Transformer,实现理论最优压缩与泛化。
Bridging Kolmogorov Complexity and Deep Learning: Asymptotically Optimal Description Length Objectives for Transformers
- 基于柯尔莫哥洛夫复杂度提出渐近最优描述长度目标
- 该目标在资源无限时可实现任意数据集的最优压缩(差常数)
- 设计可微分变分目标,实现在算法任务上的强泛化能力
最小描述长度(MDL)原则为机器学习中的奥卡姆剃刀提供了形式化框架。然而,由于缺乏对神经网络(如Transformer)模型复杂性的普适度量,其应用面临挑战。本文引入渐近最优描述长度目标的理论概念,基于柯尔莫哥洛夫复杂度理论。我们证明,在模型资源上限趋于无穷时,此类目标的最小化器可对任意数据集实现最优压缩(仅差一个加性常数)。我们进一步证明,Transformer存在此类目标,并基于其计算通用性的新证明建立基础。同时,我们构造并分析了一种基于自适应高斯混合先验的变分目标,使其具有可计算性和可微性。实验表明,该变分目标能选择低复杂度解,在算法任务上展现强泛化能力;但标准优化器从随机初始化难以找到此类解,凸显关键优化挑战。更广泛而言,本工作为识别具备强渐近保证的描述长度目标提供理论框架,指明了提升神经网络压缩与泛化能力的新路径。
原文摘要 · Abstract (English)
The Minimum Description Length (MDL) principle offers a formal framework for applying Occam's razor in machine learning. However, its application to neural networks such as Transformers is challenging due to the lack of a principled, universal measure for model complexity. This paper introduces the theoretical notion of asymptotically optimal description length objectives, grounded in the theory of Kolmogorov complexity. We establish that a minimizer of such an objective achieves optimal compression, for any dataset, up to an additive constant, in the limit as model resource bounds increase. We prove that asymptotically optimal objectives exist for Transformers, building on a new demonstration of their computational universality. We further show that such objectives can be tractable and differentiable by constructing and analyzing a variational objective based on an adaptive Gaussian mixture prior. Our empirical analysis shows that this variational objective selects for a low-complexity solution with strong generalization on an algorithmic task, but standard optimizers fail to find such solutions from a random initialization, highlighting key optimization challenges. More broadly, by providing a theoretical framework for identifying description length objectives with strong asymptotic guarantees, we outline a potential path towards training neural networks that achieve greater compression and generalization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。