← Search

Randall Balestriero

54 accepted papers

2026

Causal-JEPA: Learning World Models through Object-Level Latent Interventions

ICML 2026poster

World models require robust relational understanding to support prediction, reasoning, and control. While object-centric representations provide a useful abstraction, they are not sufficient to capture interaction-dependent dynamics. We therefore propose C-JEPA, a simple and flexible object-centric …

Cited by 0SourceScholar
2026

Distilling Geometry Priors for 3D-Consistent Video Generation

ICML 2026poster

While recent video diffusion models (VDMs) produce visually impressive results, they fundamentally struggle to maintain 3D structural consistency, often resulting in object deformation or spatial drift. We hypothesize that these failures arise because standard denoising objectives lack explicit ince…

Cited by 0SourceScholar
2026

Interpreting Physics in Video World Models

ICML 2026poster

A long-standing question in physical reasoning is whether video-based models need to rely on factorized representations of physical variables in order to make physically accurate predictions, or whether they can implicitly represent such variables in a distributed manner. While modern video world mo…

Cited by 0SourceScholar
2026

Into the Rabbit Hull: From Task-Relevant Concepts in DINO to Minkowski Geometry

ICLR 2026poster

DINOv2 sees the world well enough to guide robots and segment images, but we still do not know what it sees. We conduct the first comprehensive analysis of DINOv2’s representational structure using overcomplete dictionary learning, extracting over 32,000 visual concepts in what constitutes the large…

Cited by 0SourceScholar
2026

Joint-Embedding Predictive Learning of Latent Market States in U.S. Equities

ICML 2026poster

We investigate whether Joint-Embedding Predictive Architectures (JEPA) can learn useful representations of U.S. equity markets. We jointly train a permutation-invariant tokenizer that maps each trading day's unordered per-asset features to a fixed set of learned factor tokens, together with a tempor…

Cited by 0SourceScholar
2026

LLM-JEPA: Large Language Models Meet Joint Embedding Predictive Architectures

ICLR 2026poster

Large Language Model (LLM) pretraining, finetuning, and evaluation rely on input-space reconstruction and generative capabilities. Yet, it has been observed in vision that embedding-space training objectives, e.g., with Joint Embedding Predictive Architectures (JEPAs), are far superior to their inpu…

Cited by 0SourcecodeScholar
2026

OpenApps: Simulating Environment Variations to Measure UI Agent Reliability

ICLR 2026oral

Reliability is key to realizing the promise of autonomous UI-agents, multimodal agents that directly interact with the apps humans use, as users must be able to trust an agent to complete a given task. Current evaluations rely on fixed environments---often clones of existing apps--- which are limite…

Cited by 0SourcecodeScholar
2026

Rectified LpJEPA: Joint-Embedding Predictive Architectures with Sparse and Maximum-Entropy Representations

ICML 2026poster

Joint-Embedding Predictive Architectures (JEPA) learn view-invariant representations and admit projection-based distribution matching for collapse preventions. Existing approaches regularize representations towards isotropic Gaussian distributions, but inherently favor dense representations and fail…

Cited by 0SourceScholar
2026

Temporal Straightening for Latent Planning

ICML 2026poster

Learning good representations is essential for latent planning with world models. While pretrained visual encoders provide strong visual features, they are not tailored to planning and contain substantial information which is irrelevant to planning. Inspired by the perceptual straightening hypothesi…

Cited by 0SourceScholar
2025

$\mathbb{X}$-Sample Contrastive Loss: Improving Contrastive Learning with Sample Similarity Graphs

ICLR 2025poster

Learning good representations involves capturing the diverse ways in which data samples relate. Contrastive loss—an objective matching related samples—underlies methods from self-supervised to multimodal learning. Contrastive losses, however, can be viewed more broadly as modifying a similarity grap…

Cited by 0SourcePDFScholar
2025

Beyond [cls]: Exploring the True Potential of Masked Image Modeling Representations

ICCV 2025poster

Masked Image Modeling (MIM) has emerged as a promising approach for Self-Supervised Learning (SSL) of visual representations. However, the out-of-the-box performance of MIMs is typically inferior to competing approaches. Most users cannot afford fine-tuning due to the need for large amounts of data,…

2025

Cross-Entropy Is All You Need To Invert the Data Generating Process

