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

Pawe{\l} 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/ingest-acl/2026.acl-long.515/
DOI:
Bibkey:
Cite (ACL):
Pawe{\l} 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/ingest-acl/2026.acl-long.515.pdf
Checklist:
 2026.acl-long.515.checklist.pdf