LinearARD: Linear-Memory Attention Distillation for RoPE Restoration
线性记忆注意力蒸馏:用于RoPE修复的线性内存注意力蒸馏
机构 * Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所) ; Southwest University(西南大学) ; Dalian University of Technology(大连理工大学) ; University of Science and Technology Beijing(北京科技大学) ; University College London(伦敦大学学院)
专题命中 长上下文与记忆 :large language model(abstract);language model(abstract);分类 cs.CL、cs.AI
AI总结 本文提出LinearARD,通过冻结的原生RoPE教师模型,利用注意力结构一致性恢复RoPE缩放的学生模型,以线性内存核克服二次内存瓶颈,提升长上下文性能。