Algorithmic Language Models with Neurally Compiled Libraries

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Saldyt, Lucas, Kambhampati, Subbarao
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915302185369600
author Saldyt, Lucas
Kambhampati, Subbarao
author_facet Saldyt, Lucas
Kambhampati, Subbarao
contents Important tasks such as reasoning and planning are fundamentally algorithmic, meaning that solving them robustly requires acquiring true reasoning or planning algorithms, rather than shortcuts. Large Language Models lack true algorithmic ability primarily because of the limitations of neural network optimization algorithms, their optimization data and optimization objective, but also due to architectural inexpressivity. To solve this, our paper proposes augmenting LLMs with a library of fundamental operations and sophisticated differentiable programs, so that common algorithms do not need to be learned from scratch. We add memory, registers, basic operations, and adaptive recurrence to a transformer architecture built on LLaMA3. Then, we define a method for directly compiling algorithms into a differentiable starting library, which is used natively and propagates gradients for optimization. In this preliminary study, we explore the feasability of augmenting LLaMA3 with a differentiable computer, for instance by fine-tuning small transformers on simple algorithmic tasks with variable computational depth.
format Preprint
id arxiv_https___arxiv_org_abs_2407_04899
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Algorithmic Language Models with Neurally Compiled Libraries
Saldyt, Lucas
Kambhampati, Subbarao
Artificial Intelligence
Computation and Language
Programming Languages
Important tasks such as reasoning and planning are fundamentally algorithmic, meaning that solving them robustly requires acquiring true reasoning or planning algorithms, rather than shortcuts. Large Language Models lack true algorithmic ability primarily because of the limitations of neural network optimization algorithms, their optimization data and optimization objective, but also due to architectural inexpressivity. To solve this, our paper proposes augmenting LLMs with a library of fundamental operations and sophisticated differentiable programs, so that common algorithms do not need to be learned from scratch. We add memory, registers, basic operations, and adaptive recurrence to a transformer architecture built on LLaMA3. Then, we define a method for directly compiling algorithms into a differentiable starting library, which is used natively and propagates gradients for optimization. In this preliminary study, we explore the feasability of augmenting LLaMA3 with a differentiable computer, for instance by fine-tuning small transformers on simple algorithmic tasks with variable computational depth.
title Algorithmic Language Models with Neurally Compiled Libraries
topic Artificial Intelligence
Computation and Language
Programming Languages
url https://arxiv.org/abs/2407.04899