Mechanistic Interpretability of Reinforcement Learning Agents

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Trim, Tristan, Grayston, Triston
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910681278709760
author Trim, Tristan
Grayston, Triston
author_facet Trim, Tristan
Grayston, Triston
contents This paper explores the mechanistic interpretability of reinforcement learning (RL) agents through an analysis of a neural network trained on procedural maze environments. By dissecting the network's inner workings, we identified fundamental features like maze walls and pathways, forming the basis of the model's decision-making process. A significant observation was the goal misgeneralization, where the RL agent developed biases towards certain navigation strategies, such as consistently moving towards the top right corner, even in the absence of explicit goals. Using techniques like saliency mapping and feature mapping, we visualized these biases. We furthered this exploration with the development of novel tools for interactively exploring layer activations.
format Preprint
id arxiv_https___arxiv_org_abs_2411_00867
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Mechanistic Interpretability of Reinforcement Learning Agents
Trim, Tristan
Grayston, Triston
Machine Learning
This paper explores the mechanistic interpretability of reinforcement learning (RL) agents through an analysis of a neural network trained on procedural maze environments. By dissecting the network's inner workings, we identified fundamental features like maze walls and pathways, forming the basis of the model's decision-making process. A significant observation was the goal misgeneralization, where the RL agent developed biases towards certain navigation strategies, such as consistently moving towards the top right corner, even in the absence of explicit goals. Using techniques like saliency mapping and feature mapping, we visualized these biases. We furthered this exploration with the development of novel tools for interactively exploring layer activations.
title Mechanistic Interpretability of Reinforcement Learning Agents
topic Machine Learning
url https://arxiv.org/abs/2411.00867