kage-bench-fast-known-axis-visual-generalization

name	kage-bench-fast-known-axis-visual-generalization
title	KAGE-Bench: Fast Known-Axis Visual Generalization Evaluation for Reinforcement Learning Agents
version	0.0.2
engine	skillxiv-v0.0.2-claude-opus-4.6
license	MIT
url	https://arxiv.org/abs/2601.14232
keywords	["Agent","Learning"]
description	Pixel-based reinforcement learning agents often fail under purely visual distribution shift even when latent dynamics and rewards are unchanged, but existing benchmarks entangle multiple sources of shift and hinder systematic analysis. We introduce KAGE-Env, a JAX-native 2D platformer that factorizes the observation process into independently controllable visual axes while keeping the underlying control problem fixed. By construction, varying a visual axis affects performance only through the in...