N-Gram Induction Heads for In-Context RL: Improving Stability and Reducing Data Needs

Fuente: arXiv

Guardado en:

Detalles Bibliográficos
Autores principales:	Zisman, Ilya, Nikulin, Alexander, Sinii, Viacheslav, Tarasov, Denis, Lyubaykin, Nikita, Polubarov, Andrei, Kiselev, Igor, Kurenkov, Vladislav
Formato:	Preprint
Publicado:	2024
Materias:	Machine Learning
Acceso en línea:	Acceder al recurso
Etiquetas:	Agregar Etiqueta Sin Etiquetas, Sea el primero en etiquetar este registro!

Ejemplares similares

Latent Action Learning Requires Supervision in the Presence of Distractors
por: Nikulin, Alexander, et al.
Publicado: (2025)

Yes, Q-learning Helps Offline In-Context RL
por: Tarasov, Denis, et al.
Publicado: (2025)

Vintix: Action Model via In-Context Reinforcement Learning
por: Polubarov, Andrey, et al.
Publicado: (2025)

Object-Centric Latent Action Learning
por: Klepach, Albina, et al.
Publicado: (2025)

Vision-Language Models Unlock Task-Centric Latent Actions
por: Nikulin, Alexander, et al.
Publicado: (2026)

NinA: Normalizing Flows in Action. Training VLA Models with Normalizing Flows
por: Tarasov, Denis, et al.
Publicado: (2025)

In-Context Reinforcement Learning for Variable Action Spaces
por: Sinii, Viacheslav, et al.
Publicado: (2023)

Emergence of In-Context Reinforcement Learning from Noise Distillation
por: Zisman, Ilya, et al.
Publicado: (2023)

XLand-MiniGrid: Scalable Meta-Reinforcement Learning Environments in JAX
por: Nikulin, Alexander, et al.
Publicado: (2023)

XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning
por: Nikulin, Alexander, et al.
Publicado: (2024)

Vintix II: Decision Pre-Trained Transformer is a Scalable In-Context Reinforcement Learner
por: Polubarov, Andrei, et al.
Publicado: (2026)

Zero-Shot Adaptation of Behavioral Foundation Models to Unseen Dynamics
por: Bobrin, Maksim, et al.
Publicado: (2025)

cadrille: Multi-modal CAD Reconstruction with Reinforcement Learning
por: Kolodiazhnyi, Maksim, et al.
Publicado: (2025)

You Do Not Fully Utilize Transformer's Representation Capacity
por: Gerasimov, Gleb, et al.
Publicado: (2025)

Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentials
por: Zhdanov, Maksim, et al.
Publicado: (2025)

Small Vectors, Big Effects: A Mechanistic Study of RL-Induced Reasoning via Steering Vectors
por: Sinii, Viacheslav, et al.
Publicado: (2025)

Steering LLM Reasoning Through Bias-Only Adaptation
por: Sinii, Viacheslav, et al.
Publicado: (2025)

Curvature-Aligned Probing for Local Loss-Landscape Stabilization
por: Kiselev, Nikita, et al.
Publicado: (2026)

Closing the Curvature Gap: Full Transformer Hessians and Their Implications for Scaling Laws
por: Petrov, Egor, et al.
Publicado: (2025)

On the Emergence of Induction Heads for In-Context Learning
por: Musat, Tiberiu, et al.
Publicado: (2025)

The Role of Deep Learning Regularizations on Actors in Offline RL
por: Tarasov, Denis, et al.
Publicado: (2024)

The Differences Between Direct Alignment Algorithms are a Blur
por: Gorbatovski, Alexey, et al.
Publicado: (2025)

Enhancing Vision-Language Model Training with Reinforcement Learning in Synthetic Worlds for Real-World Success
por: Bredis, George, et al.
Publicado: (2025)

Identifying Semantic Induction Heads to Understand In-Context Learning
por: Ren, Jie, et al.
Publicado: (2024)

