← Search

Bowen Lei

5 accepted papers

2025

Adaptive Draft-Verification for Efficient Large Language Model Decoding

AAAI 2025technical

Large language model (LLM) decoding involves generating a sequence of tokens based on a given context, where each token is predicted one at a time using the model's learned probabilities. The typical autoregressive decoding method requires a separate forward pass through the model for each token g…

2023

Accelerating Dataset Distillation via Model Augmentation

CVPR 2023highlight

Dataset Distillation (DD), a newly emerging field, aims at generating much smaller but efficient synthetic training datasets from large ones. Existing DD methods based on gradient matching achieve leading performance; however, they are extremely computationally intensive as they require continuously…

2023

Calibrating the Rigged Lottery: Making All Tickets Reliable

ICLR 2023poster

Although sparse training has been successfully used in various deep learning tasks to save memory and reduce inference time, the reliability of the produced sparse models remains unexplored. Previous research has shown that deep neural networks tend to be over-confident, and we find that sparse trai…

2023

Efficient Informed Proposals for Discrete Distributions via Newton’s Series Approximation

AISTATS 2023poster

Gradients have been exploited in proposal distributions to accelerate the convergence of Markov chain Monte Carlo algorithms on discrete distributions. However, these methods require a natural differentiable extension of the target discrete distribution, which often does not exist or does not provid…

2023

Rethinking Data Distillation: Do Not Overlook Calibration

ICCV 2023poster

Neural networks trained on distilled data often produce over-confident output and require correction by calibration methods. Existing calibration methods such as temperature scaling and mixup work well for networks trained on original large-scale data. However, we find that these methods fail to cal…

Cited by 20PDFcodeScholar