← Search

Gregory Kang Ruey Lau

9 accepted papers

2026

DUET: Optimizing LLM Training Data Mixtures via Noisy Feedback from Unseen, Downstream Evaluation Tasks

ICLR 2026poster

The performance of an LLM depends heavily on the relevance of its training data to the downstream evaluation task. However, in practice, we do not have fine-grained knowledge of the data in the evaluation task (e.g., conversations between an LLM and a user are end-to-end encrypted). Hence, it is unc…

Cited by 0SourcecodeScholar
2026

WaterDrum: Watermark-based Data-centric Unlearning Metric

ICLR 2026poster

Large language model (LLM) unlearning is critical in real-world applications where it is necessary to efficiently remove the influence of private, copyrighted, or harmful data from some users. Existing utility-centric unlearning metrics (based on model utility) may fail to accurately evaluate the ex…

Cited by 0SourcecodeScholar
2025

Dipper: Diversity in Prompts for Producing Large Language Model Ensembles in Reasoning Tasks

EMNLP 2025

Large Language Models (LLMs), particularly smaller variants, still struggle with complex reasoning tasks. While inference-time prompting can guide reasoning, existing methods often rely on sequential queries. Ensemble approaches offer a promising path to performance gains, especially given recent ba

Cited by 0SourcePDFScholar
2025

PIED: Physics-Informed Experimental Design for Inverse Problems

ICLR 2025poster

In many science and engineering settings, system dynamics are characterized by governing partial differential equations (PDEs), and a major challenge is to solve inverse problems (IPs) where unknown PDE parameters are inferred based on observational data gathered under limited budget. Due to the hi…

2025

Uncovering Scaling Laws for Large Language Models via Inverse Problems

EMNLP 2025

Large Language Models (LLMs) are large-scale pretrained models that have achieved remarkable success across diverse domains. These successes have been driven by unprecedented complexity and scale in both data and computations. However, due to the high costs of training such models, brute-force trial

Cited by 0SourcePDFScholar
2024

PINNACLE: PINN Adaptive ColLocation and Experimental points selection

ICLR 2024spotlight

Physics-Informed Neural Networks (PINNs), which incorporate PDEs as soft constraints, train with a composite loss function that contains multiple training point types: different types of collocation points chosen during training to enforce each PDE and initial/boundary conditions, and experimental p…

2024

Position Paper: Data-Centric AI in the Age of Large Language Models

EMNLP 2024finding

This position paper proposes a data-centric viewpoint of AI research, focusing on large language models (LLMs). We start by making a key observation that data is instrumental in the developmental (e.g., pretraining and fine-tuning) and inferential stages (e.g., in-context learning) of LLMs, and advo…

Cited by 1SourcePDFScholar
2024

Waterfall: Scalable Framework for Robust Text Watermarking and Provenance for LLMs

EMNLP 2024main

Protecting intellectual property (IP) of text such as articles and code is increasingly important, especially as sophisticated attacks become possible, such as paraphrasing by large language models (LLMs) or even unauthorized training of LLMs on copyrighted text to infringe such IP. However, existin…

2023

Quantum Bayesian Optimization

NeurIPS 2023poster

Kernelized bandits, also known as Bayesian optimization (BO), has been a prevalent method for optimizing complicated black-box reward functions. Various BO algorithms have been theoretically shown to enjoy upper bounds on their cumulative regret which are sub-linear in the number $T$ of iterations,…

Cited by 18SourcePDFScholar