Skip to content
AI News HubLIVE
Original source2 min read

Sparse Memory Finetuning as a Low-Forgetting Alternative to LoRA and Full Finetuning

Summary

Sparse Memory Finetuning (SMF) addresses catastrophic forgetting by adding key-value memory layers and updating only the most heavily read memory rows per batch. Experiments on Qwen-2.5-0.5B-Instruct using MedMCQA show SMF improves performance by 2.5 percentage points while keeping forgetting probes within 1 point of the base model, whereas LoRA and full finetuning achieve larger gains but with clear drift. Two row-selection rules (KL-divergence and TF-IDF) are also compared.

SourcearXiv Computational LinguisticsAuthor: Prakhar Gupta, Garv Shah, Satyam Goyal, Anirudh Kanchi
Sparse Memory Finetuning as a Low-Forgetting Alternative to LoRA and Full Finetuning
Report an error

The correction channel is not available yet. You can copy the article reference below for later.

Correction instructions
Read article

[2605.03229] Sparse Memory Finetuning as a Low-Forgetting Alternative to LoRA and Full Finetuning

Skip to main content

Learn about arXiv becoming an independent nonprofit.

We gratefully acknowledge support from the Simons Foundation, member institutions, and all contributors. Donate

> cs > arXiv:2605.03229

Help | Advanced Search

All fields Title Author Abstract Comments Journal reference ACM classification MSC classification Report number arXiv identifier DOI ORCID arXiv author ID Help pages Full text

Search

GO

quick links

Login

Help Pages

About

-->

Computer Science > Computation and Language

arXiv:2605.03229 (cs)

[Submitted on 4 May 2026]

Title:Sparse Memory Finetuning as a Low-Forgetting Alternative to LoRA and Full Finetuning

Authors:Prakhar Gupta, Garv Shah, Satyam Goyal, Anirudh Kanchi

View a PDF of the paper titled Sparse Memory Finetuning as a Low-Forgetting Alternative to LoRA and Full Finetuning, by Prakhar Gupta and 3 other authors

View PDF HTML (experimental)

Abstract:Adapting a pretrained language model to a new task often hurts the general capabilities it already had, a problem known as catastrophic forgetting. Sparse Memory Finetuning (SMF) tries to avoid this by adding key-value memory layers to the model and, on each training step, updating only the small set of memory rows that the current batch reads most heavily. We re-implement SMF on Qwen-2.5-0.5B-Instruct and compare it with LoRA and full finetuning on MedMCQA, a 4-choice medical exam task, using WikiText perplexity and TriviaQA accuracy as forgetting probes. SMF improves MedMCQA by 2.5 percentage points while keeping both forgetting probes within roughly 1 point of the base model, whereas LoRA and full finetuning achieve larger gains but with clear drift on both. We also compare two row-selection rules (KL-divergence and TF-IDF), which balance the two forgetting metrics differently.

Subjects:

Computation and Language (cs.CL); Machine Learning (cs.LG)

Cite as: arXiv:2605.03229 [cs.CL]

(or arXiv:2605.03229v1 [cs.CL] for this version)

https://doi.org/10.48550/arXiv.2605.03229

Focus to learn more

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Prakhar Gupta [view email] [v1] Mon, 4 May 2026 23:46:40 UTC (1,325 KB)

Full-text links:

Access Paper:

View a PDF of the paper titled Sparse Memory Finetuning as a Low-Forgetting Alternative to LoRA and Full Finetuning, by Prakhar Gupta and 3 other authors

View PDF

HTML (experimental)

TeX Source

view license

Current browse context:

cs.CL

new | recent | 2026-05

Change to browse by:

cs cs.LG

References & Citations

NASA ADS

Google Scholar

Semantic Scholar

export BibTeX citation Loading...

BibTeX formatted citation

×

loading...

Data provided by:

Bookmark

Bibliographic Tools

Bibliographic and Citation Tools

Bibliographic Explorer Toggle

Bibliographic Explorer (What is the Explorer?)

Connected Papers Toggle

Connected Papers (What is Connected Papers?)

Litmaps Toggle

Litmaps (What is Litmaps?)

scite.ai Toggle

scite Smart Citations (What are Smart Citations?)

Code, Data, Media

Code, Data and Media Associated with this Article

alphaXiv Toggle

alphaXiv (What is alphaXiv?)

Links to Code Toggle

CatalyzeX Code Finder for Papers (What is CatalyzeX?)

DagsHub Toggle

DagsHub (What is DagsHub?)

GotitPub Toggle

Gotit.pub (What is GotitPub?)

Huggingface Toggle

Hugging Face (What is Huggingface?)

ScienceCast Toggle

ScienceCast (What is ScienceCast?)

Demos

Demos

Replicate Toggle

Replicate (What is Replicate?)

Spaces Toggle

Hugging Face Spaces (What is Spaces?)

Spaces Toggle

TXYZ.AI (What is TXYZ.AI?)

Related Papers

Recommenders and Search Tools

Link to Influence Flower

Influence Flower (What are Influence Flowers?)

Core recommender toggle

CORE Recommender (What is CORE?)

Author

Venue

Institution

Topic

About arXivLabs

arXivLabs: experimental projects with community collaborators

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.

Which authors of this paper are endorsers? | Disable MathJax (What is MathJax?)

About

Help

contact arXivClick here to contact arXiv Contact

subscribe to arXiv mailingsClick here to subscribe Subscribe

Copyright

Privacy Policy

Web Accessibility Assistance

arXiv Operational Status

Key points and analysis

Article intelligence

EngineersAdvanced

Key points

  • SMF adds key-value memory layers to avoid catastrophic forgetting
  • Only the most heavily read memory rows are updated each step
  • SMF yields 2.5% improvement on MedMCQA with minimal forgetting
  • KL-divergence and TF-IDF row selection balance forgetting differently

Highlights and analysis are generated automatically and may contain errors. Check the original source.

Sparse Memory Finetuning as a Low-Forgetting Alternative to LoRA and Full Finetuning | AI News Hub