← Search

Yilun Kuang

6 accepted papers

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
2025

Bayesian Optimization of Antibodies Informed by a Generative Model of Evolving Sequences

ICLR 2025spotlight

To build effective therapeutics, biologists iteratively mutate antibody sequences to improve binding and stability. Proposed mutations can be informed by previous measurements or by learning from large antibody databases to predict only typical antibodies. Unfortunately, the space of typical antibod…

2025

Customizing the Inductive Biases of Softmax Attention using Structured Matrices

ICML 2025poster

The core component of attention is the scoring function, which transforms the inputs into low-dimensional queries and keys and takes the dot product of each pair. While the low-dimensional projection improves efficiency, it causes information loss for certain tasks that have intrinsically high-dimen…

Cited by 0SourcePDFScholar
2024

Non-Vacuous Generalization Bounds for Large Language Models

ICML 2024poster

Modern language models can contain billions of parameters, raising the question of whether they can generalize beyond the training data or simply parrot their training corpora. We provide the first non-vacuous generalization bounds for pretrained large language models (LLMs), indicating that languag…

2024

Unlocking Tokens as Data Points for Generalization Bounds on Larger Language Models

NeurIPS 2024spotlight

Large language models (LLMs) with billions of parameters excel at predicting the next token in a sequence. Recent work computes non-vacuous compression-based generalization bounds for LLMs, but these bounds are vacuous for large models at the billion-parameter scale. Moreover, these bounds are obtai…

Cited by 7SourcePDFScholar
2023

Learning Efficient Coding of Natural Images with Maximum Manifold Capacity Representations

NeurIPS 2023poster

The efficient coding hypothesis proposes that the response properties of sensory systems are adapted to the statistics of their inputs such that they capture maximal information about the environment, subject to biological constraints. While elegant, information theoretic properties are notoriously…