← Search

Madian Khabsa

18 accepted papers

2026

High Accuracy, Less Talk (HALT): Reliable LLMs through Capability-Aligned Finetuning

ICLR 2026poster

Large Language Models (LLMs) currently respond to every prompt. However, they can produce incorrect answers when they lack knowledge or capability -- a problem known as hallucination. We instead propose post-training an LLM to generate content only when confident in its correctness and to otherwise…

Cited by 0SourceScholar
2024

Effective Long-Context Scaling of Foundation Models

NAACL 2024long

We present an effective recipe to train strong long-context LLMs that are capable of utilizing massive context windows of up to 32,000 tokens. Our models are built through continual pretraining from Llama 2 checkpoints with longer text sequences and on a dataset where long texts are upsampled. We pe…

Cited by 231SourcePDFScholar
2024

MART: Improving LLM Safety with Multi-round Automatic Red-Teaming

NAACL 2024long

Red-teaming is a common practice for mitigating unsafe behaviors in Large Language Models (LLMs), which involves thoroughly assessing LLMs to identify potential flaws and addressing them with responsible and accurate responses.While effective, manual red-teaming is costly, and existing automatic red…

Cited by 99SourcePDFScholar
2024

The Belebele Benchmark: a Parallel Reading Comprehension Dataset in 122 Language Variants

ACL 2024long

We present Belebele, a multiple-choice machine reading comprehension (MRC) dataset spanning 122 language variants. Significantly expanding the language coverage of natural language understanding (NLU) benchmarks, this dataset enables the evaluation of text models in high-, medium-, and low-resource…

2023

Generating Hashtags for Short-form Videos with Guided Signals

ACL 2023long

Short-form video hashtag recommendation (SVHR) aims to recommend hashtags to content creators from videos and corresponding descriptions. Most prior studies regard SVHR as a classification or ranking problem and select hashtags from a set of limited candidates. However, in reality, users can create…

2023

Logical Satisfiability of Counterfactuals for Faithful Explanations in NLI

AAAI 2023technical

Evaluating an explanation's faithfulness is desired for many reasons such as trust, interpretability and diagnosing the sources of model's errors. In this work, which focuses on the NLI task, we introduce the methodology of Faithfulness-through-Counterfactuals, which first generates a counterfactual…

Cited by 14SourcePDFScholar
2023

MUSTIE: Multimodal Structural Transformer for Web Information Extraction

ACL 2023long

The task of web information extraction is to extract target fields of an object from web pages, such as extracting the name, genre and actor from a movie page. Recent sequential modeling approaches have achieved state-of-the-art results on web information extraction. However, most of these methods o…

Cited by 18SourcePDFScholar
2023

MixPAVE: Mix-Prompt Tuning for Few-shot Product Attribute Value Extraction

ACL 2023findings

The task of product attribute value extraction is to identify values of an attribute from product information. Product attributes are important features, which help improve online shopping experience of customers, such as product search, recommendation and comparison. Most existing works only focus…

Cited by 31SourcePDFScholar
2023

Progressive Prompts: Continual Learning for Language Models

ICLR 2023poster

We introduce Progressive Prompts – a simple and efficient approach for continual learning in language models. Our method allows forward transfer and resists catastrophic forgetting, without relying on data replay or a large number of task-specific parameters. Progressive Prompts learns a new soft pr…

2023

Residual Prompt Tuning: improving prompt tuning with residual reparameterization

ACL 2023findings

Prompt tuning is one of the successful approaches for parameter-efficient tuning of pre-trained language models. Despite being arguably the most parameter-efficient (tuned soft prompts constitute <0.1% of total parameters), it typically performs worse than other efficient tuning methods and is quite…

2023

RoAST: Robustifying Language Models via Adversarial Perturbation with Selective Training

EMNLP 2023long findings

Fine-tuning pre-trained language models (LMs) has become the de facto standard in many NLP tasks. Nevertheless, fine-tuned LMs are still prone to robustness issues, such as adversarial robustness and model calibration. Several perspectives of robustness for LMs have been studied independently, but l…

Cited by 0SourcecodeScholar
2023

XLM-V: Overcoming the Vocabulary Bottleneck in Multilingual Masked Language Models

EMNLP 2023long main

Large multilingual language models typically rely on a single vocabulary shared across 100+ languages. As these models have increased in parameter count and depth, vocabulary size has remained largely unchanged. This \textit{vocabulary bottleneck} limits the representational capabilities of multilin…

Cited by 0SourceScholar
2022

Quantifying Adaptability in Pre-trained Language Models with 500 Tasks

NAACL 2022long

When a neural language model (LM) is adapted to perform a new task, what aspects of the task predict the eventual performance of the model? In NLP, systematic features of LM generalization to individual examples are well characterized, but systematic aspects of LM adaptability to new tasks are not n…

Cited by 15SourcePDFScholar
2022

SMARTAVE: Structured Multimodal Transformer for Product Attribute Value Extraction

EMNLP 2022finding

Automatic product attribute value extraction refers to the task of identifying values of an attribute from the product information. Product attributes are essential in improving online shopping experience for customers. Most existing methods focus on extracting attribute values from product title an…

2022

Sparse Distillation: Speeding Up Text Classification by Using Bigger Student Models

NAACL 2022long

Distilling state-of-the-art transformer models into lightweight student models is an effective way to reduce computation cost at inference time. The student models are typically compact transformers with fewer parameters, while expensive operations such as self-attention persist. Therefore, the impr…

2022

UniPELT: A Unified Framework for Parameter-Efficient Language Model Tuning

ACL 2022long

Recent parameter-efficient language model tuning (PELT) methods manage to match the performance of fine-tuning with much fewer trainable parameters and perform especially well when training data is limited. However, different PELT methods may perform rather differently on the same task, making it no…

2021

On Unifying Misinformation Detection

NAACL 2021long

In this paper, we introduce UnifiedM2, a general-purpose misinformation model that jointly models multiple domains of misinformation with a single, unified setup. The model is trained to handle four tasks: detecting news bias, clickbait, fake news, and verifying rumors. By grouping these tasks toget…

Cited by 27SourcePDFScholar
2021

On the Influence of Masking Policies in Intermediate Pre-training

EMNLP 2021main

Current NLP models are predominantly trained through a two-stage “pre-train then fine-tune” pipeline. Prior work has shown that inserting an intermediate pre-training stage, using heuristic masking policies for masked language modeling (MLM), can significantly improve final performance. However, it…