ICLR 2025oral

Supervised learning has become a cornerstone of modern machine learning, yet a comprehensive theory explaining its effectiveness remains elusive. Empirical phenomena, such as neural analogy-making and the linear representation hypothesis, suggest that supervised models can learn interpretable factor…

Cited by 2SourcePDFScholar
2025

Curvature Tuning: Provable Training-free Model Steering From a Single Parameter

NeurIPS 2025poster

The scaling of model and data sizes has reshaped the AI landscape, establishing finetuning pretrained models as the standard paradigm for solving downstream tasks. However, dominant finetuning methods typically rely on weight adaptation, often lack interpretability, and depend on heuristically chose…

Cited by 0SourcecodeScholar
2025

Ditch the Denoiser: Emergence of Noise Robustness in Self-Supervised Learning from Data Curriculum

NeurIPS 2025poster

Self-Supervised Learning (SSL) has become a powerful solution to extract rich representations from unlabeled data. Yet, SSL research is mostly focused on clean, curated and high-quality datasets. As a result, applying SSL on noisy data remains a challenge, despite being crucial to applications such…

Cited by 0SourcecodeScholar
2025

FastDINOv2: Frequency Based Curriculum Learning Improves Robustness and Training Speed

NeurIPS 2025poster

Large-scale vision foundation models such as DINOv2 boast impressive performances by leveraging massive architectures and training datasets. The expense of large-scale pre-training puts such research out of reach for many, hence limiting scientific advancements. We thus propose a novel pretraining s…

Cited by 0SourceScholar
2025

From Linearity to Non-Linearity: How Masked Autoencoders Capture Spatial Correlations

ICCV 2025poster

Masked Autoencoders (MAEs) have emerged as a powerful pretraining technique for vision foundation models. Despite their effectiveness, they require extensive hyperparameter tuning (masking ratio, patch size, encoder/decoder layers) when applied to novel datasets. While prior theoretical works have a…

Cited by 0SourcePDFScholar
2025

Joint‑Embedding vs Reconstruction: Provable Benefits of Latent Space Prediction for Self‑Supervised Learning

NeurIPS 2025spotlight

Reconstruction and joint-embedding have emerged as two leading paradigms in Self‑Supervised Learning (SSL). Reconstruction methods focus on recovering the original sample from a different view in input space. On the other hand, joint-embedding methods align the representations of different views in…

Cited by 0SourceScholar
2025

Learning from Reward-Free Offline Data: A Case for Planning with Latent Dynamics Models

NeurIPS 2025poster

A long-standing goal in AI is to develop agents capable of solving diverse tasks across a range of environments, including those never seen during training. Two dominant paradigms address this challenge: (i) reinforcement learning (RL), which learns policies via trial and error, and (ii) optimal con…

Cited by 0SourceScholar
2025

MITIGATING OVER-EXPLORATION IN LATENT SPACE OPTIMIZATION USING LES

ICML 2025poster

We develop Latent Exploration Score (LES) to mitigate over-exploration in Latent Space Optimization (LSO), a popular method for solving black-box discrete optimization problems. LSO utilizes continuous optimization within the latent space of a Variational Autoencoder (VAE) and is known to be suscept…

Cited by 0SourcePDFScholar
2025

No Location Left Behind: Measuring and Improving the Fairness of Implicit Representations for Earth Data

ICLR 2025poster

Implicit neural representations (INRs) exhibit growing promise in addressing Earth representation challenges, ranging from emissions monitoring to climate modeling. However, existing methods disproportionately prioritize global average performance, whereas practitioners require fine-grained insights…

Cited by 0SourcePDFScholar
2025

Position: An Empirically Grounded Identifiability Theory Will Accelerate Self Supervised Learning Research

ICML 2025poster

Self-Supervised Learning (SSL) powers many current AI systems. As research interest and investment grow, the SSL design space continues to expand. The Platonic view of SSL, following the Platonic Representation Hypothesis (PRH), suggests that despite different methods and engineering approaches, all…

Cited by 0SourcePDFScholar
2024

Characterizing Large Language Model Geometry Helps Solve Toxicity Detection and Generation

ICML 2024poster

Large Language Models (LLMs) drive current AI breakthroughs despite very little being known about their internal representations. In this work, we propose to shed the light on LLMs inner mechanisms through the lens of geometry. In particular, we develop in closed form $(i)$ the intrinsic dimension i…

