cs.ROOct 3, 2026

Attention-Based Surface Representation Learning for Robot State Prediction and Open-Ended Surface Classification

Authors: Oleg Kushnarev, Alexander Belyaev

Organizations: National Research Tomsk Polytechnic University, Tomsk, 634050 Russia

Abstract

For ground robots operating in outdoor environments, understanding the properties of the underlying terrain is essential for ensuring reliable operation. In most perception-based studies, this problem is formulated as categorical classification with a fixed number of classes defined during training. We propose an approach that enables new surface classes to be added as trainable vectors, which can subsequently be used to address higher-level tasks. By employing a learning paradigm based on predicting the robot's next state in time and using attention blocks, we improved classification accuracy to 98.56% on the Belyaev-Kushnarev dataset and 94.8% on BorealTC.

Explore similar work

CardsList
  1. Robot Learning on Discrete Surfaces: Theory and Applications

    Oct 1, 2026Matteo Dalle Vedove, Fares J. Abu-Dakka, Luigi Palopoli +2Geometric Representation LearningRobot Motion Generation

  2. PileBelief: Persistent Physical State for Interaction-Driven World Modeling

    Sep 19, 2026Hongyi Lin, Song Zhang, Haiquan Liu +3World Models for RoboticsWorld Models