← Search

Thomas TCK Zhang

5 accepted papers

2026

Action Chunking and Data Augmentation Yield Exponential Improvements in Behavior Cloning for Continuous Spaces

ICLR 2026poster

This paper presents a theoretical analysis of two of the most impactful interventions in modern learning from demonstration in robotics and continuous control: the practice of *action-chunking* (predicting sequences of actions in open-loop) and *exploratory augmentation* of expert demonstrations. Th…

Cited by 0SourceScholar
2025

On The Concurrence of Layer-wise Preconditioning Methods and Provable Feature Learning

ICML 2025poster

Layer-wise preconditioning methods are a family of memory-efficient optimization algorithms that introduce preconditioners per axis of each layer's weight tensors. These methods have seen a recent resurgence, demonstrating impressive performance relative to entry-wise ("diagonal") preconditioning me…

Cited by 0SourcePDFScholar
2024

Guarantees for Nonlinear Representation Learning: Non-identical Covariates, Dependent Data, Fewer Samples

ICML 2024poster

A driving force behind the diverse applicability of modern machine learning is the ability to extract meaningful features across many sources. However, many practical domains involve data that are non-identically distributed across sources, and possibly statistically dependent within its source, vio…

Cited by 1SourcePDFScholar
2024

Sample-Efficient Linear Representation Learning from Non-IID Non-Isotropic Data

ICLR 2024spotlight

A powerful concept behind much of the recent progress in machine learning is the extraction of common features across data from heterogeneous sources or tasks. Intuitively, using all of one's data to learn a common representation function benefits both computational effort and statistical generaliza…

Cited by 8SourcePDFScholar