2024

UniBench: Visual Reasoning Requires Rethinking Vision-Language Beyond Scaling

NeurIPS 2024poster

Significant research efforts have been made to scale and improve vision-language model (VLM) training approaches. Yet, with an ever-growing number of benchmarks, researchers are tasked with the heavy burden of implementing each protocol, bearing a non-trivial computational cost, and making sense of…

2023

Active Self-Supervised Learning: A Few Low-Cost Relationships Are All You Need

ICCV 2023poster

Self-Supervised Learning (SSL) has emerged as the solution of choice to learn transferable representations from unlabeled data. However, SSL requires to build samples that are known to be semantically akin, i.e. positive views. Requiring such knowledge is the main limitation of SSL and is often tac…

Cited by 18PDFcodeScholar
2023

An Information Theory Perspective on Variance-Invariance-Covariance Regularization

NeurIPS 2023poster

Variance-Invariance-Covariance Regularization (VICReg) is a self-supervised learning (SSL) method that has shown promising results on a variety of tasks. However, the fundamental mechanisms underlying VICReg remain unexplored. In this paper, we present an information-theoretic perspective on the VIC…

Cited by 8SourcePDFScholar
2023

ImageNet-X: Understanding Model Mistakes with Factor of Variation Annotations

ICLR 2023top-25%

Deep learning vision systems are widely deployed across applications where reliability is critical. However, even today's best models can fail to recognize an object when its pose, lighting, or background varies. While existing benchmarks surface examples challenging for models, they do not explain…

Cited by 50SourcePDFScholar
2023

RankMe: Assessing the Downstream Performance of Pretrained Self-Supervised Representations by Their Rank

ICML 2023oral

Joint-Embedding Self Supervised Learning (JE-SSL) has seen a rapid development, with the emergence of many method variations but only few principled guidelines that would help practitioners to successfully deploy them. The main reason for that pitfall comes from JE-SSL's core principle of not employ…

Cited by 90SourcePDFScholar
2023

SplineCam: Exact Visualization and Characterization of Deep Network Geometry and Decision Boundaries

CVPR 2023highlight

Current Deep Network (DN) visualization and interpretability methods rely heavily on data space visualizations such as scoring which dimensions of the data are responsible for their associated prediction or generating new data features or samples that best match a given DN unit or representation. In…

2023

The SSL Interplay: Augmentations, Inductive Bias, and Generalization

ICML 2023poster

Self-supervised learning (SSL) has emerged as a powerful framework to learn representations from raw data without supervision. Yet in practice, engineers face issues such as instability in tuning optimizers and collapse of representations during training. Such challenges motivate the need for a theo…

Cited by 41SourcePDFScholar
2023

The hidden uniform cluster prior in self-supervised learning

ICLR 2023poster

A successful paradigm in representation learning is to perform self-supervised pretraining using tasks based on mini-batch statistics; (e.g., SimCLR, VICReg, SwAV, MSN). We show that in the formulation of all these methods is an overlooked prior to learn features that enable uniform clustering of th…

Cited by 60SourcePDFScholar
2023

Understanding the detrimental class-level effects of data augmentation

NeurIPS 2023poster

Data augmentation (DA) encodes invariance and provides implicit regularization critical to a model's performance in image classification tasks. However, while DA improves average accuracy, recent studies have shown that its impact can be highly class dependent: achieving optimal average accuracy com…

Cited by 13SourcePDFScholar
2022

A Data-Augmentation Is Worth A Thousand Samples: Analytical Moments And Sampling-Free Training

NeurIPS 2022accept

Data-Augmentation (DA) is known to improve performance across tasks and datasets. We propose a method to theoretically analyze the effect of DA and study questions such as: how many augmented samples are needed to correctly estimate the information encoded by that DA? How does the augmentation polic…

Cited by 18SourcePDFScholar
2022

Contrastive and Non-Contrastive Self-Supervised Learning Recover Global and Local Spectral Embedding Methods

NeurIPS 2022accept

Self-Supervised Learning (SSL) surmises that inputs and pairwise positive relationships are enough to learn meaningful representations. Although SSL has recently reached a milestone: outperforming supervised methods in many modalities\dots the theoretical foundations are limited, method-specific, an…

