POME: Post Optimization Model Edit via Muon-style Projection
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
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
29 more
- 2017Axiomatic Attribution for Deep Networkscited by 3,384
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
- 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,391
- 2025Understanding R1-Zero-Like Training: A Critical Perspectivecited by 1,345
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
- 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,301
- 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,090
- 2022Mass-Editing Memory in a Transformercited by 1,001
- 2023MetaMath: Bootstrap Your Own Mathematical Questions for Large Language Modelscited by 735
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 660
- 2021Knowledge Neurons in Pretrained Transformerscited by 601
- 2021Fast Model Editing at Scalecited by 498
- 2022Memory-Based Model Editing at Scalecited by 447
- 2025Muon is Scalable for LLM Trainingcited by 428
- 2023Editing Large Language Models: Problems, Methods, and Opportunitiescited by 372
- 2023LLM-Adapters: An Adapter Family for Parameter-Efficient Fine-Tuning of Large Language Modelscited by 324
- 2024OpenCodeInterpreter: Integrating Code Generation with Execution and Refinementcited by 247
- 2023Does Localization Inform Editing? Surprising Differences in Causality-Based Localization vs. Knowledge Editing in Language Modelscited by 243
- 2019Model Fusion via Optimal Transportcited by 219
- 2023Transformer-Patcher: One Mistake worth One Neuroncited by 197
- 2023A Spectral Condition for Feature Learningcited by 109
- 2023NEFTune: Noisy Embeddings Improve Instruction Finetuningcited by 100
- 2025AnyEdit: Edit Any Knowledge Encoded in Language Modelscited by 76
- 2023Massive Editing for Large Language Models via Meta Learningcited by 63
- 2025Reinforced Lifelong Editing for Language Modelscited by 34
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Merging Beyond: Streaming LLM Updates via Activation-Guided Rotationscited by 1
- 2026Subspace Control: Turning Constrained Model Steering into Controllable Spectral Optimizationcited by 1
- 2026Muonp: Muon with Fractional Spectral Powerscited by 1
- 2026Geometric Calibration of Supervised Fine-Tuningcited by 0
- 2026Geometric Calibration of Supervised Fine-Tuning13 Oct 2026
- 2026Muonp: Muon with Fractional Spectral Powers11 Jun 2026
- 2026Subspace Control: Turning Constrained Model Steering into Controllable Spectral Optimization5 Apr 2026
- 2026Merging Beyond: Streaming LLM Updates via Activation-Guided Rotations3 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2025MuLoCo: Muon is a practical inner optimizer for DiLoCo
- 2026Advancing Model Refinement: Muon-Optimized Distillation and Quantization for LLM Deployment
- 2026AF-Muon: An AdamW-Free Muon Optimizer for Tied-Embedding Models
- 2026Muon Meets Mamba: Spectral Optimization for State Space Models
- 2026Monarch-Muon: Structure-Aware Muon for Fast and Memory-Efficient Pretraining
- 2026MONA: Muon Optimizer with Nesterov Acceleration for Scalable Language Model Training
- 2025Muon: Training and Trade-offs with Latent Attention and MoE
- 2025Towards Fast LLM Fine-tuning through Zeroth-Order Optimization with Projected Gradient-Aligned Perturbations