用多个小模型并行预测分类结果,提升效率同时保持准确
A Multi-Model Adaptation of Speculative Decoding for Classification
- 多个轻量级模型并行预测标签,多数一致则直接采纳
- 30亿参数小模型与70亿模型效果接近,准确率超80%且提速2.8至9倍
- 适合追求高推理速度的分类任务,尤其适合资源受限场景
本研究提出一种将推测解码技术从生成任务迁移至分类任务的新方法。设计了一个多模型框架,包含最多三个轻量级工作模型和一个更强大的判别模型,分别类比于推测解码中的草稿模型与目标模型。工作模型负责对输入进行独立的离散类别预测;当多数工作模型意见一致时,直接采纳该标签,避免调用计算开销大的判别模型。在存在分歧时,由判别模型介入决策。该方法减少冗余计算,利用多模型冗余增强置信度,并将判别模型仅用于复杂情形,实现效率与准确性的平衡。分析表明,未经微调的30亿参数(3B)工作模型,在简单及高阶推理任务中,其与判别模型的对齐程度可媲美经过微调的70亿参数(7B)模型。表现最佳的3B模型对对组合在情感分类上达成约80%-83%的共识率,在相似票务任务上为50%-80%。此外,3B模型组合相较判别模型实现2.8x至9x的速度提升,而7B模型组合速度提升范围为1.28x至0.28x(部分低于1,即慢于基准)。
原文摘要 · Abstract (English)
The current study introduces a novel adaptation of speculative decoding, repurposed from generation to classification tasks. We propose a multi-model framework employing up to three lightweight worker models and a single, more robust judge model analogous to draft models and target model, respectively, in speculative decoding. The worker models, tasked with the bulk of the computation, independently predict discrete class labels for a given input. When majority worker models agree on a label, it is accepted as the final label, optimizing efficiency by bypassing the computationally expensive judge model. In cases of disagreement, the judge model intervenes to resolve the label. This approach minimizes redundant computation, leverages the redundancy of multiple workers for confidence, and confines the judge model's role to challenging cases, offering a practical balance of efficiency and accuracy. Our analysis suggests that smaller out of the box instruction/chat finetuned worker models with 3 billion parameters (hereafter, 3B) demonstrate a level of alignment with judge models comparable to that of larger finetuned worker models with 7 billion parameters (hereafter, 7B) across both simple and higher order reasoning tasks. The top performing 3B worker model pair achieve an agreement rate of approximately 80-83% for sentiment and around 50-80% for similar ticket when compared to judge models. Additionally, 3B worker models provide a speedup ranging from 2.8x to 9x relative to the judge models, while 7B worker model combinations achieve a speedup ranging from 1.28x to 0.28x
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。