arXiv:2605.01205cs.CL2026-05ACL被引 2

用物理模型对齐语义段,提升跨分词器大模型蒸馏效果

SRA: Span Representation Alignment for Large Language Model Distillation

论文配图:SRA: Span Representation Alignment for Large Language Model Distillation
图 1 · 摘自论文原文
  • 将分词粒度提升为语义段,通过质心聚合增强鲁棒性
  • 在跨架构蒸馏中显著优于现有方法,准确率提升超过5% (在GLUE上)
  • 适合需要跨模型迁移的场景,尤其适用于分词不一致的情况

跨分词器知识蒸馏(CTKD)可在不同分词器的大模型与小模型间实现知识迁移。现有方法多聚焦于词元级对齐,但对分词差异敏感且易失效。本文提出SRA(Span Representation Alignment),从多粒子动力系统角度重构CTKD,将对齐单元由词元升级为鲁棒的、分词无关的语义段。每个语义段被建模为粒子簇,其状态由注意力加权的质心(CoM)表示,捕捉丰富语义信息。通过质心优先策略和几何正则化保持表征空间结构,引入对齐语义段logits蒸馏以增强知识传递。在多个跨架构蒸馏任务中,SRA持续显著优于当前最优基线,验证了其物理启发式方法的有效性。

原文摘要 · Abstract (English)

Cross-Tokenizer Knowledge Distillation (CTKD) enables knowledge transfer between a large language model and a smaller student, even when they employ different tokenizers. While existing approaches mainly focus on token-level alignment strategies, which are often brittle and sensitive to discrepancies between tokenizers, we argue that the method of aggregating tokens into more robust representations before distillation is of equal importance. In this paper, we introduce \textbf{SRA} (\textbf{S}pan \textbf{R}epresentation \textbf{A}lignment for Large Language Model Distillation), a novel framework that reframes CTKD through the physical lens of Multi-Particle Dynamical Systems. SRA shifts the fundamental unit of alignment from tokens to robust, tokenizer-agnostic spans. We model each span as a cluster of particles and represent its state by its Center of Mass (CoM) - an attention-weighted average that captures rich semantic information. We leverage the concept of span centers of mass with attention-derived weighting to prioritize the most salient spans. In addition, we employ a geometric regularizer to preserve the structural integrity of the representation space and introduce aligned span logit distillation to enhance knowledge transfer across models. In challenging cross-architecture distillation experiments, SRA consistently and significantly outperforms state-of-the-art CTKD baselines, validating our physically-grounded approach.

知识蒸馏大模型语义对齐跨分词器

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