Using Natural Language Explanations to Improve Robustness of In-context Learning
Xuanli He, Yuxiang Wu, Oana-Maria Camburu, Pasquale Minervini, Pontus Stenetorp
TL;DR
This paper tackles the brittleness of in-context learning when faced with adversarial inputs by introducing X-ICL, a framework that jointly predicts labels and natural language explanations (NLEs) within an in-context setting. It evaluates LLM-generated NLEs in zero-shot and few-shot regimes across eight adversarial NLI/paraphrase datasets and five large models, showing that few-shot ChatGPT-generated NLEs (fs-X-ICL) yield substantial robustness improvements—often exceeding 10 percentage points in accuracy—while zero-shot ChatGPT NLEs (zs-X-ICL) are less reliable. The work also compares against data-selection baselines, finding that NLE-based approaches provide stronger robustness than in-distribution optimization strategies. Qualitative and quantitative analyses reveal that concise, faithful NLEs generated by ChatGPT tend to be more effective than verbatim human-written NLEs, and they scale better across tasks. Overall, the study highlights a practical, scalable path to more robust reasoning with LLMs by leveraging synthetic NLEs in ICL, while acknowledging limits related to explanation faithfulness and generalizability to tasks beyond NLI and paraphrase detection.
Abstract
Recent studies demonstrated that large language models (LLMs) can excel in many tasks via in-context learning (ICL). However, recent works show that ICL-prompted models tend to produce inaccurate results when presented with adversarial inputs. In this work, we investigate whether augmenting ICL with natural language explanations (NLEs) improves the robustness of LLMs on adversarial datasets covering natural language inference and paraphrasing identification. We prompt LLMs with a small set of human-generated NLEs to produce further NLEs, yielding more accurate results than both a zero-shot-ICL setting and using only human-generated NLEs. Our results on five popular LLMs (GPT3.5-turbo, Llama2, Vicuna, Zephyr, and Mistral) show that our approach yields over 6% improvement over baseline approaches for eight adversarial datasets: HANS, ISCS, NaN, ST, PICD, PISP, ANLI, and PAWS. Furthermore, previous studies have demonstrated that prompt selection strategies significantly enhance ICL on in-distribution test sets. However, our findings reveal that these strategies do not match the efficacy of our approach for robustness evaluations, resulting in an accuracy drop of 8% compared to the proposed approach.
