mlsanityData to 7 Oct 2026

Neural Organ Transplantation (NOT): Checkpoint-Based Modular Adaptation for Transformer Models

arXiv 2601.13580 ↗First posted 20 Jan 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

We introduce Neural Organ Transplantation (NOT), a modular adaptation framework that enables trained transformer layers to function as reusable transferable checkpoints for domain adaptation. Unlike conventional fine-tuning approaches that tightly couple trained parameters to specific model instances and training data, NOT extracts contiguous layer subsets ("donor organs") from pre-trained models, trains them independently on domain-specific data, and saves them as standalone checkpoint files that can be transplanted into compatible recipient models without access to the original training data. Through experiments on three decoder-only transformer architectures spanning 124M to 20B parameters (GPT-2, TinyLlama, and GPT-OSS), we demonstrate that donor transplantation substantially outperforms existing adaptation methods, achieving an order-of-magnitude improvement in perplexity over LoRA while training significantly faster. The method exhibits position dependence, with early insertion positions yielding optimal results. Cross-domain transfer at billion-parameter scale reveals unexpected regularization benefits. These findings demonstrate that transformer middle layers can support efficient modular transfer for decoder-only architectures, enabling privacy-preserving expertise sharing through checkpoint distribution. We note that this approach is currently limited to decoder-only models; preliminary experiments on encoder-based architectures show reduced effectiveness.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 23 of its 33 references that are in the map. First posted on arXiv 20 Jan 2026.

Builds on

23 of 33 references.

References · 23

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  2. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  3. 2015Distilling the Knowledge in a Neural Networkcited by 10,076
  4. 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
  5. 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,878
  6. 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,769
  7. 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,355
  8. 2016Pointer Sentinel Mixture Modelscited by 2,992
  9. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
  10. 2022Editing Models with Task Arithmeticcited by 1,076
  11. 2019BERT Rediscovers the Classical NLP Pipelinecited by 1,046
  12. 2022Few-Shot Parameter-Efficient Fine-Tuning is Better and Cheaper than In-Context Learningcited by 886
11 more
  1. 2020A Primer in BERTology: What we know about how BERT workscited by 807
  2. 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
  3. 2020AdapterFusion: Non-Destructive Task Composition for Transfer Learningcited by 780
  4. 2024DoRA: Weight-Decomposed Low-Rank Adaptationcited by 740
  5. 2023AdaLoRA: Adaptive Budget Allocation for Parameter-Efficient Fine-Tuningcited by 337
  6. 2023MedAlpaca -- An Open-Source Collection of Medical Conversational AI Models and Training Datacited by 282
  7. 2014Understanding image representations by measuring their equivariance and equivalencecited by 240
  8. 2015Convergent Learning: Do different neural networks learn the same representations?cited by 215
  9. 2021Revisiting Model Stitching to Compare Neural Representationscited by 203
  10. 2023Composing Parameter-Efficient Modules with Arithmetic Operationscited by 155
  11. 2021Similarity and Matching of Neural Network Representationscited by 77

Built on by

Not cited yet.