AI & ML
MUON-RT Reusing the Teacher's Trajectory for Momentum-Free Muon for LLM Self-Distillation
arXiv stat.ML0 views
arXiv:2610.41104v1 Announce Type: cross Abstract: 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.
Source:https://arxiv.org/abs/2610.41104PreviousWhere, When, and How Many Bits? Scaling Laws for SGD with Stochastic RoundingNextA Unified Framework for Probability Estimation of Data-Dependent Events
Related posts
Merge builds product and business tools with OpenAI technology
0OpenAI Blog
How to Survive (Not Thrive) as a Salaryman
0HackerNoon
Debugging macOS LaunchAgent Failures With launchctl and Exit Code 78
0HackerNoon

Building Data Observability for Banking Pipelines
0HackerNoon
OpenMatter Network Positions Decentralized Computing as an Alternative To Data Centers for AI
0HackerNoon
BMW bumps battery for the better: The 2027 i7 60 xDrive, driven
0Ars Technica