← Search

Jerry Weihong Liu

4 accepted papers

2025

Adaptive Rank Allocation: Speeding Up Modern Transformers with RaNA Adapters

ICLR 2025poster

Large Language Models (LLMs) are computationally intensive, particularly during inference. Neuron-adaptive techniques, which selectively activate neurons in Multi-Layer Perceptron (MLP) layers, offer some speedups but suffer from limitations in modern Transformers. These include reliance on sparse a…

2025

Restructuring Vector Quantization with the Rotation Trick

ICLR 2025oral

Vector Quantized Variational AutoEncoders (VQ-VAEs) are designed to compress a continuous input to a discrete latent space and reconstruct it with minimal distortion. They operate by maintaining a set of vectors---often referred to as the codebook---and quantizing each encoder output to the nearest…

Cited by 3SourcePDFScholar
2025

SD-KDE: Score-Debiased Kernel Density Estimation

NeurIPS 2025poster

We propose a method for density estimation that leverages an estimated score function to debias kernel density estimation (SD-KDE). In our approach, each data point is adjusted by taking a single step along the score function with a specific choice of step size, followed by standard KDE with a modif…

Cited by 0SourceScholar
2025

Towards Learning High-Precision Least Squares Algorithms with Sequence Models

ICLR 2025poster

This paper investigates whether sequence models can learn to perform numerical algorithms, e.g. gradient descent, on the fundamental problem of least squares. Our goal is to inherit two properties of standard algorithms from numerical analysis: (1) machine precision, i.e. we want to obtain solutions…