用神经元激活图筛选数据,让模型更快变专家
Target-Oriented Pretraining Data Selection via Neuron-Activated Graph

- 通过高影响力神经元构建稀疏图,无须训练即可定位关键数据
- 比随机选数据提升4.9%,在HellaSwag上领先基线5.3%
- 仅激活0.12%神经元就导致性能下降23.5%,证明其功能核心性
日常任务都有目标,围绕目标预训练能让模型成为专家。本文提出无需训练且可解释的神经元激活图排序(NAG-based Ranking)框架,用于目标导向的语言模型预训练数据选择。不依赖黑箱表示,而是直接通过任意现成大模型中的高影响神经元来刻画目标输入。具体地,量化神经元影响并选取跨层最具影响力的神经元构成紧凑的神经元激活图(NAG),再根据候选数据与目标样本的NAG相似度进行排序。在六个基准测试中,该方法平均比随机采样提升4.9%,在HellaSwag上优于最先进基线5.3%。在更实用的多目标场景下,最佳设置分别超越两个基线1.1%和4.1%。进一步分析显示,关闭被NAG选中的神经元(仅占全部的0.12%)会导致性能下降23.5%,仅保留最后一层NAG则平均下降4.1%,表明NAG捕捉到了学习目标特征的稀疏“功能主干”。代码已开源:https://github.com/asillycat/NAG。
原文摘要 · Abstract (English)
Everyday tasks come with a target, and pretraining models around this target is what turns them into experts. In this paper, we study target-oriented language model (LM) pretraining by introducing Neuron-Activated Graph Ranking (NAG-based Ranking), a training-free and interpretable framework for target pretraining data selection. Rather than using black-box representations, our approach directly characterizes each target input by a sparse set of high-impact neurons in any off-the-shelf LLMs. Concretely, we quantify neuron impact and select the most influential neurons across layers into a compact Neuron-Activated Graph (NAG), and rank candidate data by NAG similarity to target examples. We conduct experiments across six benchmarks, where our NAG-based Ranking improves target-oriented pretraining by 4.9% on average over random sampling, and also outperforms state-of-the-art baselines by 5.3% accuracy on HellaSwag. It also remains effective under a more applicable multi-target setting, where our best setup surpasses two baselines by 1.1% and 4.1%, respectively. Furthermore, we provide a comprehensive analysis on why and how our NAG works, e.g., deactivating NAG-selected neurons (only 0.12% of all) causes a 23.5% performance collapse, and restricting NAG to the final layer incurs a 4.1% average drop, indicating that NAG captures a sparse "functional backbone" for learning target features. We release the code at https://github.com/asillycat/NAG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。