← 返回时间线

paper

MUON-RT Reusing the Teacher's Trajectory for Momentum-Free Muon for LLM Self-Distillation

arXiv ↗
ID
2610.41104
分类
—
首次捕获
2026-11-05
状态
unread
作者
Aryan Das, Vinay Kumar Verma, Aakansha Mishra
信号
🔥 20

信号历史

  • 2026-11-05HF Daily Papers · 🔥9
  • 2026-11-06HF Daily Papers · 🔥20
暂无信号数据

摘要

Self-distillation methods for large language models with an exponential moving average (EMA) teacher already retain a history of student updates. Muon nevertheless stores a separate momentum buffer, increasing persistent training memory. We propose Muon with Reused Teacher (Muon-RT), which reuses the teacher-student displacement to eliminate momentum buffers for hidden weight matrices. We show that this displacement encodes an exponentially weighted history of applied updates and smooths temporally uncorrelated noise under a local model. Muon-RT removes its radial component, normalizes it spectrally, and mixes it with the current gradient after Newton-Schulz (NS) orthogonalization. Under explicit regularity and alignment assumptions, we establish linear contraction of an expected objective-gap bound toward a residual error level. With Self-Distillation Fine-Tuning (SDFT) on Qwen2.5-7B-Instruct, Muon-RT improves average accuracy over AdamW by 2.6 points with 42% less persistent memory, and over momentum Muon by 5.4 points with 27% less persistent memory. With Self-Distillation Policy Optimization (SDPO) on Olmo-3-7B-Instruct, it improves average accuracy over AdamW by 2.0 points.

我的笔记

还没有笔记。

在 GitHub 上写笔记 ↗(新建 content/notes/2610.41104.md,PR 合并后本页自动更新)