Papers
Topics
Authors
Recent
Search
2000 character limit reached

Domain-Adapted Small Language Models with Hybrid Post-Processing: Achieving Cost-Efficient, Low-Latency Multi-Label Structured Prediction via LoRA Fine-Tuning on Scarce Data

Published 4 Jun 2026 in cs.LG | (2606.05781v2)

Abstract: Deploying frontier LLMs for domain-specific structured evaluation tasks incurs prohibitive latency, cost, and data-privacy overhead. We present a hybrid framework that fine-tunes a small LLM (LLaMA 3.1 8B, 2.05% trainable parameters via LoRA) on only 219 curated examples and couples it with a deterministic rule-based postprocessing layer. Applied to multi-label compliance evaluation of conversational transcripts (18 heterogeneous output fields), our system achieves 100% JSON structural validity, 83.0% human-validated overall accuracy, and 100% accuracy on the most critical classification field in blind evaluation on 53 unseen production transcripts. On a single NVIDIA A100 GPU, inference completes in $\sim$2 seconds -- 2--5x faster than frontier APIs -- at USD 0.013 per evaluation versus USD 0.025--0.055 for proprietary alternatives, yielding 46--76% cost savings. We introduce targeted hard-negative augmentation for critical decision boundaries and formalize the hybrid neural-symbolic decomposition, demonstrating that domain-adapted small LLMs with postprocessing can match frontier model accuracy while dramatically reducing operational cost, latency, and privacy risk.

Summary

  • The paper demonstrates effective domain adaptation by fine-tuning a small LLM with only 2.05% of parameters updated on scarce, targeted data.
  • It integrates a hybrid neural-symbolic approach where a neural model infers outputs and a deterministic rule-based layer enforces strict SOP compliance.
  • Empirical results show 100% accuracy on critical fields, significantly reduced costs, and lower inference latency compared to frontier LLMs.

Domain-Adapted Small LLMs for Cost-Efficient, Low-Latency Multi-Label Structured Prediction

Motivation and Problem Statement

The paper addresses the deployment of AI-powered conversational agents in regulated industries where multi-label compliance evaluation of conversational transcripts is critical. Strict SOPs necessitate high accuracy in structured prediction, yet frontier LLMs such as GPT-4o and Claude Opus incur unsustainable operational costs, elevated inference latency, and significant data privacy concerns in on-premise settings. The authors propose a framework leveraging a small LLM (LLaMA 3.1 8B), fine-tuned via parameter-efficient Low-Rank Adaptation (LoRA) on extremely scarce domain data, integrated with a hybrid deterministic rule-based post-processing layer. The target task is multi-label transcript classification comprising 18 heterogeneous fields with atomic inter-field constraints.

Methodological Innovations

Parameter-Efficient Fine-Tuning

LoRA is utilized for domain adaptation, optimizing only 2.05% (167.8M parameters) of the full 8.2B model. Fine-tuning is performed on just 219 curated examples, notably including targeted hard-negative augmentation for critical decision boundaries—specifically, Agent Disclosure labels. This enables the model to directly encode domain knowledge in its weights, circumventing context window limitations inherent to in-context learning and zero-shot prompting.

Hybrid Neural-Symbolic Inference

Evaluation is decomposed into two stages:

  • Neural: The fine-tuned LLM infers the structured JSON output, handling contextual reasoning tasks such as professional conduct assessment.
  • Symbolic: A deterministic post-processing layer enforces invariant SOP rules, e.g., exact phrase matching for Agent Disclosure and logical consistency checks for inter-field relationships. This architecture concentrates neural capacity on fields demanding natural language understanding while guaranteeing strict compliance via symbolic enforcement for deterministic fields.

Data Augmentation

Hard-negative augmentation is applied to agent disclosure, generating minimally contrastive transcript pairs by modifying only the disclosure phrase. This strategy robustly calibrates the decision boundary, as evidenced by 100% accuracy on the disclosure field in blind evaluations.

Experimental Evaluation

Dataset and Training Regime

A production dataset of 219 labeled conversational transcripts from financial services is split 90/10 for training/evaluation. Transcripts range up to 2,472 tokens, with rigorous SOP-guided auditing for all label fields. Label masking is used during training to restrict loss computation to assistant tokens corresponding to JSON outputs.

Blind Test and Deployment

Validation is conducted on 53 unseen production transcripts, independently scored by domain experts. Inference runs on a single NVIDIA A100 80GB GPU, with greedy decoding and no quantization, completing evaluations in approximately 2 seconds.

Results

Performance Metrics

  • JSON Structural Validity: 100% (compared to 85-95% for zero-shot approaches)
  • Overall Human-Validated Accuracy: 83%
  • Critical Field (Agent Disclosure): 100%
  • Most Behavioral/Informational Fields: 98–100%
  • Compliance Result and Disposition: 90.6%
  • Inference Latency: ∼\sim2 seconds per transcript
  • Cost: \$0.013 per evaluation (46–76% lower than proprietary APIs)

Notably, hard-negative augmentation for Agent Disclosure completely eliminates classification errors for this field. Error analysis highlighted distributional bias in Disposition, suggesting augmentation opportunities.

Cost and Privacy Implications

Compared to API-based approaches, the proposed system yields up to 76% cost reduction and a 2–5×\times decrease in inference latency, with full data sovereignty via on-premise deployment. Few-shot prompting with frontier models is orders-of-magnitude more expensive and slower.

Theoretical Implications

The hybrid decomposition principle validates a generalizable methodology for regulated domains: neural models absorb contextual and ambiguous aspects of compliance, while deterministic symbolic layers enforce strict rules and inter-field constraints. Weight-based domain adaptation confers schema consistency and robustness against attention degradation in long context windows.

Limitations and Future Directions

The paper acknowledges statistical power limits due to the size of blind test sets (n=53), field imbalance, and brittleness in regex-based symbolic rules. Disposition bias was detected, implying the need for further targeted augmentation. Future work may include more sophisticated symbolic rule handling, addressing conversational pattern diversity, and scaling the methodology to broader compliance tasks.

Conclusion

This work demonstrates that LoRA-fine-tuned small LLMs, supplemented by deterministic symbolic post-processing, achieve high-fidelity structured prediction for domain-specific compliance evaluation with minimal training data. The architecture delivers significant operational cost, latency, and privacy advantages over frontier LLMs, and shows that small models with targeted augmentation can substitute for scale in regulated deployment scenarios. The hybrid neural-symbolic blueprint is theoretically compelling for tasks combining language understanding with strict rule enforcement, offering a scalable pathway for practical and privacy-preserving AI in regulated industries.

Paper to Video (Beta)

No one has generated a video about this paper yet.

Whiteboard

No one has generated a whiteboard explanation for this paper yet.

Open Problems

We haven't generated a list of open problems mentioned in this paper yet.

Collections

Sign up for free to add this paper to one or more collections.

Tweets

Sign up for free to view the 1 tweet with 10 likes about this paper.