通过精巧架构设计,让模型更小更快更省资源。
Smaller, Faster, Cheaper: Architectural Designs for Efficient Machine Learning
- 优化数据输入输出机制,提升小模型性能
- 改进视觉变换器注意力结构,增强表达能力
- 利用归一化流特性实现高效知识蒸馏
计算机视觉模型能力的显著提升主要依赖于数据集、模型参数和计算预算的快速扩张,导致对计算基础设施的需求持续增加。然而,随着模型在更多资源受限环境中的部署,迫切需要在保持高性能的同时降低计算开销的架构设计。本论文聚焦于通过架构原则实现性能提升与计算需求减少的平衡。首先,研究数据输入与输出机制,确保核心神经处理单元最大限度利用可用信息,使小型模型具备更强性能。其次,针对视觉变换器中的受限注意力机制,探索去除统一上下文窗口的设计,以提升模型表达能力。第三,分析归一化流(Normalizing Flows)的自然结构,并利用其特性实现更高效的模型知识蒸馏。这些贡献表明,精心设计的神经架构能显著提升机器学习算法的效率,推动模型向更小、更快、更廉价的方向发展。
原文摘要 · Abstract (English)
Major advancements in the capabilities of computer vision models have been primarily fueled by rapid expansion of datasets, model parameters, and computational budgets, leading to ever-increasing demands on computational infrastructure. However, as these models are deployed in increasingly diverse and resource-constrained environments, there is a pressing need for architectures that can deliver high performance while requiring fewer computational resources. This dissertation focuses on architectural principles through which models can achieve increased performance while reducing their computational demands. We discuss strides towards this goal through three directions. First, we focus on data ingress and egress, investigating how information may be passed into and retrieved from our core neural processing units. This ensures that our models make the most of available data, allowing smaller architectures to become more performant. Second, we investigate modifications to the core neural architecture, applied to restricted attention in vision transformers. This section explores how removing uniform context windows in restricted attention increases the expressivity of the underlying neural architecture. Third, we explore the natural structures of Normalizing Flows and how we can leverage these properties to better distill model knowledge. These contributions demonstrate that careful design of neural architectures can increase the efficiency of machine learning algorithms, allowing them to become smaller, faster, and cheaper.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。