Automatic Prompt Optimization for Dataset-Level Feature Discovery
Not on the map. Why?
Abstract (arXiv)
Feature extraction from unstructured text is a critical step in many downstream classification pipelines, yet current approaches largely rely on hand-crafted prompts or fixed feature schemas. We formulate feature discovery as a dataset-level prompt optimization problem: given a labelled text corpus, the goal is to induce a global set of interpretable and discriminative feature definitions whose realizations optimize a downstream supervised learning objective. To this end, we propose a multi-agent prompt optimization framework in which language-model agents jointly propose feature definitions, extract feature values, and evaluate feature quality using dataset-level performance and interpretability feedback. Instruction prompts are iteratively refined based on this structured feedback, enabling optimization over prompts that induce shared feature sets rather than per-example predictions. This formulation departs from prior prompt optimization methods that rely on per-sample supervision and provides a principled mechanism for automatic feature discovery from unstructured text.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 42 of its 51 references that are in the map. First posted on arXiv 20 Jan 2026.
Builds on
42 of 51 references.
References · 42
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2025Qwen3 Technical Reportcited by 8,820
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2022Large Language Models are Zero-Shot Reasonerscited by 4,374
- 2019The Curious Case of Neural Text Degenerationcited by 2,916
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,501
- 2025Gemma 3 Technical Reportcited by 1,824
- 2022Large Language Models Are Human-Level Prompt Engineerscited by 1,045
- 2023ChatGPT Outperforms Crowd-Workers for Text-Annotation Taskscited by 870
- 2023Large Language Models as Optimizerscited by 864
- 2023DSPy: Compiling Declarative Language Model Calls into Self-Improving Pipelinescited by 832
30 more
- 2023Automatic Prompt Optimization with "Gradient Descent" and Beam Searchcited by 653
- 2019AutoML: A Survey of the State-of-the-Artcited by 390
- 2024Optimizing Instructions and Demonstrations for Multi-Stage Language Model Programscited by 291
- 2025Small Language Models are the Future of Agentic AIcited by 270
- 2023ToxicChat: Unveiling Hidden Challenges of Toxicity Detection in Real-World User-AI Conversationcited by 231
- 2023Efficient Guided Generation for Large Language Modelscited by 200
- 2023AnnoLLM: Making Large Language Models to Be Better Crowdsourced Annotatorscited by 172
- 2025Multi-Agent Design: Optimizing Agents with Better Prompts and Topologiescited by 113
- 2023ChatGPT-4 Outperforms Experts and Crowd Workers in Annotating Political Twitter Messages with Zero-Shot Learningcited by 100
- 2024BioMedLM: A 2.7B Parameter Language Model Trained On Biomedical Textcited by 86
- 2023Testing the Reliability of ChatGPT for Text Annotation and Classification: A Cautionary Remarkcited by 54
- 2025Feedback Descent: Open-Ended Text Optimization via Pairwise Comparisoncited by 44
- 2024Can Large Language Models be Trusted for Evaluation? Scalable Meta-Evaluation of LLMs as Evaluators via Agent Debatecited by 41
- 2024Dynamic and Adaptive Feature Generation with LLMcited by 27
- 2024Using LLMs for the Extraction and Normalization of Product Attribute Valuescited by 20
- 2023It's Just a Matter of Time: Detecting Depression with Time-Enriched Multimodal Transformerscited by 18
- 2024Large Language Models as Financial Data Annotators: A Study on Effectiveness and Efficiencycited by 18
- 2025The Rise of Small Language Models in Healthcare: A Comprehensive Surveycited by 18
- 2025SLOT: Structuring the Output of Large Language Modelscited by 18
- 2019Listening between the Lines: Learning Personal Attributes from Conversationscited by 16
- 2024LLM-based feature generation from text for interpretable machine learningcited by 13
- 2024CancerLLM: A Large Language Model in Cancer Domaincited by 12
- 2023Knowledge Engineering using Large Language Modelscited by 9
- 2024An LLM Feature-based Framework for Dialogue Constructiveness Assessmentcited by 9
- 2025Enhanced Large Language Models for Effective Screening of Depression and Anxietycited by 9
- 2025Position: Scaling LLM Agents Requires Asymptotic Analysis with LLM Primitivescited by 7
- 2025metaTextGrad: Automatically optimizing language model optimizerscited by 6
- 2025Dr.Copilot: A Multi-Agent Prompt Optimized Assistant for Improving Patient-Doctor Communication in Romaniancited by 5
- 2024PAE: LLM-based Product Attribute Extraction for E-Commerce Fashion Trendscited by 4
- 2024Automated Multi-Label Annotation for Mental Health Illnesses Using Large Language Modelscited by 3
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Structured Prompt Optimization Meets Reinforcement Learning for Global and Local Interpretability over Complex Text
- 2023Hard Prompts Made Easy: Gradient-Based Discrete Optimization for Prompt Tuning and Discovery
- 2026UPA: Unsupervised Prompt Agent via Tree-Based Search and Selection
- 2026Automated Interpretability and Feature Discovery in Language Models with Agents
- 2025Don't Just Demo, Teach Me the Principles: A Principle-Based Multi-Agent Prompting Strategy for Text Classification
- 2023Prompt Algebra for Task Composition
- 2026Optimizing Prompts for Large Language Models: A Causal Approach
- 2023Embroid: Unsupervised Prediction Smoothing Can Improve Few-Shot Classification