activity
20162026
most citedBatch Normalization Explained

12 citations · 67 across the 40 of their papers we have counts for

collaborators

42 papers

cs.LG2026

Behavioral Monitoring of JEPA World Models with Jacobian Centroids

Thomas Walker, Randall Balestriero, Richard Baraniuk

Detecting failures in World Model (WM)-based planning requires monitoring whether the model is behaviorally aligned with the current task, which in turn requires studying its inter…

cs.LG2026

The Geometric Structure of Models Learning Sparse Data

Thomas Walker, T. Mitchell Roddenberry, Ahmed Imtiaz Humayun +2

The manifold hypothesis (MH) is often used to explain how machine learning can overcome the curse of dimensionality. However, the MH is only applicable in regimes where the trainin…

cs.LG2026

The Linear Centroids Hypothesis: Features as Directions Learned by Local Experts

Thomas Walker, Ahmed Imtiaz Humayun, Randall Balestriero +1

The Linear Representation Hypothesis (LRH) identifies features of a trained deep network (DN) as linear directions in the activation spaces, i.e., output spaces of intermediate lay…

cs.LG2025

GrokAlign: Geometric Characterisation and Acceleration of Grokking

Thomas Walker, Ahmed Imtiaz Humayun, Randall Balestriero +1

A key challenge for the machine learning community is to understand and accelerate the training dynamics of deep networks that lead to delayed generalisation and emergent robustnes…

cs.LG2024

On the Geometry of Deep Learning

Randall Balestriero, Ahmed Imtiaz Humayun, Richard Baraniuk

In this paper, we overview one promising avenue of progress at the mathematical foundation of deep learning: the connection between deep networks and function approximation by affi…

cs.LG2024★ 1 cited

Mitigating over-exploration in latent space optimization using LES

Omer Ronen, Ahmed Imtiaz Humayun, Richard Baraniuk +2

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. LS…

cs.LG2024

Deep Networks Always Grok and Here is Why

Ahmed Imtiaz Humayun, Randall Balestriero, Richard Baraniuk

Grokking, or delayed generalization, is a phenomenon where generalization in a deep neural network (DNN) occurs long after achieving near zero training error. Previous studies have…

cs.AI2023★ 2 cited

Characterizing Large Language Model Geometry Helps Solve Toxicity Detection and Generation

Randall Balestriero, Romain Cosentino, Sarath Shekkizhar

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…

cs.LG2023

Training Dynamics of Deep Network Linear Regions

Ahmed Imtiaz Humayun, Randall Balestriero, Richard Baraniuk

The study of Deep Network (DN) training dynamics has largely focused on the evolution of the loss function, evaluated on or around train and test set data points. In fact, many DN…

cs.CV2023

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

Ahmed Imtiaz Humayun, Randall Balestriero, Guha Balakrishnan +1

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 thei…

cs.LG2022★ 12 cited

Batch Normalization Explained

Randall Balestriero, Richard G. Baraniuk

A critically important, ubiquitous, and yet poorly understood ingredient in modern deep networks (DNs) is batch normalization (BN), which centers and normalizes the feature maps. T…

stat.AP2022★ 7 cited

DeepTensor: Low-Rank Tensor Decomposition with Deep Network Priors

Vishwanath Saragadam, Randall Balestriero, Ashok Veeraraghavan +1

DeepTensor is a computationally efficient framework for low-rank decomposition of matrices and tensors using deep generative networks. We decompose a tensor as the product of low-r…

cs.LG2022

Singular Value Perturbation and Deep Network Optimization

Rudolf H. Riedi, Randall Balestriero, Richard G. Baraniuk

We develop new theoretical results on matrix perturbation to shed light on the impact of architecture on the performance of a deep network. In particular, we explain analytically w…

cs.LG2022

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

Ahmed Imtiaz Humayun, Randall Balestriero, Anastasios Kyrillidis +1

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 t…

cs.CV2022★ 3 cited

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

Ahmed Imtiaz Humayun, Randall Balestriero, Richard Baraniuk

We present Polarity Sampling, a theoretically justified plug-and-play method for controlling the generation quality and diversity of pre-trained deep generative networks DGNs). Lev…

cs.LG2022

NeuroView-RNN: It's About Time

CJ Barberan, Sina Alemohammad, Naiming Liu +2

Recurrent Neural Networks (RNNs) are important tools for processing sequential data such as time-series or video. Interpretability is defined as the ability to be understood by a p…

cs.LG2022

Spatial Transformer K-Means

Romain Cosentino, Randall Balestriero, Yanis Bahroun +3

K-means defines one of the most employed centroid-based clustering algorithms with performances tied to the data's embedding. Intricate data embeddings have been designed to push $…

cs.CV2021

NeuroView: Explainable Deep Network Decision Making

CJ Barberan, Randall Balestriero, Richard G. Baraniuk

Deep neural networks (DNs) provide superhuman performance in numerous computer vision tasks, yet it remains unclear exactly which of a DN's units contribute to a particular decisio…

cs.LG2021★ 2 cited

MaGNET: Uniform Sampling from Deep Generative Network Manifolds Without Retraining

Ahmed Imtiaz Humayun, Randall Balestriero, Richard Baraniuk

Deep Generative Networks (DGNs) are extensively employed in Generative Adversarial Networks (GANs), Variational Autoencoders (VAEs), and their variants to approximate the data mani…

cs.LG2021

Fast Jacobian-Vector Product for Deep Networks

Randall Balestriero, Richard Baraniuk

Jacobian-vector products (JVPs) form the backbone of many recent developments in Deep Networks (DNs), with applications including faster constrained optimization, regularization wi…

cs.LG2021

Max-Affine Spline Insights Into Deep Network Pruning

Haoran You, Randall Balestriero, Zhihan Lu +6

In this paper, we study the importance of pruning in Deep Networks (DNs) and the yin & yang relationship between (1) pruning highly overparametrized DNs that have been trained from…

cs.CV2020

Interpretable Image Clustering via Diffeomorphism-Aware K-Means

Romain Cosentino, Randall Balestriero, Yanis Bahroun +3

We design an interpretable clustering algorithm aware of the nonlinear structure of image manifolds. Our approach leverages the interpretability of -means applied in the image s…

stat.ML2020

Sparse Multi-Family Deep Scattering Network

Romain Cosentino, Randall Balestriero

In this work, we propose the Sparse Multi-Family Deep Scattering Network (SMF-DSN), a novel architecture exploiting the interpretability of the Deep Scattering Network (DSN) and im…

cs.LG2020

Enhanced Recurrent Neural Tangent Kernels for Non-Time-Series Data

Sina Alemohammad, Randall Balestriero, Zichao Wang +1

Kernels derived from deep neural networks (DNNs) in the infinite-width regime provide not only high performance in a range of machine learning tasks but also new theoretical insigh…