← Search

Rongzhi Zhang

10 accepted papers

2025

DORM: Preference Data Weights Optimization for Reward Modeling in LLM Alignment

EMNLP 2025

Aligning large language models (LLMs) with human preferences relies heavily on high-quality reward models. However, existing approaches struggle with two critical challenges: noisy preference labels and the varying importance of preference samples. We introduce DORM, a method that enhances reward mo

Cited by 0SourcePDFScholar
2025

Hephaestus: Improving Fundamental Agent Capabilities of Large Language Models through Continual Pre-Training

NAACL 2025long

Due to the scarcity of agent-oriented pre-training data, LLM-based autonomous agents typically rely on complex prompting or extensive fine-tuning, which often fails to introduce new capabilities while preserving strong generalizability. We introduce Hephaestus-Forge, the first large-scale pre-traini…

Cited by 1SourcePDFScholar
2025

MLE-Dojo: Interactive Environments for Empowering LLM Agents in Machine Learning Engineering

NeurIPS 2025poster

We introduce MLE-Dojo, a Gym-style framework for systematically reinforcement learning, evaluating, and improving autonomous large language model (LLM) agents in iterative machine learning engineering (MLE) workflows. Unlike existing benchmarks that primarily rely on static datasets or single-attemp…

Cited by 0SourcecodeScholar
2024

Aligning Large Language Models with Representation Editing: A Control Perspective

NeurIPS 2024poster

Aligning large language models (LLMs) with human objectives is crucial for real-world applications. However, fine-tuning LLMs for alignment often suffers from unstable training and requires substantial computing resources. Test-time alignment techniques, such as prompting and guided decoding, do not…

2024

PLaD: Preference-based Large Language Model Distillation with Pseudo-Preference Pairs

ACL 2024findings

Large Language Models (LLMs) have exhibited impressive capabilities in various tasks, yet their vast parameter sizes restrict their applicability in resource-constrained settings. Knowledge distillation (KD) offers a viable solution by transferring expertise from large teacher models to compact stud…

Cited by 5SourcePDFScholar
2024

ProgGen: Generating Named Entity Recognition Datasets Step-by-step with Self-Reflexive Large Language Models

ACL 2024findings

Although Large Language Models (LLMs) exhibit remarkable adaptability across domains, these models often fall short in structured knowledge extraction tasks such as named entity recognition (NER). This paper explores an innovative, cost-efficient strategy to harness LLMs with modest NER capabilities…

2023

Cold-Start Data Selection for Better Few-shot Language Model Fine-tuning: A Prompt-based Uncertainty Propagation Approach

ACL 2023long

We present PATRON, a prompt-based data selection method for pre-trained language model fine-tuning under cold-start scenarios, i.e., no initial labeled data are available. In PATRON, we design (1) a prompt-based uncertainty propagation approach to estimate the importance of data points and (2) a par…

2023

ReGen: Zero-Shot Text Classification via Training Data Generation with Progressive Dense Retrieval

ACL 2023findings

With the development of large language models (LLMs), zero-shot learning has attracted much attention for various NLP tasks. Different from prior works that generate training data with billion-scale natural language generation (NLG) models, we propose a retrieval-enhanced framework to create trainin…

2022

AcTune: Uncertainty-Based Active Self-Training for Active Fine-Tuning of Pretrained Language Models

NAACL 2022long

Although fine-tuning pre-trained language models (PLMs) renders strong performance in many NLP tasks, it relies on excessive labeled data. Recently, researchers have resorted to active fine-tuning for enhancing the label efficiency of PLM fine-tuning, but existing methods of this type usually ignore…

2022

Prompt-Based Rule Discovery and Boosting for Interactive Weakly-Supervised Learning

ACL 2022long

Weakly-supervised learning (WSL) has shown promising results in addressing label scarcity on many NLP tasks, but manually designing a comprehensive, high-quality labeling rule set is tedious and difficult. We study interactive weakly-supervised learning—the problem of iteratively and automatically d…