让大模型既能准确分类,又能生成可信解释。
CLSGen: A Dual-Head Fine-Tuning Framework for Joint Probabilistic Classification and Verbalized Explanation
- 双头架构同时训练分类与解释生成,避免能力退化。
- 在多个数据集上分类准确率优于基线模型,AUROC和F1提升显著。
- 适合需要可解释决策的场景,如医疗、金融风控。
随着大语言模型(LLM)的发展,其在处理长上下文和生成自然语言解释方面展现出巨大潜力。然而,将LLM用于实际决策时,缺乏可靠的定量概率输出成为关键障碍。传统微调方法虽能生成概率估计,但常引发灾难性遗忘和语言坍缩,导致解释生成能力丧失,削弱模型可解释性。为此,我们提出CLSGen,一种专为二分类任务设计的新型微调框架,包含新模型结构、训练策略与数据构建方法,实现稳健的概率估计而不牺牲解释生成能力。在多个基准数据集上的实验表明,使用CLSGen微调的模型在分类性能(AUROC和F1-score)上超越现有基线;解释结果与预测标签高度一致且可读性强。
原文摘要 · Abstract (English)
With the recent progress of Large Language Models (LLMs), there is a growing interest in applying these models to solve complex and challenging problems. Modern LLMs, capable of processing long contexts and generating verbalized explanations, offer significant potential in addressing real-world applications. However, a critical hurdle in deploying LLMs for practical decision-making is their inability to provide reliable, quantitative probabilities. While task-specific fine-tuning of LLMs using traditional discriminative objectives (similar to encoder-only models) can yield probability estimates, this often leads to catastrophic forgetting and linguistic collapse. Consequently, the model loses its ability to generate explanations, severely undermining its interpretability and usability. To address this challenge, we propose CLSGen, a novel LLM fine-tuning framework designed for binary classification tasks. The CLSGen framework encompasses a new model architecture, training methodology, and data construction strategy to enable robust probability estimation without sacrificing the model's inherent explanation-generation capabilities. Experimental results across multiple benchmark datasets demonstrate that models fine-tuned with CLSGen outperform existing baselines in classification metrics (AUROC and F1-score). Regarding explanation, the results showed strong alignment between predicted labels and generated justifications, as well as high readability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。