MACA: A Framework for Distilling Trustworthy LLMs into Efficient Retrievers
This work addresses the need for efficient and accurate retrieval in enterprise settings, offering a domain-specific solution that is incremental by building on existing distillation and ranking methods.
The paper tackles the problem of handling short, underspecified queries in enterprise retrieval systems by proposing MACA, a framework that distills a trustworthy LLM re-ranker into an efficient student retriever, avoiding costly online LLM calls. On a proprietary banking FAQ corpus, the MACA student improved MiniLM Accuracy@1 from 0.23 to 0.48.
Modern enterprise retrieval systems must handle short, underspecified queries such as ``foreign transaction fee refund'' and ``recent check status''. In these cases, semantic nuance and metadata matter but per-query large language model (LLM) re-ranking and manual labeling are costly. We present Metadata-Aware Cross-Model Alignment (MACA), which distills a calibrated metadata aware LLM re-ranker into a compact student retriever, avoiding online LLM calls. A metadata-aware prompt verifies the teacher's trustworthiness by checking consistency under permutations and robustness to paraphrases, then supplies listwise scores, hard negatives, and calibrated relevance margins. The student trains with MACA's MetaFusion objective, which combines a metadata conditioned ranking loss with a cross model margin loss so it learns to push the correct answer above semantically similar candidates with mismatched topic, sub-topic, or entity. On a proprietary consumer banking FAQ corpus and BankFAQs, the MACA teacher surpasses a MAFA baseline at Accuracy@1 by five points on the proprietary set and three points on BankFAQs. MACA students substantially outperform pretrained encoders; e.g., on the proprietary corpus MiniLM Accuracy@1 improves from 0.23 to 0.48, while keeping inference free of LLM calls and supporting retrieval-augmented generation.