← Search

Zi Yang

10 accepted papers

2026

CIRBench: Evaluating Large Language Models as LLVM IR Optimizers

ICML 2026spotlight

Large language models are beginning to introduce a new paradigm for compilation: instead of only assisting at the source level, they can operate directly on **intermediate representations (IRs)**, the compiler’s internal code representation, Early studies suggest that LLM-guided optimization can som…

Cited by 0SourceScholar
2026

DiaBlo: Diagonal Blocks Are Sufficient For Finetuning

ICLR 2026poster

Fine-tuning is a critical step for adapting large language models (LLMs) to domain-specific downstream tasks. To mitigate the substantial computational and memory costs of full-model fine-tuning, Parameter-Efficient Fine-Tuning (PEFT) methods have been proposed to update only a small subset of model…

Cited by 0SourcecodeScholar
2026

Revisiting Nonstationary Kernel Design for Multi-Output Gaussian Processes

ICLR 2026poster

Multi-output Gaussian processes (MOGPs) provide a Bayesian framework for modeling non-linear functions with multiple outputs, in which nonstationary kernels are essential for capturing input-dependent variations in observations. However, from a spectral (dual) perspective, existing nonstationary ker…

Cited by 0SourceScholar
2025

CoLA: Compute-Efficient Pre-Training of LLMs via Low-Rank Activation

EMNLP 2025

The full-size MLPs and the projection layers in attention introduce tremendous model sizes of large language models (LLMs), consuming extensive computational resources in pre-training. We empirically observe that the activations of pre-trained LLMs exhibit low-rank property. Motivated by such observ

2025

IR-OptSet: An Optimization-Sensitive Dataset for Advancing LLM-Based IR Optimizer

NeurIPS 2025poster

Compiler optimization is essential for improving program performance, yet modern compilers still depend on manually crafted transformation rules over intermediate representations (IRs). As compilers grow in complexity, maintaining these rule-based optimizations becomes increasingly labor-intensive a…

Cited by 0SourceScholar
2025

Multi-View Oriented GPLVM: Expressiveness and Efficiency

NeurIPS 2025poster

The multi-view Gaussian process latent variable model (MV-GPLVM) aims to learn a unified representation from multi-view data but is hindered by challenges such as limited kernel expressiveness and low computational efficiency. To overcome these issues, we first introduce a new duality between the sp…

Cited by 0SourceScholar
2024

CoMERA: Computing- and Memory-Efficient Training via Rank-Adaptive Tensor Optimization

NeurIPS 2024poster

Training large AI models such as LLMs and DLRMs costs massive GPUs and computing time. The high training cost has become only affordable to big tech companies, meanwhile also causing increasing concerns about the environmental impact. This paper presents CoMERA, a **Co**mputing- and **M**emory-**E**…

2024

MagR: Weight Magnitude Reduction for Enhancing Post-Training Quantization

NeurIPS 2024poster

In this paper, we present a simple optimization-based preprocessing technique called Weight Magnitude Reduction (MagR) to improve the performance of post-training quantization. For each linear layer, we adjust the pre-trained floating-point weights by solving an $\ell_\infty$-regularized optimizatio…

2018

Best arm identification in multi-armed bandits with delayed feedback

AISTATS 2018poster

In this paper, we propose a generalization of the best arm identification problem in stochastic multi-armed bandits (MAB) to the setting where every pull of an arm is associated with delayed feedbacks. The delay in feedbacks increases the effective sample complexity of the algorithm, but can be offs…

Cited by 0SourcePDFScholar