Contrastive Representations for Temporal Reasoning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ziarko, Alicja, Bortkiewicz, Michal, Zawalski, Michal, Eysenbach, Benjamin, Milos, Piotr
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916975673868288
author Ziarko, Alicja
Bortkiewicz, Michal
Zawalski, Michal
Eysenbach, Benjamin
Milos, Piotr
author_facet Ziarko, Alicja
Bortkiewicz, Michal
Zawalski, Michal
Eysenbach, Benjamin
Milos, Piotr
contents In classical AI, perception relies on learning state-based representations, while planning, which can be thought of as temporal reasoning over action sequences, is typically achieved through search. We study whether such reasoning can instead emerge from representations that capture both perceptual and temporal structure. We show that standard temporal contrastive learning, despite its popularity, often fails to capture temporal structure due to its reliance on spurious features. To address this, we introduce Combinatorial Representations for Temporal Reasoning (CRTR), a method that uses a negative sampling scheme to provably remove these spurious features and facilitate temporal reasoning. CRTR achieves strong results on domains with complex temporal structure, such as Sokoban and Rubik's Cube. In particular, for the Rubik's Cube, CRTR learns representations that generalize across all initial states and allow it to solve the puzzle using fewer search steps than BestFS, though with longer solutions. To our knowledge, this is the first method that efficiently solves arbitrary Cube states using only learned representations, without relying on an external search algorithm.
format Preprint
id arxiv_https___arxiv_org_abs_2508_13113
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Contrastive Representations for Temporal Reasoning
Ziarko, Alicja
Bortkiewicz, Michal
Zawalski, Michal
Eysenbach, Benjamin
Milos, Piotr
Machine Learning
Artificial Intelligence
In classical AI, perception relies on learning state-based representations, while planning, which can be thought of as temporal reasoning over action sequences, is typically achieved through search. We study whether such reasoning can instead emerge from representations that capture both perceptual and temporal structure. We show that standard temporal contrastive learning, despite its popularity, often fails to capture temporal structure due to its reliance on spurious features. To address this, we introduce Combinatorial Representations for Temporal Reasoning (CRTR), a method that uses a negative sampling scheme to provably remove these spurious features and facilitate temporal reasoning. CRTR achieves strong results on domains with complex temporal structure, such as Sokoban and Rubik's Cube. In particular, for the Rubik's Cube, CRTR learns representations that generalize across all initial states and allow it to solve the puzzle using fewer search steps than BestFS, though with longer solutions. To our knowledge, this is the first method that efficiently solves arbitrary Cube states using only learned representations, without relying on an external search algorithm.
title Contrastive Representations for Temporal Reasoning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2508.13113