CLSGen: A Dual-Head Fine-Tuning Framework for Joint Probabilistic Classification and Verbalized Explanation
Authors: WonJin Yoon, Kangyu Zhu, Ian Bulovic, Autumn Sehy, Yanjun Gao, Dmitriy Dligach, Majid Afshar, Timothy A. Miller
Organizations: Boston Children’s Hospital · Harvard Medical School · University of Colorado Anschutz Medical Campus · Loyola University Chicago · University of Wisconsin Madison
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.
Figures & tables
Figure 1: Classification and generation quality during classification-only training.
Figure 2: Diagram of data construction pipeline (left) and CLSGen fine-tuning (right).
Dataset
Train
Dev.
Test
Total
LCD benchmark
34,759
7,505
7,568
49,832
CliniFact
1,260
316
394
1,970
Table 1: Dataset statistics. Both 30 and 90 days tasks of the LCD benchmark have identical splits.
Inference-only methods
Traditional
CLSGen (Ours)
Dataset/task
Metric
Verb prob
Label pred
Self cons
CNN
Default
Threshold
LCD benchmark 30 days
AUROC
0.6598
N/A
0.7455
0.8582
0.8586
F1
0.1409
0.2023
0.2853
0.2182
0.2263
0.2455
Prec.
0.0809
0.1443
0.2315
0.5889
0.5606
0.5616
Recall
0.5467
0.3383
0.3716
0.1364
0.1418
0.1571
Parsable
0.8098
0.8660
0.9432
N/A
0.9991
Table 2: Performance comparison across datasets and methods. Default uses a fixed threshold of 0.5, while Threshold uses thresholds tuned on the development set. Standard deviations are provided in Appendix A.1 (Table 4 ).
Threshold-free
Cohen’s Kappa
LLM-as-a-Judge
Dataset
AUROC-Align
Default
Threshold
RLI
R-L Kappa ( κRL )
Readability
LCD 30 days
0.9841
0.6074
0.6187
0.0003
0.9872
0.9996
LCD 90 days
0.9597
0.6060
0.6014
0.0018
0.9751
0.9937
CliniFact
0.9891
0.8709
0.8802
0.0482
0.8910
1.0000
Table 3: Evaluation of classification-generated text consistency metrics. RLI stands for Rationale-Label Inconsistency, and R-L Kappa stands for Rationale-Label Kappa metric.
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
Dataset/task
Metric
Verb prob
Label pred
CNN
LCD benchmark 30 days
AUROC
0.0124
N/A
0.0015
F1
0.0047
0.0123
0.0457
Prec.
0.0027
0.0097
0.0315
Recall
0.0173
0.0169
0.0354
Parsable
0.0021
0.0025
N/A
LCD benchmark 90 days
AUROC
0.0110
N/A
0.0049
Appendix
Table 4: Standard deviations for the performance results in Table 2 , computed over 10 independent runs. AUROC is unavailable for label prediction, and parsability is not applicable to CNN.
Hyperparameter
Value
Base Model
Llama-3.1-8B-Instruct
Max Token Length (Input / Output)
7,040 / 384
Training Epochs
20 (LCD) / 20 (CliniFact)
Batch Size (per GPU)
1
Gradient Accumulation Steps
8
Effective Batch Size
32
Appendix
Table 5: Hyperparameters for CLSGen fine-tuning.
Configuration
Value
Grid Search
Learning Rate
{5e-4, 1e-4, 5e-5}
Fixed Hyperparameters
Batch Size
16
Max Epochs
10
Optimizer
AdamW
Appendix
Table 6: Hyperparameters and architecture for the CNN baseline.
Dataset
Status
False
True
True / Total
LCD-bench 30 days (Dev)
Original dataset
7,203
302
4.0%
Successfully generated
7,102
164
2.3%
Failed to generate
101
138
57.7%
LCD-bench 90 days (Dev)
Original dataset
7,203
302
4.0%
Successfully generated
6,950
198
2.8%
Failed to generate
253
104
29.1%
Appendix
Table 7: Development dataset data processing pipeline statistics.
Ping An Technology (Shenzhen) Co., Ltd., Shenzhen, China · Tsinghua Shenzhen International Graduate School, Shenzhen, China · Hubei University of Technology, Hubei, China