← Search

Mohammad Saleh

9 accepted papers

2025

Building Math Agents with Multi-Turn Iterative Preference Learning

ICLR 2025poster

Recent studies have shown that large language models' (LLMs) mathematical problem-solving capabilities can be enhanced by integrating external tools, such as code interpreters, and employing multi-turn Chain-of-Thought (CoT) reasoning. While current methods focus on synthetic data generation and Sup…

Cited by 24SourcePDFScholar
2025

LiPO: Listwise Preference Optimization through Learning-to-Rank

NAACL 2025long

Aligning language models (LMs) with curated human feedback is critical to control their behaviors in real-world applications. Several recent policy optimization methods, such as DPO and SLiC, serve as promising alternatives to the traditional Reinforcement Learning from Human Feedback (RLHF) approac…

Cited by 44SourcePDFScholar
2025

RRM: Robust Reward Model Training Mitigates Reward Hacking

ICLR 2025poster

Reward models (RMs) play a pivotal role in aligning large language models (LLMs) with human preferences. However, traditional RM training, which relies on response pairs tied to specific prompts, struggles to disentangle prompt-driven preferences from prompt-independent artifacts, such as response l…

Cited by 0SourcePDFScholar
2024

Statistical Rejection Sampling Improves Preference Optimization

ICLR 2024poster

Improving the alignment of language models with human preferences remains an active research challenge. Previous approaches have primarily utilized online Reinforcement Learning from Human Feedback (RLHF). Recently, offline methods such as Sequence Likelihood Calibration (SLiC) and Direct Preference…

Cited by 199SourcePDFScholar
2023

Calibrating Sequence likelihood Improves Conditional Language Generation

ICLR 2023poster

Conditional language models are predominantly trained with maximum likelihood estimation (MLE), giving probability mass to sparsely observed target sequences. While MLE trained models assign high probability to plausible sequences given the context, the model probabilities often do not accurately ra…

Cited by 140SourcePDFScholar
2023

Improving the Robustness of Summarization Models by Detecting and Removing Input Noise

EMNLP 2023long findings

The evaluation of abstractive summarization models typically uses test data that is identically distributed as training data. In real-world practice, documents to be summarized may contain input noise caused by text extraction artifacts or data pipeline bugs. The robustness of model performance unde…

Cited by 0SourceScholar
2023

Out-of-Distribution Detection and Selective Generation for Conditional Language Models

ICLR 2023top-25%

Machine learning algorithms typically assume independent and identically distributed samples in training and at test time (IID). Much work has shown that high-performing ML classifiers can degrade significantly and provide overly-confident, wrong classification predictions, particularly for out-of-…

Cited by 106SourcePDFScholar
2020

PEGASUS: Pre-training with Extracted Gap-sentences for Abstractive Summarization

ICML 2020poster

Recent work pre-training Transformers with self-supervised objectives on large text corpora has shown great success when fine-tuned on downstream NLP tasks including text summarization. However, pre-training objectives tailored for abstractive text summarization have not been explored. Furthermore t…