用小模型实现快速精准的AI内容评估,成本降80倍
Luna-2: Scalable Single-Token Evaluation with Small Language Models
- 用小语言模型+轻量头结构,实现确定性单标记评估
- 在安全与幻觉评测中精度媲美大模型,推理成本降80倍以上
- 适合需要低延迟、本地部署的AI系统安全防护场景
实时内容安全防护要求评估既准确又快速廉价,但现有基于大模型的评判方式(LLMAJ)因多标记生成导致慢、贵且结果不可控。本文提出Luna-2,一种新架构,将解码器仅的小语言模型(SLMs)转化为确定性评估模型,可可靠计算多种任务特定指标(如毒性、幻觉、工具选择质量等),精度与前沿大模型相当,同时大幅降低计算成本和延迟。每个指标以轻量级LoRA/PEFT头形式部署在共享的SLM骨干网络上,支持单块GPU并发运行数百个专用评估指标,实现本地化部署,兼顾隐私与低延迟。在内容安全与幻觉基准测试中,Luna-2达到顶尖大模型评估器的精度,推理成本降低超80倍,延迟降低超20倍。本文详述模型架构、训练方法,并报告真实世界中的准确率、延迟与吞吐量结果。上线后,已保护超1亿次AI会话,每月处理超1000亿标记,年节省评估成本超3000万美元。
原文摘要 · Abstract (English)
Real-time guardrails require evaluation that is accurate, cheap, and fast - yet today's default, LLM-as-a-judge (LLMAJ), is slow, expensive, and operationally non-deterministic due to multi-token generation. We present Luna-2, a novel architecture that leverages decoder-only small language models (SLMs) into a deterministic evaluation model to reliably compute complex task-specific LLMAJ metrics (e.g. toxicity, hallucination, tool selection quality, etc.) at an accuracy at par or higher than LLMAJ using frontier LLMs while drastically reducing the cost and latency of computation. Each metric is implemented as a lightweight LoRA/PEFT head on top of a shared SLM backbone, enabling hundreds of specialized metrics to run concurrently on a single GPU, deployable locally next to AI systems in a privacy-preserving and latency optimizing manner. Across content safety and hallucination benchmarks, Luna-2 matches the accuracy of state-of-the-art LLM-based evaluators while reducing inference cost by over 80x and latency by over 20x. In this paper, we outline the model architecture, training methodology and report real-world empirical results on accuracy, latency, and throughput results. In production, Luna-2 is protecting 100M+ AI sessions and processing over 100B tokens per month for our customers with eval cost savings of over $30M annually.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。