← Search

Alvin Cheung

10 accepted papers

2026

FrontierCS: Evolving Challenges for Evolving Intelligence

ICML 2026poster

We introduce FrontierCS, a benchmark of 240 open-ended problems across diverse areas of computer science, designed and reviewed by experts, including CS PhDs and top-tier competitive programming participants and problem setters. Unlike existing benchmarks that focus on tasks with known optimal solut…

Cited by 0SourceScholar
2024

AST-T5: Structure-Aware Pretraining for Code Generation and Understanding

ICML 2024poster

Large language models (LLMs) have made significant advancements in code-related tasks, yet many LLMs treat code as simple sequences, neglecting its structured nature. We introduce AST-T5, a novel pretraining paradigm that leverages the Abstract Syntax Tree (AST) for enhanced code generation, transpi…

2024

Evaluation of LLMs on Syntax-Aware Code Fill-in-the-Middle Tasks

ICML 2024oral

We introduce **S**yntax-**A**ware **F**ill-**i**n-the-**M**iddle (SAFIM), a new benchmark for evaluating Large Language Models (LLMs) on the code Fill-in-the-Middle (FIM) task. This benchmark focuses on syntax-aware completions of program structures such as code blocks and conditional expressions, a…

2024

Online Speculative Decoding

ICML 2024poster

Speculative decoding is a pivotal technique to accelerate the inference of large language models (LLMs) by employing a smaller draft model to predict the target model's outputs. However, its efficacy can be limited due to the low predictive accuracy of the draft model, particularly when faced with d…

2024

SlimFit: Memory-Efficient Fine-Tuning of Transformer-based Models Using Training Dynamics

NAACL 2024long

Transformer-based models, such as BERT and ViT, have achieved state-of-the-art results across different natural language processing (NLP) and computer vision (CV) tasks. However, these models are extremely memory intensive during their fine-tuning process, making them difficult to deploy on GPUs wit…

2024

Verified Code Transpilation with LLMs

NeurIPS 2024poster

Domain-specific languages (DSLs) have become integral to various software workflows. Such languages offer domain-specific optimizations and abstractions that improve code readability and maintainability. However, leveraging these languages requires developers to rewrite existing code using the spec…

Cited by 10SourcePDFScholar
2023

Model-Generated Pretraining Signals Improves Zero-Shot Generalization of Text-to-Text Transformers

ACL 2023long

This paper explores the effectiveness of model-generated signals in improving zero-shot generalization of text-to-text Transformers such as T5. We study various designs to pretrain T5 using an auxiliary model to construct more challenging token replacements for the main model to denoise. Key aspects…

2022

GACT: Activation Compressed Training for Generic Network Architectures

ICML 2022spotlight

Training large neural network (NN) models requires extensive memory resources, and Activation Compression Training (ACT) is a promising approach to reduce training memory footprint. This paper presents GACT, an ACT framework to support a broad range of machine learning tasks for generic NN architect…

2021

PlotCoder: Hierarchical Decoding for Synthesizing Visualization Code in Programmatic Context

ACL 2021long

Creating effective visualization is an important part of data analytics. While there are many libraries for creating visualization, writing such code remains difficult given the myriad of parameters that users need to provide. In this paper, we propose the new task of synthesizing visualization prog…