Cited by 156SourcePDFScholar
2022

MaGNET: Uniform Sampling from Deep Generative Network Manifolds Without Retraining

ICLR 2022poster

Deep Generative Networks (DGNs) are extensively employed in Generative Adversarial Networks (GANs), Variational Autoencoders (VAEs), and their variants to approximate the data manifold, and data distribution on that manifold. However, training samples are often obtained based on preferences, costs,…

2022

No More Than 6ft Apart: Robust K-Means via Radius Upper Bounds

ICASSP 2022accepted

Centroid based clustering methods such as k-means, k-medoids and k-centers are heavily applied as a go-to tool in exploratory data analysis. In many cases, those methods are used to obtain representative centroids of the data manifold for visualization or summarization of a dataset. Real world datas…

Cited by 0SourceScholar
2022

Polarity Sampling: Quality and Diversity Control of Pre-Trained Generative Networks via Singular Values

CVPR 2022oral

We present Polarity Sampling, a theoretically justified plug-and-play method for controlling the generation quality and diversity of any pre-trained deep generative network (DGN). Leveraging the fact that DGNs are, or can be approximated by, continuous piecewise affine splines, we derive the analyti…

Cited by 38PDFcodeScholar
2022

The Effects of Regularization and Data Augmentation are Class Dependent

NeurIPS 2022accept

Regularization is a fundamental technique to prevent over-fitting and to improve generalization performances by constraining a model's complexity. Current Deep Networks heavily rely on regularizers such as Data-Augmentation (DA) or weight-decay, and employ structural risk minimization, i.e. cross-va…

Cited by 123SourcePDFScholar
2022

projUNN: efficient method for training deep networks with unitary matrices

NeurIPS 2022accept

In learning with recurrent or very deep feed-forward networks, employing unitary matrices in each layer can be very effective at maintaining long-range stability. However, restricting network parameters to be unitary typically comes at the cost of expensive parameterizations or increased training ru…

2021

Wearing A Mask: Compressed Representations of Variable-Length Sequences Using Recurrent Neural Tangent Kernels

ICASSP 2021accepted

High dimensionality poses many challenges to the use of data, from visualization and interpretation, to prediction and storage for historical preservation. Techniques abound to reduce the dimensionality of fixed-length sequences, yet these methods rarely generalize to variable-length sequences. To a…

Cited by 0SourceScholar
2020

Analytical Probability Distributions and Exact Expectation-Maximization for Deep Generative Networks

NeurIPS 2020poster

Deep Generative Networks (DGNs) with probabilistic modeling of their output and latent space are currently trained via Variational Autoencoders (VAEs). In the absence of a known analytical form for the posterior and likelihood expectation, VAEs resort to approximations, including (Amortized) Variati…

Cited by 8SourcePDFScholar
2019

From Hard to Soft: Understanding Deep Network Nonlinearities via Vector Quantization and Statistical Inference

ICLR 2019poster

Nonlinearity is crucial to the performance of a deep (neural) network (DN). To date there has been little progress understanding the menagerie of available nonlinearities, but recently progress has been made on understanding the r\^{o}le played by piecewise affine and convex nonlinearities like the…

Cited by 20SourcePDFScholar
2019

The Geometry of Deep Networks: Power Diagram Subdivision

NeurIPS 2019poster

We study the geometry of deep (neural) networks (DNs) with piecewise affine and convex nonlinearities. The layers of such DNs have been shown to be max-affine spline operators (MASOs) that partition their input space and apply a region-dependent affine mapping to their input to produce their output…

2018

Spline Filters For End-to-End Deep Learning

ICML 2018oral

We propose to tackle the problem of end-to-end learning for raw waveform signals by introducing learnable continuous time-frequency atoms. The derivation of these filters is achieved by defining a functional space with a given smoothness order and boundary conditions. From this space, we derive the…

2017

Fast Chirplet Transform to Enhance CNN Machine Listening - Validation on Animal calls and Speech

ICLR 2017workshop

The scattering framework offers an optimal hierarchical convolutional decomposition according to its kernels. Convolutional Neural Net (CNN) can be seen asan optimal kernel decomposition, nevertheless it requires large amount of trainingdata to learn its kernels. We propose a trade-off between the…

Cited by 5SourceScholar