arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2607.02460cs.LGcs.AI

面向无标注LLM自蒸馏的神经元感知数据选择

Neuron-Aware Data Selection for Annotation-Free LLM Self-Distillation

University of Pittsburgh(匹兹堡大学)

Zhuowei Chen, Xiang Lorraine Li

AI总结:

提出Neuron-OPSD框架,利用神经元激活指导训练数据选择和教师上下文构建,通过在线策略蒸馏实现无标注自蒸馏,提升领域内性能并保持跨领域泛化。

摘要:

在没有真实世界交互反馈或人工标注监督的情况下对大型语言模型(LLM)进行后训练仍然具有挑战性,特别是在专家标注成本高昂的专业领域。最近的无标注自进化方法通过使用模型自身的输出作为监督信号来解决这个问题,通过额外上下文构建教师模型,并通过多数投票聚合多个rollout的预测来生成伪标签。然而,这些方法并非没有缺点:基于SFT和GRPO的变体会导致域外性能下降,而基于奖励的在线策略RL会膨胀校准误差。在本文中,我们提出神经元在线策略自蒸馏(Neuron-OPSD),一个用于无标注自蒸馏的数据中心框架,利用内部神经元激活来指导训练数据选择和教师上下文构建。然后,模型通过从教师分布进行在线策略蒸馏来训练,在任何阶段都不需要真实标签。在专业领域基准测试中,Neuron-OPSD提高了领域内任务性能,同时保持了跨领域泛化,并减轻了先前无标注基线的校准崩溃。该框架特别适用于在线交互或外部监督成本高昂或不可行的场景,并且在概念上不同于依赖已记录奖励标签轨迹的离线RL方法。

英文摘要:

Post-training large language models (LLMs) without real-world interaction feedback or human-labeled supervision remains challenging, particularly in specialized domains where expert annotations are costly to obtain. Recent annotation-free self-evolution methods address this by using the model's own outputs as supervision signals, constructing a teacher via additional context and aggregating predictions across multiple rollouts through majority voting to produce pseudo-labels. However, these approaches are not without drawbacks: SFT- and GRPO-based variants suffer out-of-domain performance degradation, while reward-based on-policy RL inflates calibration error. In this paper, we propose Neuron On-Policy Self-Distillation (Neuron-OPSD), a data-centric framework for annotation-free self-distillation that leverages internal neuron activations to guide both training-data selection and teacher context construction. The model is then trained via on-policy distillation from the teacher distribution, requiring no ground-truth labels at any stage. Across specialized-domain benchmarks, Neuron-OPSD improves in-domain task performance while preserving cross-domain generalization and mitigating calibration collapse over prior annotation-free baselines. This framework is particularly relevant to settings where online interaction or external supervision is costly or infeasible, and is conceptually distinct from offline RL approaches that rely on logged, reward-labeled trajectories.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