← Search

Aman Madaan

15 accepted papers

2024

AutoMix: Automatically Mixing Language Models

NeurIPS 2024poster

Large language models (LLMs) are now available from cloud API providers in various sizes and configurations. While this diversity offers a broad spectrum of choices, effectively leveraging the options to optimize computational cost and performance remains challenging. In this work, we present AutoMi…

2024

In-Context Principle Learning from Mistakes

ICML 2024poster

In-context learning (ICL, also known as few-shot prompting) has been the standard method of adapting LLMs to downstream tasks, by learning from a few input-output examples. Nonetheless, all ICL-based approaches only learn from correct input-output pairs. In this paper, we revisit this paradigm, by l…

Cited by 29SourcePDFScholar
2024

Learning Performance-Improving Code Edits

ICLR 2024spotlight

With the decline of Moore's law, optimizing program performance has become a major focus of software research. However, high-level optimizations such as API and algorithm changes remain elusive due to the difficulty of understanding the semantics of code. Simultaneously, pretrained large language mo…

2024

Program-Aided Reasoners (Better) Know What They Know

NAACL 2024long

Prior work shows that program-aided reasoning, in which large language models (LLMs) are combined with programs written in programming languages such as Python, can significantly improve accuracy on various reasoning tasks. However, while accuracy is essential, it is also important for such reasoner…

2024

Synatra: Turning Indirect Knowledge into Direct Demonstrations for Digital Agents at Scale

NeurIPS 2024poster

LLMs can now act as autonomous agents that interact with digital environments and complete specific objectives (e.g., arranging an online meeting). However, accuracy is still far from satisfactory, partly due to a lack of large-scale, direct demonstrations for digital tasks. Obtaining supervised dat…

Cited by 19SourcePDFScholar
2023

Let's Sample Step by Step: Adaptive-Consistency for Efficient Reasoning and Coding with LLMs

EMNLP 2023long main

A popular approach for improving the correctness of output from large language models (LLMs) is Self-Consistency - poll the LLM multiple times and output the most frequent solution. Existing Self-Consistency techniques always generate a constant number of samples per question, where a better approac…

Cited by 0SourcecodeScholar
2023

Self-Refine: Iterative Refinement with Self-Feedback

NeurIPS 2023poster

Like humans, large language models (LLMs) do not always generate the best output on their first try. Motivated by how humans refine their written text, we introduce Self-Refine, an approach for improving initial outputs from LLMs through iterative feedback and refinement. The main idea is to generat…

Cited by 1546SourcePDFScholar
2023

What Makes Chain-of-Thought Prompting Effective? A Counterfactual Study

EMNLP 2023long findings

The effectiveness of Chain-of-thought prompting (CoT) has been widely recognized, but the underlying mechanisms behind its success, the reason why it just works for a wide range of tasks, remains an open question. To investigate this, we employ a counterfactual prompting approach, systematically man…

Cited by 49SourceScholar
2022

Conditional set generation using Seq2seq models

EMNLP 2022main

Conditional set generation learns a mapping from an input sequence of tokens to a set. Several NLP tasks, such as entity typing and dialogue emotion tagging, are instances of set generation. Seq2Seq models are a popular choice to model set generation but they treat a set as a sequence and do not ful…

Cited by 10SourcePDFScholar
2022

Language Models of Code are Few-Shot Commonsense Learners

EMNLP 2022main

We address the general task of structured commonsense reasoning: given a natural language input, the goal is to generate a graph such as an event or a reasoning-graph.To employ large language models (LMs) for this task, existing approaches ‘serialize’ the output graph as a flat list of nodes and edg…

2022

Learning to repair: Repairing model output errors after deployment using a dynamic memory of feedback

NAACL 2022findings

Large language models (LMs), while powerful, are not immune to mistakes, but can be difficult to retrain. Our goal is for an LM to continue to improve after deployment, without retraining, using feedback from the user. Our approach pairs an LM with (i) a growing memory of cases where the user identi…

2022

Memory-assisted prompt editing to improve GPT-3 after deployment

EMNLP 2022main

Large LMs such as GPT-3 are powerful, but can commit mistakes that are obvious to humans. For example, GPT-3 would mistakenly interpret “What word is similar to good?” to mean a homophone, while the user intended a synonym. Our goal is to effectively correct such errors via user interactions with th…

2021

Think about it! Improving defeasible reasoning by first modeling the question scenario.

EMNLP 2021main

Defeasible reasoning is the mode of reasoning where conclusions can be overturned by taking into account new evidence. Existing cognitive science literature on defeasible reasoning suggests that a person forms a “mental model” of the problem scenario before answering questions. Our research goal ask…