mlsanityData to 14 Oct 2026

POME: Post Optimization Model Edit via Muon-style Projection

arXiv 2510.06627 ↗First posted 8 Oct 2025cs.LGcited by 4

Not on the map. Why?

Abstract (arXiv)

We introduce Post-Optimization Model Edit (POME), a new algorithm that enhances the performance of fine-tuned large language models using only their pretrained and fine-tuned checkpoints, without requiring extra data or further optimization. The core idea is to apply a muon-style projection to Delta W, the difference between the fine-tuned and pretrained weights. This projection uses truncated singular value decomposition (SVD) to equalize the influence of dominant update directions and prune small singular values, which often represent noise. As a simple post-processing step, POME is completely decoupled from the training pipeline. It requires zero modifications and imposes no overhead, making it universally compatible with any optimizer or distributed framework. POME delivers consistent gains, boosting average performance by +2.5% on GSM8K and +1.0% on code generation. Its broad applicability -- from 7B foundation models to 72B RLHF-instructed models -- establishes it as a practical, zero-cost enhancement for any fine-tuning pipeline. Code is available at https://github.com/NUS-HPC-AI-Lab/POME.

Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 41 of its 51 references that are in the map. First posted on arXiv 8 Oct 2025.

Builds on

41 of 51 references.

References · 41

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  2. 2022Training language models to follow instructions with human feedbackcited by 15,971
  3. 2017Proximal Policy Optimization Algorithmscited by 15,215
  4. 2024The Llama 3 Herd of Modelscited by 14,715
  5. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  6. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  7. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  8. 2021Evaluating Large Language Models Trained on Codecited by 8,320
  9. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
  10. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
  11. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  12. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
29 more
  1. 2017Axiomatic Attribution for Deep Networkscited by 3,384
  2. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
  3. 2022Locating and Editing Factual Associations in GPTcited by 2,691
  4. 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
  5. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
  6. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
  7. 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,391
  8. 2025Understanding R1-Zero-Like Training: A Critical Perspectivecited by 1,345
  9. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
  10. 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,301
  11. 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,090
  12. 2022Mass-Editing Memory in a Transformercited by 1,001
  13. 2023MetaMath: Bootstrap Your Own Mathematical Questions for Large Language Modelscited by 735
  14. 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 660
  15. 2021Knowledge Neurons in Pretrained Transformerscited by 601
  16. 2021Fast Model Editing at Scalecited by 498
  17. 2022Memory-Based Model Editing at Scalecited by 447
  18. 2025Muon is Scalable for LLM Trainingcited by 428
  19. 2023Editing Large Language Models: Problems, Methods, and Opportunitiescited by 372
  20. 2023LLM-Adapters: An Adapter Family for Parameter-Efficient Fine-Tuning of Large Language Modelscited by 324
  21. 2024OpenCodeInterpreter: Integrating Code Generation with Execution and Refinementcited by 247
  22. 2023Does Localization Inform Editing? Surprising Differences in Causality-Based Localization vs. Knowledge Editing in Language Modelscited by 243
  23. 2019Model Fusion via Optimal Transportcited by 219
  24. 2023Transformer-Patcher: One Mistake worth One Neuroncited by 197
  25. 2023A Spectral Condition for Feature Learningcited by 109
  26. 2023NEFTune: Noisy Embeddings Improve Instruction Finetuningcited by 100
  27. 2025AnyEdit: Edit Any Knowledge Encoded in Language Modelscited by 76
  28. 2023Massive Editing for Large Language Models via Meta Learningcited by 63
  29. 2025Reinforced Lifelong Editing for Language Modelscited by 34

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025MuLoCo: Muon is a practical inner optimizer for DiLoCo
  2. 2026Advancing Model Refinement: Muon-Optimized Distillation and Quantization for LLM Deployment
  3. 2026AF-Muon: An AdamW-Free Muon Optimizer for Tied-Embedding Models
  4. 2026Muon Meets Mamba: Spectral Optimization for State Space Models
  5. 2026Monarch-Muon: Structure-Aware Muon for Fast and Memory-Efficient Pretraining
  6. 2026MONA: Muon Optimizer with Nesterov Acceleration for Scalable Language Model Training
  7. 2025Muon: Training and Trade-offs with Latent Attention and MoE
  8. 2025Towards Fast LLM Fine-tuning through Zeroth-Order Optimization with Projected Gradient-Aligned Perturbations