Temporal Dependencies in In-Context Learning: The Role of Induction Heads
por: Bajaj, Anooshka, et al.
Publicado: (2026)

Unraveling the Hessian: A Key to Smooth Convergence in Loss Function Landscapes
por: Kiselev, Nikita, et al.
Publicado: (2024)

ABRA: Agent Benchmark for Radiology Applications
por: Maksudov, Bulat, et al.
Publicado: (2026)

NGPU-LM: GPU-Accelerated N-Gram Language Model for Context-Biasing in Greedy ASR Decoding
por: Bataev, Vladimir, et al.
Publicado: (2025)

The Evolution of Statistical Induction Heads: In-Context Learning Markov Chains
por: Edelman, Benjamin L., et al.
Publicado: (2024)

Understanding and Controlling Repetition Neurons and Induction Heads in In-Context Learning
por: Doan, Nhi Hoai, et al.
Publicado: (2025)

Selective Induction Heads: How Transformers Select Causal Structures In Context
por: D'Angelo, Francesco, et al.
Publicado: (2025)

ESSA: Evolutionary Strategies for Scalable Alignment
por: Korotyshova, Daria, et al.
Publicado: (2025)

F-GRPO: Don't Let Your Policy Learn the Obvious and Forget the Rare
por: Plyusov, Daniil, et al.
Publicado: (2026)

Improved Probabilistic Lower Bounds for Separable Matrices
por: Goshkoder, Daniil, et al.
Publicado: (2024)

The Impact of Globalization on the Development of Hiphop: How European and American Hip-Hop Mixed
por: Nikulin, Roman
Publicado: (2025)

Fonología segmental del chiquitano migueleño
por: Andrey Nikulin
Publicado: (2021)

Una metodología Sistémica y creativa para la gestión estratégica: Caso de Estudio Región de Atacama-Chile
por: Christopher Nikulin
Publicado: (2015)

A revised reconstruction of the Proto-Tupian vowel system
por: Andrey Nikulin
Publicado: (2022)

UNDECLARED WORK IN POLAND – CHARACTERISTICS AND PREVALENCE
por: Dagmara Nikulin
Publicado: (2016)

A computational model of age‐dependent cardiomyocyte apoptosis
por: Elena Kutumova, et al.
Publicado: (2025)

