PIAST: Rapid Prompting with In-context Augmentation for Scarce Training data

Paweł Batorski, Paul Swoboda


Abstract
LLMs are highly sensitive to prompt design, but handcrafting effective prompts is difficult and often requires intricate crafting of few-shot examples. We propose a fast automatic prompt construction algorithm that augments human instructions by generating a small set of few shot examples. Our method iteratively replaces/drops/keeps few-shot examples using Monte Carlo Shapley estimation of example utility. For faster execution, we use aggressive subsampling and a replay buffer for faster evaluations. Our method can be run using different compute time budgets. Under a limited budget, it outperforms prior automatic prompting methods on text simplification and mathematical reasoning (GSM8K, DeepMath, Math500), while achieving second-best results on classification and summarization and third-best on MedQA. With an extended, yet still modest budget, PIAST sets a new state of the art among automatic prompting methods on classification, simplification, GSM8K, DeepMath, and Math500. Overall, our results suggest that optimizing in-context examples, rather than exhaustively searching over instruction rewrites is the dominant lever for fast and data-efficient prompt engineering. We will release code and data upon acceptance.
Anthology ID:
2026.acl-long.515
Volume:
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
11222–11242
Language:
URL:
https://preview.aclanthology.org/check-for-anonymous-pdfs/2026.acl-long.515/
DOI:
Bibkey:
Cite (ACL):
Paweł Batorski and Paul Swoboda. 2026. PIAST: Rapid Prompting with In-context Augmentation for Scarce Training data. In Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pages 11222–11242, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
PIAST: Rapid Prompting with In-context Augmentation for Scarce Training data (Batorski & Swoboda, ACL 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/check-for-anonymous-pdfs/2026.acl-long.515.pdf
Checklist:
 2026.acl-long.515.checklist.pdf