Reversing Large Language Models for Efficient Training and Fine-Tuning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Gal, Eshed, Eliasof, Moshe, Turek, Javier, Ascher, Uri, Treister, Eran, Haber, Eldad
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908693225799680
author Gal, Eshed
Eliasof, Moshe
Turek, Javier
Ascher, Uri
Treister, Eran
Haber, Eldad
author_facet Gal, Eshed
Eliasof, Moshe
Turek, Javier
Ascher, Uri
Treister, Eran
Haber, Eldad
contents Large Language Models (LLMs) are known for their expensive and time-consuming training. Thus, oftentimes, LLMs are fine-tuned to address a specific task, given the pretrained weights of a pre-trained LLM considered a foundation model. In this work, we introduce memory-efficient, reversible architectures for LLMs, inspired by symmetric and symplectic differential equations, and investigate their theoretical properties. Different from standard, baseline architectures that store all intermediate activations, the proposed models use time-reversible dynamics to retrieve hidden states during backpropagation, relieving the need to store activations. This property allows for a drastic reduction in memory consumption, allowing for the processing of larger batch sizes for the same available memory, thereby offering improved throughput. In addition, we propose an efficient method for converting existing, non-reversible LLMs into reversible architectures through fine-tuning, rendering our approach practical for exploiting existing pre-trained models. Our results show comparable or improved performance on several datasets and benchmarks, on several LLMs, building a scalable and efficient path towards reducing the memory and computational costs associated with both training from scratch and fine-tuning of LLMs.
format Preprint
id arxiv_https___arxiv_org_abs_2512_02056
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Reversing Large Language Models for Efficient Training and Fine-Tuning
Gal, Eshed
Eliasof, Moshe
Turek, Javier
Ascher, Uri
Treister, Eran
Haber, Eldad
Computation and Language
Artificial Intelligence
Large Language Models (LLMs) are known for their expensive and time-consuming training. Thus, oftentimes, LLMs are fine-tuned to address a specific task, given the pretrained weights of a pre-trained LLM considered a foundation model. In this work, we introduce memory-efficient, reversible architectures for LLMs, inspired by symmetric and symplectic differential equations, and investigate their theoretical properties. Different from standard, baseline architectures that store all intermediate activations, the proposed models use time-reversible dynamics to retrieve hidden states during backpropagation, relieving the need to store activations. This property allows for a drastic reduction in memory consumption, allowing for the processing of larger batch sizes for the same available memory, thereby offering improved throughput. In addition, we propose an efficient method for converting existing, non-reversible LLMs into reversible architectures through fine-tuning, rendering our approach practical for exploiting existing pre-trained models. Our results show comparable or improved performance on several datasets and benchmarks, on several LLMs, building a scalable and efficient path towards reducing the memory and computational costs associated with both training from scratch and fine-tuning of LLMs.
title Reversing Large Language Models for Efficient Training and Fine-Tuning
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2512.02056