← Search

Vimal Thilak

7 accepted papers

2026

Rethinking JEPA: Compute‑Efficient Video Self-Supervised Learning with Frozen Teachers

ICLR 2026poster

Video Joint Embedding Predictive Architectures (V‑JEPA) learn generalizable off-the-shelf video representations by predicting masked regions in latent space with an exponential moving average (EMA)‑updated teacher. While EMA prevents representation collapse, it complicates scalable model selection a…

Cited by 0SourceScholar
2026

Text-Conditional JEPA for Learning Semantically Rich Visual Representations

ICML 2026poster

Image-based Joint-Embedding Predictive Architecture (I-JEPA) offers a promising approach to visual self-supervised learning through masked feature prediction. However with the inherent visual uncertainty at masked positions, feature prediction remains challenging and may fail to learn semantic repre…

Cited by 0SourceScholar
2025

Parameters vs FLOPs: Scaling Laws for Optimal Sparsity for Mixture-of-Experts Language Models

ICML 2025poster

Scaling the capacity of language models has consistently proven to be a reliable approach for improving performance and unlocking new capabilities. Capacity can be primarily defined by two dimensions: the number of model parameters and the compute per example. While scaling typically involves increa…

Cited by 6SourcePDFScholar
2025

Towards Automatic Assessment of Self-Supervised Speech Models using Rank

ICASSP 2025accepted

This study explores using embedding rank as an unsupervised evaluation metric for general-purpose speech encoders trained via self-supervised learning (SSL). Traditionally, assessing the performance of these encoders is resource-intensive and requires labeled data from the downstream tasks. Inspired…

Cited by 0SourceScholar
2024

How JEPA Avoids Noisy Features: The Implicit Bias of Deep Linear Self Distillation Networks

NeurIPS 2024poster

Two competing paradigms exist for self-supervised learning of data representations. Joint Embedding Predictive Architectures (JEPAs) is a class of architectures in which semantically similar inputs are encoded into representations that are predictive of each other. A recent successful approach…

Cited by 6SourcePDFScholar
2024

LiDAR: Sensing Linear Probing Performance in Joint Embedding SSL Architectures

ICLR 2024spotlight

Joint embedding (JE) architectures have emerged as a promising avenue for ac- quiring transferable data representations. A key obstacle to using JE methods, however, is the inherent challenge of evaluating learned representations without access to a downstream task, and an annotated dataset. Without…

Cited by 7SourcePDFScholar
2024

Vanishing Gradients in Reinforcement Finetuning of Language Models

ICLR 2024poster

Pretrained language models are commonly aligned with human preferences and downstream tasks via reinforcement finetuning (RFT), which refers to maximizing a (possibly learned) reward function using policy gradient algorithms. This work identifies a fundamental optimization obstacle in RFT: we prove…