arXiv:2504.15983cs.CLcs.AI2025-04ICLR被引 1

提出无需训练的W-PCA方法,快速筛选轻量级语言模型。

W-PCA Based Gradient-Free Proxy for Efficient Search of Lightweight Language Models

  • 用参数量和主成分贡献率设计零梯度评估代理
  • 在GLUE/SQuAD上训练时间更短,测试性能更高
  • 适合高效搜索轻量模型,尤其擅长避免梯度计算

高效自然语言处理系统的需求推动了轻量级语言模型的发展。以往工作主要依赖人工设计或基于训练的神经架构搜索(NAS)。近期零样本NAS方法可在无需训练的情况下评估模型,但现有方法常存在评估指标偏差和计算效率低的问题。本文提出针对轻量级语言模型的权重加权主成分分析(W-PCA)方法,利用两个评估代理:参数量及前馈层中累积贡献超过η的主成分数量。通过消除梯度计算,显著降低评估耗时,提升设计效率。我们在GLUE和SQuAD数据集上进行对比分析,结果表明该方法相比单次训练型NAS大幅减少训练时间,并在测试阶段超越此前最优训练型方法。此外,在从FlexiBERT搜索空间采样的数据集上进行排名评估,其相关性优于其他需梯度计算的零样本NAS方法,且求解时间更短。

原文摘要 · Abstract (English)

The demand for efficient natural language processing (NLP) systems has led to the development of lightweight language models. Previous work in this area has primarily focused on manual design or training-based neural architecture search (NAS) methods. Recently, zero-shot NAS methods have been proposed for evaluating language models without the need for training. However, prevailing approaches to zero-shot NAS often face challenges such as biased evaluation metrics and computational inefficiencies. In this paper, we introduce weight-weighted PCA (W-PCA), a novel zero-shot NAS method specifically tailored for lightweight language models. Our approach utilizes two evaluation proxies: the parameter count and the number of principal components with cumulative contribution exceeding $η$ in the feed-forward neural (FFN) layer. Additionally, by eliminating the need for gradient computations, we optimize the evaluation time, thus enhancing the efficiency of designing and evaluating lightweight language models. We conduct a comparative analysis on the GLUE and SQuAD datasets to evaluate our approach. The results demonstrate that our method significantly reduces training time compared to one-shot NAS methods and achieves higher scores in the testing phase compared to previous state-of-the-art training-based methods. Furthermore, we perform ranking evaluations on a dataset sampled from the FlexiBERT search space. Our approach exhibits superior ranking correlation and further reduces solving time compared to other zero-shot NAS methods that require gradient computation.

轻量模型零样本搜索架构优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。