cs.AIMar 17, 2026

POaaS: Minimal-Edit Prompt Optimization as a Service to Lift Accuracy and Cut Hallucinations on On-Device sLLMs

Authors: Jungwoo Shim, Dae Won Kim, Sun Wook Kim, Soo Young Kim, Myungcheol Lee, Jae-geun Cha, Hyunhwa Choi

Organizations: Electronics and Telecommunications Research Institute, Republic of Korea

Abstract

Small language models (sLLMs) are increasingly deployed on-device, where imperfect user prompts--typos, unclear intent, or missing context--can trigger factual errors and hallucinations. Existing automatic prompt optimization (APO) methods were designed for large cloud LLMs and rely on search that often produces long, structured instructions; when executed under an on-device constraint where the same small model must act as optimizer and solver, these pipelines can waste context and even hurt accuracy. We propose POaaS, a minimal-edit prompt optimization layer that routes each query to lightweight specialists (Cleaner, Paraphraser, Fact-Adder) and merges their outputs under strict drift and length constraints, with a conservative skip policy for well-formed prompts. Under a strict fixed-model setting with Llama-3.2-3B-Instruct and Llama-3.1-8B-Instruct, POaaS improves both task accuracy and factuality while representative APO baselines degrade them, and POaaS recovers up to +7.4% under token deletion and mixup. Overall, per-query conservative optimization is a practical alternative to search-heavy APO for on-device sLLMs.

Figures & tables

Appendix figures & tables1 asset

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. FAPO: Fully Automated Prompt Optimization of Multi-Step LLM Pipelines

    Jun 17, 2026Paul Kassianik, Baturay Saglam, Huaibo Zhao +4Large Language Model-Guided OptimizationLLM Agent Workflow Optimization

  2. From Monolithic to Modular: Segment-level Automatic Prompt Optimization

    Jul 21, 2026Nikita Kulin, Viktor Zhuravlev, Artur Khairullin +4Large Language Model-Guided OptimizationAutomatic Prompt Optimization

  3. Adaptive Prompt Structure Factorization: A Framework for Self-Discovering and Optimizing Compositional Prompt Programs

    Apr 8, 2026Haoyue Liu, Zhichao Wang, Yongxin Guo +2LLM Input OptimizationLLM Prompting