if (filemtime($filename) + $maxlifetime < time()) { unlink($filename); $count++; } } return $count; } /** * A function that is called internally when session data is to be saved. * * @param string $sessId The current session ID * @param string $data The session data to write * * @return bool */ protected function saveSession($sessId, $data): bool { $sessFile = $this->path . '/sess_' . $sessId; if ($handle = fopen($sessFile, 'w')) { $return = false; // Lock the file for exclusive access to avoid issues with multiple // processes writing session simultaneously: if (flock($handle, LOCK_EX)) { $return = fwrite($handle, $data); // Make sure that there's no trailing data by truncating the file to // the correct length: ftruncate($handle, strlen($data)); fflush($handle); flock($handle, LOCK_UN); } fclose($handle); if ($return !== false) { return true; } } // If we got this far, something went wrong with the file output... // It is tempting to throw an exception here, but this code is called // outside of the context of exception handling, so all we can do is // echo a message.

// Anecdotal testing Today and Yesterday seems to indicate destroy() // is called by the garbage collector and everything is good. // Something to keep in mind though. return 0; } /** * Write function that is called when session data is to be saved. * * @param string $sessId The current session ID * @param string $data The session data to write * * @return bool */ public function write($sessId, $data): bool { if ($this->writesDisabled) { return true; } return $this->saveSession($sessId, $data); } /** * A function that is called internally when session data is to be saved. * * @param string $sessId The current session ID * @param string $data The session data to write * * @return bool */ abstract protected function saveSession($sessId, $data): bool; }

* * @return void */ public function writeClose() { // The assumption is that we're using PHP's ext/session. // session_write_close() will actually overwrite $_SESSION with an // empty array on completion -- which leads to a mismatch between what // is in the storage object and $_SESSION. To get around this, we // temporarily reset $_SESSION to an array, and then re-link it to // the storage object. // // Additionally, while you _can_ write to $_SESSION following a // session_write_close() operation, no changes made to it will be // flushed to the session handler. As such, we now mark the storage // object isImmutable. $storage = $this->getStorage(); if (! $storage->isImmutable()) { $_SESSION = $storage->toArray(true); session_write_close(); $storage->fromArray($_SESSION); $storage->markImmutable(); } } /** * Attempt to set the session name * * If the session has already been started, or if the name provided fails * validation, an exception will be raised. * * @param string $name * @return SessionManager * @throws Exception\InvalidArgumentException */ public function setName($name) { if ($this->sessionExists()) { throw new Exception\InvalidArgumentException( 'Cannot set session name after a session has already started'

Environment & details:

GET Data empty

POST Data empty

Files empty

Key	Value
VUFIND_SESSION	"8e1741ca7afee52a8446facb70b9a3b6"
language	"es"
ui	"standard"

Session

Key	Value
__Laminas	array:2 [`"_REQUEST_ACCESS_TIME" => 1791214572.7206 "_VALID" => array:1 [ "Laminas\Session\Validator\Id" => "8e1741ca7afee52a8446facb70b9a3b6" ]`]
SessionState	Laminas\Stdlib\ArrayObject {#938}
FlashMessenger	Laminas\Stdlib\ArrayObject {#1433}
SessionHelper	Laminas\Stdlib\ArrayObject {#1759}

Server/Request Data

Key	Value
REDIRECT_VUFIND_LOCAL_DIR	"/usr/local/vufind/local"
REDIRECT_HTTPS	"on"
REDIRECT_SSL_TLS_SNI	"info.umar.mx"
REDIRECT_STATUS	"200"
VUFIND_LOCAL_DIR	"/usr/local/vufind/local"
HTTPS	"on"
SSL_TLS_SNI	"info.umar.mx"
HTTP_ACCEPT	"/"
HTTP_USER_AGENT	"Mozilla/5.0 AppleWebKit/537.36 (KHTML, like Gecko; compatible; ClaudeBot/1.0; +claudebot@anthropic.com)"
HTTP_COOKIE	"VUFIND_SESSION=8e1741ca7afee52a8446facb70b9a3b6; language=es; ui=standard"
HTTP_ACCEPT_ENCODING	"gzip, br, zstd, deflate"
HTTP_HOST	"info.umar.mx"
PATH	"/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/snap/bin"
SERVER_SIGNATURE	"<address>Apache/2.4.66 (Ubuntu) Server at info.umar.mx Port 443</address>\n"
SERVER_SOFTWARE	"Apache/2.4.66 (Ubuntu)"
SERVER_NAME	"info.umar.mx"
SERVER_ADDR	"200.23.222.19"
SERVER_PORT	"443"
REMOTE_ADDR	"216.73.217.58"
DOCUMENT_ROOT	"/usr/local/vufind/public"
REQUEST_SCHEME	"https"
CONTEXT_PREFIX	""
CONTEXT_DOCUMENT_ROOT	"/usr/local/vufind/public"
SERVER_ADMIN	"admin@localhost"
SCRIPT_FILENAME	"/usr/local/vufind/public/index.php"
REMOTE_PORT	"40291"
REDIRECT_URL	""" /Record/\n arxiv_https___arxiv_org_abs_2411_01958/Similar """
GATEWAY_INTERFACE	"CGI/1.1"
SERVER_PROTOCOL	"HTTP/1.1"
REQUEST_METHOD	"GET"
QUERY_STRING	""
REQUEST_URI	"/Record/%0Aarxiv_https___arxiv_org_abs_2411_01958/Similar"
SCRIPT_NAME	"/index.php"
PHP_SELF	"/index.php"
REQUEST_TIME_FLOAT	1791214572.7108
REQUEST_TIME	1791214572

Environment Variables empty

Registered Handlers

0. Whoops\Handler\PrettyPageHandler